added checkbox to check if user want to delete the video frames dir in video swapping

.gitignore

@@ -1,5 +1,6 @@
 venv310/
 SinglePhoto/
+MultiSrc/
 VideoSwapping/
 VideoSwappingAllFaces/
 MultiSrcSingleDst/

app.py

@@ -54,7 +54,7 @@ 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, progress=gr.Progress()):
+def swap_video(src_img, src_idx, video, dst_idx, delete_frames_dir=True, progress=gr.Progress()):
     log = ""
     start_time = time.time()
     src_path = "VideoSwapping/data_src.jpg"
@@ -82,12 +82,10 @@ def swap_video(src_img, src_idx, video, dst_idx, progress=gr.Progress()):
 
     from VideoSwapping import extract_frames, frames_to_video
 
-    # Extract frames only if not already present
     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")
 
-    # Prepare swapped frames list and resume if possible
     swapped_files = set(os.listdir(swapped_dir))
     total_frames = len(frame_paths)
     start_loop_time = time.time()
@@ -128,21 +126,23 @@ def swap_video(src_img, src_idx, video, dst_idx, progress=gr.Progress()):
     log += f"Combined swapped frames into video {output_video_path}\n"
     progress(0.8, desc="Muxing audio")
 
-    # Add audio from original video
     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  # fallback to video without audio
+        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 os.path.exists(frames_dir):
+        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"
@@ -154,9 +154,6 @@ def swap_video(src_img, src_idx, video, dst_idx, progress=gr.Progress()):
     return final_output_path, log
 
 def add_audio_to_video(original_video_path, video_no_audio_path, output_path):
-    """
-    Uses ffmpeg to mux audio from original_video_path into video_no_audio_path.
-    """
     cmd = [
         "ffmpeg",
         "-y",
@@ -327,11 +324,7 @@ 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, progress=gr.Progress()):
-    """
-    Swaps the single source image to all faces in each frame of the destination video.
-    For each frame, swaps one by one, always using the latest swapped image for the next face.
-    """
+def swap_video_all_faces(src_img, video, num_faces_to_swap, delete_frames_dir=True, progress=gr.Progress()):
     log = ""
     start_time = time.time()
     src_path = "VideoSwappingAllFaces/data_src.jpg"
@@ -423,8 +416,11 @@ def swap_video_all_faces(src_img, video, num_faces_to_swap, progress=gr.Progress
             os.remove(src_path)
         if os.path.exists(dst_video_path):
             os.remove(dst_video_path)
-        if os.path.exists(frames_dir):
+        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"
@@ -510,7 +506,7 @@ def swap_faces_custom(src_imgs, dst_img, mapping_str, progress=gr.Progress(track
     log += f"Elapsed time: {elapsed:.2f} seconds\n"
     return output_path, log
 
-def swap_video_custom_mapping(src_imgs, video, mapping_str, progress=gr.Progress()):
+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)
@@ -635,8 +631,11 @@ def swap_video_custom_mapping(src_imgs, video, mapping_str, progress=gr.Progress
     try:
         if os.path.exists(dst_video_path):
             os.remove(dst_video_path)
-        if os.path.exists(frames_dir):
+        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"
@@ -647,7 +646,7 @@ def swap_video_custom_mapping(src_imgs, video, mapping_str, progress=gr.Progress
     log += f"Elapsed time: {elapsed:.2f} seconds\n"
     return final_output_path, log
 
-# Add this to your Gradio UI:
+# Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown(wellcomingMessage)
     with gr.Tab("Single Photo Swapping"):
@@ -725,6 +724,7 @@ 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)
             ],
             outputs=[
                 gr.Video(label="Swapped Video"),
@@ -738,6 +738,7 @@ 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)
             ],
             outputs=[
                 gr.Video(label="Swapped Video"),
@@ -751,6 +752,7 @@ 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)
             ],
             outputs=[
                 gr.Video(label="Swapped Video"),