Upload 3 files

app.py

@@ -0,0 +1,29 @@
+
+import gradio as gr
+import os
+from run import run_faceswap
+
+def process_face_swap(source_image, target_media):
+    source_path = "face.png"
+    target_path = "target"
+
+    source_image.save(source_path)
+    ext = os.path.splitext(target_media.name)[-1]
+    target_path += ext
+    target_media.save(target_path)
+
+    output_path = run_faceswap(source_path, target_path)
+    return output_path
+
+iface = gr.Interface(
+    fn=process_face_swap,
+    inputs=[
+        gr.Image(label="Source Face (face.png)"),
+        gr.File(label="Target Image or Video (.jpg/.mp4)")
+    ],
+    outputs=gr.File(label="Swapped Output"),
+    title="Lightweight Roop Face Swap",
+    description="Upload face.png and target image/video. Output will be result.jpg or result.mp4"
+)
+
+iface.launch()

requirements.txt

@@ -0,0 +1,6 @@
+
+gradio==3.50.2
+moviepy
+opencv-python-headless
+Pillow
+tqdm

run.py

@@ -0,0 +1,43 @@
+
+import os
+import shutil
+from moviepy.editor import VideoFileClip
+import cv2
+from PIL import Image
+from tqdm import tqdm
+
+def run_faceswap(source_path, target_path):
+    if target_path.endswith(".jpg") or target_path.endswith(".png"):
+        # Image face swap (mockup, actual logic should call Roop)
+        output_path = "result.jpg"
+        shutil.copy(target_path, output_path)
+        return output_path
+
+    elif target_path.endswith(".mp4"):
+        os.makedirs("frames_input", exist_ok=True)
+        os.makedirs("frames_output", exist_ok=True)
+
+        clip = VideoFileClip(target_path)
+        fps = clip.fps
+
+        for i, frame in enumerate(clip.iter_frames()):
+            Image.fromarray(frame).save(f"frames_input/frame_{i:05d}.jpg")
+
+        # Mock face swap on each frame (copy as is)
+        for frame_file in sorted(os.listdir("frames_input")):
+            shutil.copy(f"frames_input/{frame_file}", f"frames_output/{frame_file}")
+
+        frame_files = sorted(os.listdir("frames_output"))
+        frame_images = [cv2.imread(f"frames_output/{f}") for f in frame_files]
+        h, w, _ = frame_images[0].shape
+        out_path = "result.mp4"
+        video_writer = cv2.VideoWriter(out_path, cv2.VideoWriter_fourcc(*'mp4v'), fps, (w, h))
+
+        for img in frame_images:
+            video_writer.write(img)
+        video_writer.release()
+
+        return out_path
+
+    else:
+        raise ValueError("Unsupported file format")