Update app.py

app.py

@@ -37,39 +37,30 @@ 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
 
-# Function to process refacing
+# Run function for refacing video
 def run(*vars):
-    video_path = vars[0]  # Original video path
-    if not video_path:
-        raise ValueError("Video file not found. Please upload a valid video.")
-    
+    video_path = vars[0]
     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(num_faces):
-        if origins[k] and destinations[k]:
+    for k in range(0, num_faces):
+        if origins[k] is not None and destinations[k] is not None:
             faces.append({
                 'origin': origins[k],
                 'destination': destinations[k],
                 'threshold': thresholds[k]
             })
 
-    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()
+    # Call refacer to process video and get file path
+    refaced_video_path = refacer.reface(video_path, faces)  # refaced video path
+    print(f"Refaced video can be found at {refaced_video_path}")
 
-        # 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.")
+    return refaced_video_path  # Return the file path to show in Gradio output
 
 # Prepare Gradio components
 origin = []
@@ -80,22 +71,21 @@ 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")
+        video = gr.Video(label="Original video", format="mp4")
+        video2 = gr.Video(label="Refaced video", interactive=False, format="mp4")
 
-    for i in range(num_faces):
+    for i in range(0, 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")
 
-    # Configure button click
     button.click(fn=run, inputs=[video] + origin + destination + thresholds, outputs=[video2])
 
-# Launch Gradio app
+# Launch the Gradio app
 demo.queue().launch(show_error=True, share=args.share_gradio, server_name="0.0.0.0", server_port=args.server_port)