Update run.py

run.py

@@ -1,10 +1,12 @@
 import os
+import time
 from moviepy.editor import VideoFileClip
 from PIL import Image
 import insightface
 import cv2
 import numpy as np
 
+# Load model once
 model = insightface.app.FaceAnalysis(name="buffalo_l", providers=["CPUExecutionProvider"])
 model.prepare(ctx_id=0)
 
@@ -25,6 +27,7 @@ def swap_face(source_path, target_path, output_path):
                    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):
@@ -36,10 +39,18 @@ def swap_face(source_path, target_path, output_path):
                           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
+
+    # ✅ 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.")