AddedSingleSrcMultiVideo

.gitignore

@@ -8,6 +8,7 @@ MultiSrcMultiDst/
 SingleSrcMultiDst/
 CustomSwap/
 CustomVideoSwap/
+SingleSrcMultiVideo/
 __pycache__/*
 .gradio/
 

app.py

@@ -693,6 +693,118 @@ 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
 
+def swap_single_src_multi_video(src_img, dst_videos, dst_indices, delete_frames_dir=True, add_audio=True, progress=gr.Progress(track_tqdm=True)):
+    """
+    Swaps a single source image onto multiple videos, with a mapping for destination face indices.
+    Each video is processed one by one.
+    """
+    log = ""
+    results = []
+    start_time = time.time()
+    base_dir = "SingleSrcMultiVideo"
+    dst_dir = os.path.join(base_dir, "dst")
+    frames_dir = os.path.join(base_dir, "video_frames")
+    swapped_dir = os.path.join(base_dir, "swap_frames")
+    output_dir = os.path.join(base_dir, "output")
+    os.makedirs(dst_dir, exist_ok=True)
+    os.makedirs(frames_dir, exist_ok=True)
+    os.makedirs(swapped_dir, exist_ok=True)
+    os.makedirs(output_dir, exist_ok=True)
+
+    # Parse indices
+    if isinstance(dst_indices, str):
+        dst_indices_list = [int(idx.strip()) for idx in dst_indices.split(",") if idx.strip().isdigit()]
+    else:
+        dst_indices_list = [int(idx) for idx in dst_indices]
+
+    # Save source image
+    src_path = os.path.join(base_dir, "data_src.jpg")
+    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"
+
+    from VideoSwapping import extract_frames, frames_to_video
+
+    for i, video in enumerate(dst_videos):
+        dst_idx = dst_indices_list[i] if i < len(dst_indices_list) else 1
+        video_name = f"video_{i}.mp4"
+        dst_video_path = os.path.join(dst_dir, video_name)
+        output_video_path = os.path.join(output_dir, f"output_{i}.mp4")
+        output_video_with_audio = os.path.join(output_dir, f"output_with_audio_{i}.mp4")
+
+        # Copy video to dst_dir
+        if isinstance(video, str) and os.path.exists(video):
+            shutil.copy(video, dst_video_path)
+            log += f"Copied video {video} to {dst_video_path}\n"
+        else:
+            dst_video_path = video
+
+        # Extract frames
+        frame_paths = extract_frames(dst_video_path, frames_dir)
+        log += f"Extracted {len(frame_paths)} frames from {dst_video_path} to {frames_dir}\n"
+        progress(i / len(dst_videos), desc=f"Processing video {i+1}/{len(dst_videos)}")
+
+        swapped_files = set(os.listdir(swapped_dir))
+        total_frames = len(frame_paths)
+        temp_frame_path = os.path.join(swapped_dir, "temp.jpg")
+        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):
+                continue
+            try:
+                shutil.copy(frame_path, temp_frame_path)
+                try:
+                    swapped_img = swapper.swap_faces(src_path, 1, temp_frame_path, int(dst_idx))
+                    cv2.imwrite(temp_frame_path, swapped_img)
+                except Exception as e:
+                    log += f"Failed to swap face in frame {idx} of video {i}: {e}\n"
+                shutil.copy(temp_frame_path, out_path)
+                if os.path.exists(temp_frame_path):
+                    os.remove(temp_frame_path)
+            except Exception as e:
+                cv2.imwrite(out_path, cv2.imread(frame_path))
+                log += f"Failed to swap frame {idx} of video {i}: {e}\n"
+
+        # Combine frames to video
+        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"
+
+        # Mux audio if requested
+        if add_audio:
+            ok, audio_log = add_audio_to_video(dst_video_path, output_video_path, output_video_with_audio)
+            if ok:
+                log += f"Added audio to {output_video_with_audio}\n"
+                results.append(output_video_with_audio)
+            else:
+                log += f"Audio muxing failed for video {i}: {audio_log}\n"
+                results.append(output_video_path)
+        else:
+            results.append(output_video_path)
+            log += f"Audio was not added for video {i} as per user request.\n"
+
+        # Cleanup frames for next video
+        if delete_frames_dir and os.path.exists(frames_dir):
+            shutil.rmtree(frames_dir)
+            os.makedirs(frames_dir, exist_ok=True)
+            log += f"Deleted video_frames directory after video {i}.\n"
+        elif not delete_frames_dir:
+            log += f"Kept video_frames directory after video {i} as requested.\n"
+        if os.path.exists(swapped_dir):
+            shutil.rmtree(swapped_dir)
+            os.makedirs(swapped_dir, exist_ok=True)
+            log += f"Cleared swap_frames directory after video {i}.\n"
+
+    elapsed = time.time() - start_time
+    log += f"Elapsed time: {elapsed:.2f} seconds\n"
+    return results, log
+
+
 
 # Gradio UI
 with gr.Blocks() as demo:
@@ -810,6 +922,21 @@ with gr.Blocks() as demo:
                 gr.Textbox(label="Log Output", lines=8, interactive=False)
             ],
         )
+    with gr.Tab("SingleSrcMultiVideo"):
+        gr.Interface(
+            fn=swap_single_src_multi_video,
+            inputs=[
+                gr.Image(label="Source Image"),
+                gr.File(label="Target Videos (select multiple)", file_count="multiple", type="filepath"),
+                gr.Textbox(label="Destination Face Indices (comma-separated, e.g. 1,2,1)"),
+                gr.Checkbox(label="Delete video_frames directory after processing", value=True),
+                gr.Checkbox(label="Add audio from original video", value=True)
+            ],
+            outputs=[
+                gr.Gallery(label="Swapped Videos", type="filepath"),
+                gr.Textbox(label="Log Output", lines=8, interactive=False)
+            ],
+        )
     
 
 if __name__ == "__main__":