added checkbox of audio mixing and share commandline

app.py

@@ -6,6 +6,7 @@ import shutil
 import subprocess
 import time
 from SinglePhoto import FaceSwapper
+import argparse
 
 wellcomingMessage = """
     <h1>Face Swapping Suite</h1>
@@ -14,6 +15,7 @@ wellcomingMessage = """
 
 swapper = FaceSwapper()
 
+#Photo Swapping Functions
 def swap_single_photo(src_img, src_idx, dst_img, dst_idx, progress=gr.Progress(track_tqdm=True)):
     log = ""
     start_time = time.time()
@@ -54,123 +56,43 @@ def swap_single_photo(src_img, src_idx, dst_img, dst_idx, progress=gr.Progress(t
         log += f"Elapsed time: {elapsed:.2f} seconds\n"
         return None, log
 
-def swap_video(src_img, src_idx, video, dst_idx, delete_frames_dir=True, progress=gr.Progress()):
+def swap_single_src_multi_dst(src_img, dst_imgs, dst_indices, progress=gr.Progress(track_tqdm=True)):
     log = ""
-    start_time = time.time()
-    src_path = "VideoSwapping/data_src.jpg"
-    dst_video_path = "VideoSwapping/data_dst.mp4"
-    frames_dir = "VideoSwapping/video_frames"
-    swapped_dir = "VideoSwapping/swapped_frames"
-    output_video_path = "VideoSwapping/output_tmp_output_video.mp4"
-    final_output_path = "VideoSwapping/output_with_audio.mp4"
-
-    os.makedirs(os.path.dirname(src_path), exist_ok=True)
-    os.makedirs(os.path.dirname(dst_video_path), exist_ok=True)
-    os.makedirs(frames_dir, exist_ok=True)
-    os.makedirs(swapped_dir, exist_ok=True)
-
-    src_img_bgr = cv2.cvtColor(src_img, cv2.COLOR_RGB2BGR)
-    cv2.imwrite(src_path, src_img_bgr)
-    log += f"Saved source image to {src_path}\n"
-    progress(0.05, desc="Saved source image")
-
-    if isinstance(video, str) and os.path.exists(video):
-        shutil.copy(video, dst_video_path)
-        log += f"Copied video to {dst_video_path}\n"
-    else:
-        dst_video_path = video
-
-    from VideoSwapping import extract_frames, frames_to_video
+    results = []
+    src_dir = "SingleSrcMultiDst/src"
+    dst_dir = "SingleSrcMultiDst/dst"
+    output_dir = "SingleSrcMultiDst/output"
+    os.makedirs(src_dir, exist_ok=True)
+    os.makedirs(dst_dir, exist_ok=True)
+    os.makedirs(output_dir, exist_ok=True)
 
-    frame_paths = extract_frames(dst_video_path, frames_dir)
-    log += f"Extracted {len(frame_paths)} frames to {frames_dir}\n"
-    progress(0.15, desc="Extracted frames")
+    if isinstance(dst_img, tuple):
+        dst_img = dst_img[0]
+    dst_img_bgr = cv2.cvtColor(dst_img, cv2.COLOR_RGB2BGR)
+    dst_path = os.path.join(dst_dir, "data_dst.jpg")
+    cv2.imwrite(dst_path, dst_img_bgr)
+    log += f"Saved destination image to {dst_path}\n"
+    progress(0.05, desc="Saved destination image")
 
-    swapped_files = set(os.listdir(swapped_dir))
-    total_frames = len(frame_paths)
-    start_loop_time = time.time()
-    for idx, frame_path in enumerate(frame_paths):
-        swapped_name = f"swapped_{idx:05d}.jpg"
-        out_path = os.path.join(swapped_dir, swapped_name)
-        if swapped_name in swapped_files and os.path.exists(out_path):
-            log += f"Frame {idx}: already swapped, skipping.\n"
-            elapsed = time.time() - start_loop_time
-            avg_time = elapsed / (idx + 1) if idx + 1 > 0 else 0
-            remaining = avg_time * (total_frames - (idx + 1))
-            mins, secs = divmod(int(remaining), 60)
-            progress(0.15 + 0.6 * (idx + 1) / total_frames, desc=f"Swapping {idx+1}/{total_frames} | {mins:02d}:{secs:02d} left")
-            continue
+    for i, src_img in enumerate(src_imgs):
+        if isinstance(src_img, tuple):
+            src_img = src_img[0]
+        src_path = os.path.join(src_dir, f"data_src_{i}.jpg")
+        output_path = os.path.join(output_dir, f"output_swapped_{i}.jpg")
+        src_img_bgr = cv2.cvtColor(src_img, cv2.COLOR_RGB2BGR)
+        cv2.imwrite(src_path, src_img_bgr)
+        log += f"Saved source image {i} to {src_path}\n"
         try:
-            try:
-                swapped = swapper.swap_faces(src_path, int(src_idx), frame_path, int(dst_idx))
-            except ValueError as ve:
-                if int(dst_idx) != 1 and "Target image contains" in str(ve):
-                    swapped = swapper.swap_faces(src_path, int(src_idx), frame_path, 1)
-                    log += f"Frame {idx}: dst_idx {dst_idx} not found, used 1 instead.\n"
-                else:
-                    raise ve
-            cv2.imwrite(out_path, swapped)
-            log += f"Swapped frame {idx} and saved to {out_path}\n"
+            result = swapper.swap_faces(src_path, 1, dst_path, int(dst_idx))
+            cv2.imwrite(output_path, result)
+            results.append(output_path)
+            log += f"Swapped and saved result to {output_path}\n"
         except Exception as e:
-            cv2.imwrite(out_path, cv2.imread(frame_path))
-            log += f"Failed to swap frame {idx}: {e}\n"
-        elapsed = time.time() - start_loop_time
-        avg_time = elapsed / (idx + 1) if idx + 1 > 0 else 0
-        remaining = avg_time * (total_frames - (idx + 1))
-        mins, secs = divmod(int(remaining), 60)
-        progress(0.15 + 0.6 * (idx + 1) / total_frames, desc=f"Swapping {idx+1}/{total_frames} | {mins:02d}:{secs:02d} left")
-    cap = cv2.VideoCapture(dst_video_path)
-    fps = cap.get(cv2.CAP_PROP_FPS)
-    cap.release()
-    frames_to_video(swapped_dir, output_video_path, fps)
-    log += f"Combined swapped frames into video {output_video_path}\n"
-    progress(0.8, desc="Muxing audio")
-
-    ok, audio_log = add_audio_to_video(dst_video_path, output_video_path, final_output_path)
-    if ok:
-        log += f"Added audio to {final_output_path}\n"
-    else:
-        log += f"Audio muxing failed: {audio_log}\n"
-        final_output_path = output_video_path
-
-    try:
-        if os.path.exists(src_path):
-            os.remove(src_path)
-        if os.path.exists(dst_video_path):
-            os.remove(dst_video_path)
-        if delete_frames_dir and os.path.exists(frames_dir):
-            shutil.rmtree(frames_dir)
-            log += "Deleted video_frames directory.\n"
-        elif not delete_frames_dir:
-            log += "Kept video_frames directory as requested.\n"
-        if os.path.exists(swapped_dir):
-            shutil.rmtree(swapped_dir)
-        log += "Cleaned up temp files and folders.\n"
-    except Exception as cleanup_error:
-        log += f"Cleanup error: {cleanup_error}\n"
+            results.append(f"Error: {e}")
+            log += f"Error swapping source {i}: {e}\n"
+        progress((i + 1) / len(src_imgs), desc=f"Swapping source {i+1}/{len(src_imgs)}")
     progress(1, desc="Done")
-    elapsed = time.time() - start_time
-    log += f"Elapsed time: {elapsed:.2f} seconds\n"
-    return final_output_path, log
-
-def add_audio_to_video(original_video_path, video_no_audio_path, output_path):
-    cmd = [
-        "ffmpeg",
-        "-y",
-        "-i", video_no_audio_path,
-        "-i", original_video_path,
-        "-c:v", "copy",
-        "-c:a", "aac",
-        "-map", "0:v:0",
-        "-map", "1:a:0?",
-        "-shortest",
-        output_path
-    ]
-    try:
-        subprocess.run(cmd, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-        return True, ""
-    except subprocess.CalledProcessError as e:
-        return False, e.stderr.decode()
+    return results, log
 
 def swap_multi_src_single_dst(src_imgs, dst_img, dst_idx, progress=gr.Progress(track_tqdm=True)):
     log = ""
@@ -324,7 +246,205 @@ def swap_single_src_multi_dst(src_img, dst_imgs, dst_indices, progress=gr.Progre
     progress(1, desc="Done")
     return results, log
 
-def swap_video_all_faces(src_img, video, num_faces_to_swap, delete_frames_dir=True, progress=gr.Progress()):
+def swap_faces_custom(src_imgs, dst_img, mapping_str, progress=gr.Progress(track_tqdm=True)):
+    """
+    src_imgs: list of source images (numpy arrays)
+    dst_img: destination image (numpy array)
+    mapping_str: comma-separated string, e.g. "2,1,3"
+    """
+    log = ""
+    start_time = time.time()
+    dst_path = "CustomSwap/data_dst.jpg"
+    output_path = "CustomSwap/output_swapped.jpg"
+    src_dir = "CustomSwap/src"
+    temp_dir = "CustomSwap/temp"
+    os.makedirs(src_dir, exist_ok=True)
+    os.makedirs(temp_dir, exist_ok=True)
+    os.makedirs(os.path.dirname(dst_path), exist_ok=True)
+    os.makedirs(os.path.dirname(output_path), exist_ok=True)
+
+    # Save destination image
+    dst_img_bgr = cv2.cvtColor(dst_img, cv2.COLOR_RGB2BGR)
+    cv2.imwrite(dst_path, dst_img_bgr)
+    log += f"Saved destination image to {dst_path}\n"
+
+    # Save all source images
+    src_paths = []
+    for i, src_img in enumerate(src_imgs):
+        src_path = os.path.join(src_dir, f"data_src_{i+1}.jpg")
+        if isinstance(src_img, tuple):
+            src_img = src_img[0]
+        if src_img is None:
+            log += f"Source image {i+1} is None, skipping.\n"
+            continue
+        if isinstance(src_img, np.ndarray):
+            src_img_bgr = cv2.cvtColor(src_img, cv2.COLOR_RGB2BGR)
+            cv2.imwrite(src_path, src_img_bgr)
+            src_paths.append(src_path)
+            log += f"Saved source image {i+1} to {src_path}\n"
+        elif isinstance(src_img, str) and os.path.exists(src_img):
+            shutil.copy(src_img, src_path)
+            src_paths.append(src_path)
+            log += f"Copied source image {i+1} from {src_img} to {src_path}\n"
+        else:
+            log += f"Source image {i+1} is not a valid image, skipping.\n"
+
+    # Parse mapping
+    try:
+        mapping = [int(x.strip()) for x in mapping_str.split(",") if x.strip().isdigit()]
+    except Exception as e:
+        log += f"Error parsing mapping: {e}\n"
+        elapsed = time.time() - start_time
+        log += f"Elapsed time: {elapsed:.2f} seconds\n"
+        return None, log
+
+    # Use a temp file for iterative swapping
+    temp_dst_path = os.path.join(temp_dir, "temp_dst.jpg")
+    shutil.copy(dst_path, temp_dst_path)
+
+    for face_idx, src_idx in enumerate(mapping, start=1):
+        if src_idx < 1 or src_idx > len(src_paths):
+            log += f"Invalid source index {src_idx} for face {face_idx}, skipping.\n"
+            continue
+        try:
+            swapped_img = swapper.swap_faces(src_paths[src_idx-1], 1, temp_dst_path, face_idx)
+            cv2.imwrite(temp_dst_path, swapped_img)
+            log += f"Swapped face {face_idx} in destination with source {src_idx}\n"
+        except Exception as e:
+            log += f"Failed to swap face {face_idx} with source {src_idx}: {e}\n"
+
+    shutil.copy(temp_dst_path, output_path)
+    log += f"Saved swapped image to {output_path}\n"
+    if os.path.exists(temp_dst_path):
+        os.remove(temp_dst_path)
+    elapsed = time.time() - start_time
+    log += f"Elapsed time: {elapsed:.2f} seconds\n"
+    return output_path, log
+
+#Video Swapping Functions
+def add_audio_to_video(original_video_path, video_no_audio_path, output_path):
+    cmd = [
+        "ffmpeg",
+        "-y",
+        "-i", video_no_audio_path,
+        "-i", original_video_path,
+        "-c:v", "copy",
+        "-c:a", "aac",
+        "-map", "0:v:0",
+        "-map", "1:a:0?",
+        "-shortest",
+        output_path
+    ]
+    try:
+        subprocess.run(cmd, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        return True, ""
+    except subprocess.CalledProcessError as e:
+        return False, e.stderr.decode()
+
+def swap_video(src_img, src_idx, video, dst_idx, delete_frames_dir=True, add_audio=True, progress=gr.Progress()):
+    log = ""
+    start_time = time.time()
+    src_path = "VideoSwapping/data_src.jpg"
+    dst_video_path = "VideoSwapping/data_dst.mp4"
+    frames_dir = "VideoSwapping/video_frames"
+    swapped_dir = "VideoSwapping/swapped_frames"
+    output_video_path = "VideoSwapping/output_tmp_output_video.mp4"
+    final_output_path = "VideoSwapping/output_with_audio.mp4"
+
+    os.makedirs(os.path.dirname(src_path), exist_ok=True)
+    os.makedirs(os.path.dirname(dst_video_path), exist_ok=True)
+    os.makedirs(frames_dir, exist_ok=True)
+    os.makedirs(swapped_dir, exist_ok=True)
+
+    src_img_bgr = cv2.cvtColor(src_img, cv2.COLOR_RGB2BGR)
+    cv2.imwrite(src_path, src_img_bgr)
+    log += f"Saved source image to {src_path}\n"
+    progress(0.05, desc="Saved source image")
+
+    if isinstance(video, str) and os.path.exists(video):
+        shutil.copy(video, dst_video_path)
+        log += f"Copied video to {dst_video_path}\n"
+    else:
+        dst_video_path = video
+
+    from VideoSwapping import extract_frames, frames_to_video
+
+    frame_paths = extract_frames(dst_video_path, frames_dir)
+    log += f"Extracted {len(frame_paths)} frames to {frames_dir}\n"
+    progress(0.15, desc="Extracted frames")
+
+    swapped_files = set(os.listdir(swapped_dir))
+    total_frames = len(frame_paths)
+    start_loop_time = time.time()
+    for idx, frame_path in enumerate(frame_paths):
+        swapped_name = f"swapped_{idx:05d}.jpg"
+        out_path = os.path.join(swapped_dir, swapped_name)
+        if swapped_name in swapped_files and os.path.exists(out_path):
+            log += f"Frame {idx}: already swapped, skipping.\n"
+            elapsed = time.time() - start_loop_time
+            avg_time = elapsed / (idx + 1) if idx + 1 > 0 else 0
+            remaining = avg_time * (total_frames - (idx + 1))
+            mins, secs = divmod(int(remaining), 60)
+            progress(0.15 + 0.6 * (idx + 1) / total_frames, desc=f"Swapping {idx+1}/{total_frames} | {mins:02d}:{secs:02d} left")
+            continue
+        try:
+            try:
+                swapped = swapper.swap_faces(src_path, int(src_idx), frame_path, int(dst_idx))
+            except ValueError as ve:
+                if int(dst_idx) != 1 and "Target image contains" in str(ve):
+                    swapped = swapper.swap_faces(src_path, int(src_idx), frame_path, 1)
+                    log += f"Frame {idx}: dst_idx {dst_idx} not found, used 1 instead.\n"
+                else:
+                    raise ve
+            cv2.imwrite(out_path, swapped)
+            log += f"Swapped frame {idx} and saved to {out_path}\n"
+        except Exception as e:
+            cv2.imwrite(out_path, cv2.imread(frame_path))
+            log += f"Failed to swap frame {idx}: {e}\n"
+        elapsed = time.time() - start_loop_time
+        avg_time = elapsed / (idx + 1) if idx + 1 > 0 else 0
+        remaining = avg_time * (total_frames - (idx + 1))
+        mins, secs = divmod(int(remaining), 60)
+        progress(0.15 + 0.6 * (idx + 1) / total_frames, desc=f"Swapping {idx+1}/{total_frames} | {mins:02d}:{secs:02d} left")
+    cap = cv2.VideoCapture(dst_video_path)
+    fps = cap.get(cv2.CAP_PROP_FPS)
+    cap.release()
+    frames_to_video(swapped_dir, output_video_path, fps)
+    log += f"Combined swapped frames into video {output_video_path}\n"
+    progress(0.8, desc="Muxing audio")
+
+    if add_audio:
+        ok, audio_log = add_audio_to_video(dst_video_path, output_video_path, final_output_path)
+        if ok:
+            log += f"Added audio to {final_output_path}\n"
+        else:
+            log += f"Audio muxing failed: {audio_log}\n"
+            final_output_path = output_video_path
+    else:
+        final_output_path = output_video_path
+        log += "Audio was not added as per user request.\n"
+
+    try:
+        if os.path.exists(src_path):
+            os.remove(src_path)
+        if os.path.exists(dst_video_path):
+            os.remove(dst_video_path)
+        if delete_frames_dir and os.path.exists(frames_dir):
+            shutil.rmtree(frames_dir)
+            log += "Deleted video_frames directory.\n"
+        elif not delete_frames_dir:
+            log += "Kept video_frames directory as requested.\n"
+        if os.path.exists(swapped_dir):
+            shutil.rmtree(swapped_dir)
+        log += "Cleaned up temp files and folders.\n"
+    except Exception as cleanup_error:
+        log += f"Cleanup error: {cleanup_error}\n"
+    progress(1, desc="Done")
+    elapsed = time.time() - start_time
+    log += f"Elapsed time: {elapsed:.2f} seconds\n"
+    return final_output_path, log
+
+def swap_video_all_faces(src_img, video, num_faces_to_swap, delete_frames_dir=True, add_audio=True, progress=gr.Progress()):
     log = ""
     start_time = time.time()
     src_path = "VideoSwappingAllFaces/data_src.jpg"
@@ -404,12 +524,16 @@ def swap_video_all_faces(src_img, video, num_faces_to_swap, delete_frames_dir=Tr
     log += f"Combined swapped frames into video {output_video_path}\n"
     progress(0.8, desc="Muxing audio")
 
-    ok, audio_log = add_audio_to_video(dst_video_path, output_video_path, final_output_path)
-    if ok:
-        log += f"Added audio to {final_output_path}\n"
+    if add_audio:
+        ok, audio_log = add_audio_to_video(dst_video_path, output_video_path, final_output_path)
+        if ok:
+            log += f"Added audio to {final_output_path}\n"
+        else:
+            log += f"Audio muxing failed: {audio_log}\n"
+            final_output_path = output_video_path
     else:
-        log += f"Audio muxing failed: {audio_log}\n"
         final_output_path = output_video_path
+        log += "Audio was not added as per user request.\n"
 
     try:
         if os.path.exists(src_path):
@@ -431,88 +555,7 @@ def swap_video_all_faces(src_img, video, num_faces_to_swap, delete_frames_dir=Tr
     log += f"Elapsed time: {elapsed:.2f} seconds\n"
     return final_output_path, log
 
-def swap_faces_custom(src_imgs, dst_img, mapping_str, progress=gr.Progress(track_tqdm=True)):
-    """
-    src_imgs: list of source images (numpy arrays)
-    dst_img: destination image (numpy array)
-    mapping_str: comma-separated string, e.g. "2,1,3"
-    """
-    log = ""
-    start_time = time.time()
-    dst_path = "CustomSwap/data_dst.jpg"
-    output_path = "CustomSwap/output_swapped.jpg"
-    src_dir = "CustomSwap/src"
-    temp_dir = "CustomSwap/temp"
-    os.makedirs(src_dir, exist_ok=True)
-    os.makedirs(temp_dir, exist_ok=True)
-    os.makedirs(os.path.dirname(dst_path), exist_ok=True)
-    os.makedirs(os.path.dirname(output_path), exist_ok=True)
-
-    # Save destination image
-    dst_img_bgr = cv2.cvtColor(dst_img, cv2.COLOR_RGB2BGR)
-    cv2.imwrite(dst_path, dst_img_bgr)
-    log += f"Saved destination image to {dst_path}\n"
-
-    # Save all source images
-    src_paths = []
-    for i, src_img in enumerate(src_imgs):
-        src_path = os.path.join(src_dir, f"data_src_{i+1}.jpg")
-        if isinstance(src_img, tuple):
-            src_img = src_img[0]
-        if src_img is None:
-            log += f"Source image {i+1} is None, skipping.\n"
-            continue
-        if isinstance(src_img, np.ndarray):
-            src_img_bgr = cv2.cvtColor(src_img, cv2.COLOR_RGB2BGR)
-            cv2.imwrite(src_path, src_img_bgr)
-            src_paths.append(src_path)
-            log += f"Saved source image {i+1} to {src_path}\n"
-        elif isinstance(src_img, str) and os.path.exists(src_img):
-            shutil.copy(src_img, src_path)
-            src_paths.append(src_path)
-            log += f"Copied source image {i+1} from {src_img} to {src_path}\n"
-        else:
-            log += f"Source image {i+1} is not a valid image, skipping.\n"
-
-    # Parse mapping
-    try:
-        mapping = [int(x.strip()) for x in mapping_str.split(",") if x.strip().isdigit()]
-    except Exception as e:
-        log += f"Error parsing mapping: {e}\n"
-        elapsed = time.time() - start_time
-        log += f"Elapsed time: {elapsed:.2f} seconds\n"
-        return None, log
-
-    # Use a temp file for iterative swapping
-    temp_dst_path = os.path.join(temp_dir, "temp_dst.jpg")
-    shutil.copy(dst_path, temp_dst_path)
-
-    for face_idx, src_idx in enumerate(mapping, start=1):
-        if src_idx < 1 or src_idx > len(src_paths):
-            log += f"Invalid source index {src_idx} for face {face_idx}, skipping.\n"
-            continue
-        try:
-            swapped_img = swapper.swap_faces(src_paths[src_idx-1], 1, temp_dst_path, face_idx)
-            cv2.imwrite(temp_dst_path, swapped_img)
-            log += f"Swapped face {face_idx} in destination with source {src_idx}\n"
-        except Exception as e:
-            log += f"Failed to swap face {face_idx} with source {src_idx}: {e}\n"
-
-    shutil.copy(temp_dst_path, output_path)
-    log += f"Saved swapped image to {output_path}\n"
-    if os.path.exists(temp_dst_path):
-        os.remove(temp_dst_path)
-    elapsed = time.time() - start_time
-    log += f"Elapsed time: {elapsed:.2f} seconds\n"
-    return output_path, log
-
-def swap_video_custom_mapping(src_imgs, video, mapping_str, delete_frames_dir=True, progress=gr.Progress()):
-    """
-    Swaps faces in each frame of the video according to the user mapping.
-    src_imgs: list of source images (numpy arrays)
-    video: path or numpy array of the video
-    mapping_str: comma-separated string, e.g. "2,1,3"
-    """
+def swap_video_custom_mapping(src_imgs, video, mapping_str, delete_frames_dir=True, add_audio=True, progress=gr.Progress()):
     log = ""
     start_time = time.time()
     src_dir = "CustomVideoSwap/src"
@@ -621,12 +664,16 @@ def swap_video_custom_mapping(src_imgs, video, mapping_str, delete_frames_dir=Tr
     log += f"Combined swapped frames into video {output_video_path}\n"
     progress(0.9, desc="Muxing audio")
 
-    ok, audio_log = add_audio_to_video(dst_video_path, output_video_path, final_output_path)
-    if ok:
-        log += f"Added audio to {final_output_path}\n"
+    if add_audio:
+        ok, audio_log = add_audio_to_video(dst_video_path, output_video_path, final_output_path)
+        if ok:
+            log += f"Added audio to {final_output_path}\n"
+        else:
+            log += f"Audio muxing failed: {audio_log}\n"
+            final_output_path = output_video_path
     else:
-        log += f"Audio muxing failed: {audio_log}\n"
         final_output_path = output_video_path
+        log += "Audio was not added as per user request.\n"
 
     try:
         if os.path.exists(dst_video_path):
@@ -646,6 +693,7 @@ def swap_video_custom_mapping(src_imgs, video, mapping_str, delete_frames_dir=Tr
     log += f"Elapsed time: {elapsed:.2f} seconds\n"
     return final_output_path, log
 
+
 # Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown(wellcomingMessage)
@@ -724,7 +772,8 @@ with gr.Blocks() as demo:
                 gr.Number(value=1, label="Source Face Index"),
                 gr.Video(label="Target Video"),
                 gr.Number(value=1, label="Destination Face Index"),
-                gr.Checkbox(label="Delete video_frames directory after processing", value=True)
+                gr.Checkbox(label="Delete video_frames directory after processing", value=True),
+                gr.Checkbox(label="Add audio from original video", value=True)
             ],
             outputs=[
                 gr.Video(label="Swapped Video"),
@@ -738,7 +787,8 @@ with gr.Blocks() as demo:
                 gr.Image(label="Source Image"),
                 gr.Video(label="Target Video"),
                 gr.Number(value=1, label="Number of Faces to Swap"),
-                gr.Checkbox(label="Delete video_frames directory after processing", value=True)
+                gr.Checkbox(label="Delete video_frames directory after processing", value=True),
+                gr.Checkbox(label="Add audio from original video", value=True)
             ],
             outputs=[
                 gr.Video(label="Swapped Video"),
@@ -752,7 +802,8 @@ with gr.Blocks() as demo:
                 gr.Gallery(label="Source Images", type="numpy", columns=3),
                 gr.Video(label="Target Video"),
                 gr.Textbox(label="Mapping (comma-separated, e.g. 2,1,3)"),
-                gr.Checkbox(label="Delete video_frames directory after processing", value=True)
+                gr.Checkbox(label="Delete video_frames directory after processing", value=True),
+                gr.Checkbox(label="Add audio from original video", value=True)
             ],
             outputs=[
                 gr.Video(label="Swapped Video"),
@@ -762,4 +813,7 @@ with gr.Blocks() as demo:
     
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--share", action="store_true", help="Launch Gradio with share=True")
+    args = parser.parse_args()
+    demo.launch(share=args.share)