Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import subprocess
 import os
 import gdown
+import glob
 
 # Ensure 'checkpoint' directory exists
 os.makedirs("checkpoint", exist_ok=True)
@@ -61,9 +62,16 @@ def run_xportrait(
     # Run the command
     try:
         subprocess.run(command, check=True)
-        return f"Output saved in: {output_dir}"
+        
+        # Find the generated video file in the output directory
+        video_files = glob.glob(os.path.join(output_dir, "*.mp4")) + glob.glob(os.path.join(output_dir, "*.avi"))
+        print(video_files)
+        if video_files:
+            return f"Output video saved at: {video_files[0]}", video_files[0]
+        else:
+            return "No video file was found in the output directory.", None
     except subprocess.CalledProcessError as e:
-        return f"An error occurred: {e}"
+        return f"An error occurred: {e}", None
 
 # Set up Gradio interface
 app = gr.Interface(
@@ -81,7 +89,7 @@ app = gr.Interface(
         gr.Number(value=4, label="Number of Mix"),
         gr.Number(value=30, label="DDIM Steps")
     ],
-    outputs="text",
+    outputs=["text", "video"],
     title="XPortrait Model Runner",
     description="Run XPortrait with customizable parameters."
 )