Upload 3 files

app.py

@@ -0,0 +1,28 @@
+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==4.25.0
+insightface==0.7.3
+onnxruntime==1.16.3
+moviepy==1.0.3
+numpy==1.23.5
+opencv-python==4.10.0.82

run.py

@@ -0,0 +1,42 @@
+import os
+from moviepy.editor import VideoFileClip
+import cv2
+import numpy as np
+import insightface
+
+model = insightface.app.FaceAnalysis(name="buffalo_l", providers=["CPUExecutionProvider"])
+model.prepare(ctx_id=0)
+
+def swap_face(source_path, target_path, output_path):
+    source_img = cv2.imread(source_path)
+    faces = model.get(source_img)
+    if len(faces) == 0:
+        raise Exception("No face found in source.")
+    source_face = faces[0]
+
+    if target_path.endswith(".jpg"):
+        target_img = cv2.imread(target_path)
+        faces = model.get(target_img)
+        if len(faces) == 0:
+            raise Exception("No face found in target.")
+        target_img[faces[0].bbox.astype(int)[1]:faces[0].bbox.astype(int)[3],
+                   faces[0].bbox.astype(int)[0]:faces[0].bbox.astype(int)[2]] =                    source_img[source_face.bbox.astype(int)[1]:source_face.bbox.astype(int)[3],
+                              source_face.bbox.astype(int)[0]:source_face.bbox.astype(int)[2]]
+        cv2.imwrite(output_path, target_img)
+    else:
+        clip = VideoFileClip(target_path)
+        def process_frame(frame):
+            frame_bgr = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
+            faces = model.get(frame_bgr)
+            if len(faces) > 0:
+                frame_bgr[faces[0].bbox.astype(int)[1]:faces[0].bbox.astype(int)[3],
+                          faces[0].bbox.astype(int)[0]:faces[0].bbox.astype(int)[2]] =                           source_img[source_face.bbox.astype(int)[1]:source_face.bbox.astype(int)[3],
+                                     source_face.bbox.astype(int)[0]:source_face.bbox.astype(int)[2]]
+            return cv2.cvtColor(frame_bgr, cv2.COLOR_BGR2RGB)
+        new_clip = clip.fl_image(process_frame)
+        new_clip.write_videofile(output_path, audio=True)
+
+def run_faceswap(source_path, target_path):
+    output_path = "result.jpg" if target_path.endswith(".jpg") else "result.mp4"
+    swap_face(source_path, target_path, output_path)
+    return output_path