Update app.py

app.py

@@ -3,7 +3,8 @@ import gradio as gr
 from refacer import Refacer
 import os
 import requests
-import tempfile  # Importing tempfile module
+import tempfile
+import subprocess
 
 # Hugging Face URL to download the model
 model_url = "https://huggingface.co/ofter/4x-UltraSharp/resolve/main/inswapper_128.onnx"
@@ -48,18 +49,25 @@ def run(video_path, *vars):
     refaced_video_path = refacer.reface(video_path, faces)
     print(f"Refaced video can be found at {refaced_video_path}")
     
-    # Open the refaced video file and write to a temporary file
-    with open(refaced_video_path, "rb") as f:
-        # Use tempfile to store the refaced video in memory
-        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
-        temp_file.write(f.read())
-        temp_file.close()  # Close the temporary file
+    # Use tempfile to create a temporary video file
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as temp_file:
+        temp_file_path = temp_file.name
+        
+        # Run ffmpeg to process the video and write the result to the temporary file
+        ffmpeg_command = [
+            "ffmpeg", "-i", refaced_video_path, "-c:v", "libx264", 
+            "-crf", "23", "-preset", "fast", temp_file_path
+        ]
+        
+        try:
+            subprocess.run(ffmpeg_command, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        except subprocess.CalledProcessError as e:
+            print(f"ffmpeg error: {e.stderr.decode()}")  # Print stderr output for debugging
 
-        # Reopen the temp file to use in the Gradio UI
-        with open(temp_file.name, "rb") as temp_f:
-            video_buffer = io.BytesIO(temp_f.read())
+        # Read the temporary file and convert it to a BytesIO buffer
+        with open(temp_file_path, "rb") as f:
+            video_buffer = io.BytesIO(f.read())
         
-        # Return the video buffer for Gradio to display in the UI
         return video_buffer  # Gradio will handle the video display
 
 # Prepare Gradio components