Update app.py

app.py

@@ -11,7 +11,7 @@ import logging
 import tempfile
 import sys
 from datetime import datetime,timedelta
-
+import tempfile
 import insightface
 from insightface.app import FaceAnalysis
 from huggingface_hub import hf_hub_download
@@ -151,6 +151,46 @@ async def log_faceswap_hit(token: str, status: str = "success"):
 # --------------------- Face Swap Pipeline ---------------------
 swap_lock = threading.Lock()
 
+def swap_faces_on_video(source_img, video_file):
+    """
+    source_img: numpy RGB image
+    video_file: path to uploaded video
+    Returns path to swapped video
+    """
+    temp_dir = tempfile.mkdtemp(prefix="faceswap_video_")
+    input_path = video_file.name  # Gradio gives a NamedTemporaryFile
+    output_path = os.path.join(temp_dir, f"swapped_{uuid.uuid4().hex}.mp4")
+
+    cap = cv2.VideoCapture(input_path)
+    fps = cap.get(cv2.CAP_PROP_FPS)
+    width  = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+
+    fourcc = cv2.VideoWriter_fourcc(*"mp4v")
+    out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
+
+    frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+    print(f"Processing {frame_count} frames...")
+
+    while True:
+        ret, frame = cap.read()
+        if not ret:
+            break
+
+        # Convert frame to RGB
+        frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+
+        # Apply existing functions
+        swapped_rgb = multi_face_swap(source_img, frame_rgb)
+        final_rgb = mandatory_enhancement(swapped_rgb)
+
+        # Convert back to BGR and write frame
+        out.write(cv2.cvtColor(final_rgb, cv2.COLOR_RGB2BGR))
+
+    cap.release()
+    out.release()
+    return 
+    
 def enhance_image_with_codeformer(rgb_img, temp_dir=None):
     if temp_dir is None:
         temp_dir = os.path.join(tempfile.gettempdir(), f"enhance_{uuid.uuid4().hex[:8]}")
@@ -370,6 +410,31 @@ def build_multi_faceswap_gradio():
         btn.click(process, [src, tgt], [out, error])
 
     return demo
+
+# ------------------- Gradio Interface -------------------
+def build_video_faceswap_gradio():
+    with gr.Blocks() as demo:
+        gr.Markdown("## 🎬 Video Face Swap")
+
+        with gr.Row():
+            src = gr.Image(type="numpy", label="Source Face Image")
+            vid = gr.File(label="Target Video (.mp4)")
+
+        out = gr.Video(label="Swapped Video")
+        error = gr.Textbox(label="Logs", interactive=False)
+
+        def process(src_img, video_file):
+            try:
+                swapped_video = swap_faces_on_video(src_img, video_file)
+                return swapped_video, ""
+            except Exception as e:
+                return None, str(e)
+
+        btn = gr.Button("Swap Faces in Video")
+        btn.click(process, [src, vid], [out, error])
+
+    return demo
+
     
 def mandatory_enhancement(rgb_img):
     """