added swap video in all faces function

.gitignore

@@ -1,6 +1,7 @@
 venv310/
 SinglePhoto/
 VideoSwapping/
+VideoSwappingAllFaces/
 MultiSrcSingleDst/
 MultiSrcMultiDst/
 SingleSrcMultiDst/

SinglePhoto.py

@@ -35,6 +35,17 @@ class FaceSwapper:
         result = self.swapper.get(target_img, target_face, source_face, paste_back=True)
         return result
 
+    def count_faces(self, img_path):
+        """
+        Counts the number of faces in the given image file.
+        """
+        img = cv2.imread(img_path)
+        # Use your face detector here. For example, with OpenCV's Haar cascade:
+        face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + "haarcascade_frontalface_default.xml")
+        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+        faces = face_cascade.detectMultiScale(gray, 1.1, 4)
+        return len(faces)
+
 def main():
     # Paths relative to root
     source_path = os.path.join("SinglePhoto", "data_src.jpg")

app.py

@@ -319,6 +319,111 @@ 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.
+    """
+    log = ""
+    src_path = "VideoSwappingAllFaces/data_src.jpg"
+    dst_video_path = "VideoSwappingAllFaces/data_dst.mp4"
+    frames_dir = "VideoSwappingAllFaces/video_frames"
+    swapped_dir = "VideoSwappingAllFaces/swapped_frames"
+    output_video_path = "VideoSwappingAllFaces/output_tmp_output_video.mp4"
+    final_output_path = "VideoSwappingAllFaces/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))
+    temp_dir = os.path.join(swapped_dir, "temp_swap")
+    os.makedirs(temp_dir, exist_ok=True)
+    start_time = time.time()
+    total_frames = len(frame_paths)
+
+    for idx, frame_path in enumerate(frame_paths):
+        swapped_name = f"swapped_{idx:05d}.jpg"
+        out_path = os.path.join(swapped_dir, swapped_name)
+        temp_frame_path = os.path.join(temp_dir, "temp.jpg")
+        if swapped_name in swapped_files and os.path.exists(out_path):
+            log += f"Frame {idx}: already swapped, skipping.\n"
+            elapsed = time.time() - start_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:
+            shutil.copy(frame_path, temp_frame_path)
+            for face_idx in range(1, int(num_faces_to_swap) + 1):
+                try:
+                    swapped_img = swapper.swap_faces(src_path, 1, temp_frame_path, face_idx)
+                    cv2.imwrite(temp_frame_path, swapped_img)
+                except Exception as e:
+                    log += f"Failed to swap face {face_idx} in frame {idx}: {e}\n"
+            shutil.copy(temp_frame_path, out_path)
+            log += f"Swapped all faces in frame {idx} and saved to {out_path}\n"
+            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}: {e}\n"
+        elapsed = time.time() - start_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")
+    if os.path.exists(temp_dir):
+        shutil.rmtree(temp_dir)
+
+    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 os.path.exists(frames_dir):
+            shutil.rmtree(frames_dir)
+        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")
+    return final_output_path, log
+
 with gr.Blocks() as demo:
     gr.Markdown(wellcomingMessage)
     with gr.Tab("Single Photo Swapping"):
@@ -349,6 +454,19 @@ with gr.Blocks() as demo:
                 gr.Textbox(label="Log Output", lines=8, interactive=False)
             ],
         )
+    with gr.Tab("Video All Faces"):
+        gr.Interface(
+            fn=swap_video_all_faces,
+            inputs=[
+                gr.Image(label="Source Image"),
+                gr.Video(label="Target Video"),
+                gr.Number(value=1, label="Number of Faces to Swap"),
+            ],
+            outputs=[
+                gr.Video(label="Swapped Video"),
+                gr.Textbox(label="Log Output", lines=8, interactive=False)
+            ],
+        )
     with gr.Tab("SingleSrc MultiDst"):
         gr.Interface(
             fn=swap_single_src_multi_dst,
@@ -388,6 +506,7 @@ with gr.Blocks() as demo:
                 gr.Textbox(label="Log Output", lines=8, interactive=False)
             ],
         )
+    
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=False)