Update app.py

app.py

@@ -57,10 +57,16 @@ def run(*vars):
             })
 
     # 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}")
+    refaced_video_path = refacer.reface(video_path, faces)
 
-    return refaced_video_path  # Return the file path to show in Gradio output
+    # 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)
+
+    return video_bytes  # Return the video bytes for Gradio output
 
 # Prepare Gradio components
 origin = []
@@ -72,7 +78,7 @@ with gr.Blocks() as demo:
         gr.Markdown("# Refacer")
     with gr.Row():
         video = gr.Video(label="Original video", format="mp4")
-        video2 = gr.Video(label="Refaced video", interactive=False, format="mp4")
+        video2 = gr.Video(label="Refaced video", interactive=False, format="mp4")  # Gradio UI only
 
     for i in range(0, num_faces):
         with gr.Tab(f"Face #{i+1}"):
@@ -85,6 +91,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         button = gr.Button("Reface", variant="primary")
 
+    # Connect the button to the run function
     button.click(fn=run, inputs=[video] + origin + destination + thresholds, outputs=[video2])
 
 # Launch the Gradio app