Update run.py

run.py

@@ -1,12 +1,9 @@
 import os
-import time
 from moviepy.editor import VideoFileClip
-from PIL import Image
 import insightface
 import cv2
-import numpy as np
 
-# Load model once
+# Load InsightFace model
 model = insightface.app.FaceAnalysis(name="buffalo_l", providers=["CPUExecutionProvider"])
 model.prepare(ctx_id=0)
 
@@ -22,35 +19,28 @@ def swap_face(source_path, target_path, output_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]]
+
+        # Crop and resize source face
+        sx1, sy1, sx2, sy2 = source_face.bbox.astype(int)
+        source_crop = source_img[sy1:sy2, sx1:sx2]
+
+        tx1, ty1, tx2, ty2 = faces[0].bbox.astype(int)
+        target_w, target_h = tx2 - tx1, ty2 - ty1
+        source_resized = cv2.resize(source_crop, (target_w, target_h))
+
+        # Paste on target
+        target_img[ty1:ty2, tx1:tx2] = source_resized
         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)
-
-    # ✅ Safety check – wait for output file to be created
-    for _ in range(5):
-        if os.path.exists(output_path):
-            return output_path
-        time.sleep(1)
-
-    raise Exception("Output file not generated.")
+                tx1, ty1, tx2, ty2 = faces[0].bbox.astype(int)
+                target_w, target_h = tx2 - tx1, ty2 - ty1
+
+                sx1, sy1, sx2, sy2 = source_face.bbox.astype(int)
+                source_crop = source_img