Update app.py

app.py

@@ -37,36 +37,39 @@ args = parser.parse_args()
 
 # Initialize the Refacer class
 refacer = Refacer(force_cpu=args.force_cpu, colab_performance=args.colab_performance)
-
 num_faces = args.max_num_faces
 
-# Run function for refacing video
+# Function to process refacing
 def run(*vars):
-    video_path = vars[0]
+    video_path = vars[0]  # Original video path
+    if not video_path:
+        raise ValueError("Video file not found. Please upload a valid video.")
+    
     origins = vars[1:(num_faces+1)]
     destinations = vars[(num_faces+1):(num_faces*2)+1]
     thresholds = vars[(num_faces*2)+1:]
 
     faces = []
-    for k in range(0, num_faces):
-        if origins[k] is not None and destinations[k] is not None:
+    for k in range(num_faces):
+        if origins[k] and destinations[k]:
             faces.append({
                 'origin': origins[k],
                 'destination': destinations[k],
                 'threshold': thresholds[k]
             })
 
-    # Call refacer to process video and get file path
-    refaced_video_path = refacer.reface(video_path, faces)
-
-    # Convert the refaced video to bytes
-    with open(refaced_video_path, "rb") as f:
-        video_bytes = f.read()
-
-    # Delete the temporary video file
-    os.remove(refaced_video_path)
+    try:
+        # Call refacer to process video and return bytes
+        refaced_video_path = refacer.reface(video_path, faces)
+        with open(refaced_video_path, "rb") as f:
+            video_bytes = f.read()
 
-    return video_bytes  # Return the video bytes for Gradio output
+        # Cleanup temporary file
+        os.remove(refaced_video_path)
+        return video_bytes
+    except Exception as e:
+        print(f"Error during refacing: {e}")
+        raise RuntimeError("Video refacing failed. Check logs.")
 
 # Prepare Gradio components
 origin = []
@@ -77,22 +80,22 @@ with gr.Blocks() as demo:
     with gr.Row():
         gr.Markdown("# Refacer")
     with gr.Row():
-        video = gr.Video(label="Original video", format="mp4")
-        video2 = gr.Video(label="Refaced video", interactive=False, format="mp4")  # Gradio UI only
+        video = gr.Video(label="Original Video", format="mp4")
+        video2 = gr.Video(label="Refaced Video", interactive=False, format="mp4")
 
-    for i in range(0, num_faces):
+    for i in range(num_faces):
         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"))
+                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")
 
-    # Connect the button to the run function
+    # Configure button click
     button.click(fn=run, inputs=[video] + origin + destination + thresholds, outputs=[video2])
 
-# Launch the Gradio app
+# Launch Gradio app
 demo.queue().launch(show_error=True, share=args.share_gradio, server_name="0.0.0.0", server_port=args.server_port)