Update app.py

app.py

@@ -3,6 +3,8 @@ import subprocess
 import os
 from pydub import AudioSegment
 import tempfile
+
+# Set OpenGL platform
 os.environ["PYOPENGL_PLATFORM"] = "osmesa"
 
 # Paths to TalkShow demo script and configuration
@@ -11,6 +13,9 @@ CONFIG_FILE = "./config/LS3DCG.json"
 BODY_MODEL_PATH = "experiments/2022-10-19-smplx_S2G-LS3DCG/ckpt-99.pth"
 OUTPUT_MP4_PATH = "./output/demo_output.mp4"
 
+# Ensure the output directory exists
+os.makedirs(os.path.dirname(OUTPUT_MP4_PATH), exist_ok=True)
+
 def convert_audio_to_wav(input_audio):
     # Use a temporary file to save the .wav output
     temp_wav = tempfile.NamedTemporaryFile(delete=False, suffix=".wav")
@@ -43,6 +48,8 @@ def run_demo_and_get_video(input_audio):
             return OUTPUT_MP4_PATH
         else:
             return "Error: Output video not generated."
+    except subprocess.CalledProcessError as e:
+        return f"Error: {str(e)}"
     finally:
         # Clean up the temporary .wav file after processing
         os.remove(wav_path)
@@ -51,7 +58,9 @@ def run_demo_and_get_video(input_audio):
 interface = gr.Interface(
     fn=run_demo_and_get_video,
     inputs=gr.Audio(source="upload", type="filepath"),
-    outputs=gr.Video()
+    outputs=gr.Video(type="file"),
+    title="TalkSHOW Demo",
+    description="Generate a video from audio input using TalkSHOW model."
 )
 
 # Launch the app