Update app.py

app.py

@@ -151,46 +151,6 @@ 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]}")
@@ -410,32 +370,7 @@ 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):
     """
     Always runs CodeFormer on the final image.
@@ -910,19 +845,14 @@ async def multi_face_swap_api(
 
 # --------------------- Mount Gradio ---------------------
 
-# multi_faceswap_app = build_multi_faceswap_gradio()
-# fastapi_app = mount_gradio_app(
-#     fastapi_app,
-#     multi_faceswap_app,
-#     path="/gradio-multi"
-# )
-video_faceswap_app = build_video_faceswap_gradio()
+multi_faceswap_app = build_multi_faceswap_gradio()
 fastapi_app = mount_gradio_app(
     fastapi_app,
-    video_faceswap_app,
-    path="/gradio-video"
+    multi_faceswap_app,
+    path="/gradio-couple-faceswap"
 )
 
 
+
 if __name__ == "__main__":
     uvicorn.run(fastapi_app, host="0.0.0.0", port=7860)