Update app.py

app.py

@@ -31,13 +31,16 @@ refacer = Refacer(force_cpu=True)
 
 # Dummy function to simulate frame-level processing
 def process_frame(frame, origin_face, destination_face, threshold):
-    # Simulate face swapping or any processing needed
-    result_frame = refacer.reface(frame, [{
-        'origin': origin_face,
-        'destination': destination_face,
-        'threshold': threshold
-    }])
-    return result_frame
+    try:
+        result_frame = refacer.reface(frame, [{
+            'origin': origin_face,
+            'destination': destination_face,
+            'threshold': threshold
+        }])
+        return result_frame
+    except Exception as e:
+        print(f"Error in processing frame: {e}")
+        return frame
 
 # Function to process the video in parallel using multiprocessing
 def process_video(video_path, origins, destinations, thresholds, max_processes=2):
@@ -63,7 +66,8 @@ def process_video(video_path, origins, destinations, thresholds, max_processes=2
     # Saving the processed frames back into a video
     output_video_path = "processed_video.mp4"
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # Compression using mp4 codec
-    out = cv2.VideoWriter(output_video_path, fourcc, 30.0, (640, 360))  # Reduce resolution to speed up processing
+    height, width, _ = frames[0].shape
+    out = cv2.VideoWriter(output_video_path, fourcc, 30.0, (width, height))
 
     for frame in processed_frames:
         out.write(frame)
@@ -73,13 +77,11 @@ def process_video(video_path, origins, destinations, thresholds, max_processes=2
 
 # Gradio Interface function
 def run(video_path, *vars):
-    # Split the inputs into origins, destinations, and thresholds based on num_faces
-    num_faces = 5  # You can adjust this based on your UI
+    num_faces = 5
     origins = vars[:num_faces]
     destinations = vars[num_faces:2*num_faces]
     thresholds = vars[2*num_faces:]
 
-    # Ensure there are no index errors by limiting the number of inputs
     if len(origins) != num_faces or len(destinations) != num_faces or len(thresholds) != num_faces:
         return "Please provide input for all faces."
 
@@ -100,14 +102,14 @@ with gr.Blocks() as demo:
         video_input = gr.Video(label="Original video", format="mp4")
         video_output = gr.Video(label="Refaced video", interactive=False, format="mp4")
 
-    for i in range(5):  # Set max faces to 5
+    for i in range(5):
         with gr.Tab(f"Face #{i+1}"):
             with gr.Row():
                 origin.append(gr.Image(label="Face to replace"))
                 destination.append(gr.Image(label="Destination face"))
             with gr.Row():
                 thresholds.append(gr.Slider(label="Threshold", minimum=0.0, maximum=1.0, value=0.2))
-    
+
     with gr.Row():
         button = gr.Button("Reface", variant="primary")