Update app.py

app.py

@@ -1,41 +1,50 @@
 import gradio as gr
 import subprocess
+import os
 from pydub import AudioSegment
+import tempfile
 
-# Path to the TalkShow demo script and configuration
+# Paths to TalkShow demo script and configuration
 DEMO_SCRIPT_PATH = "scripts/demo.py"
 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"
-STYLE_WAV_PATH = "./demo_audio/style.wav"
 
 def convert_audio_to_wav(input_audio):
+    # Use a temporary file to save the .wav output
+    temp_wav = tempfile.NamedTemporaryFile(delete=False, suffix=".wav")
+    temp_wav_path = temp_wav.name
+
     # Convert the input audio to .wav format
     audio = AudioSegment.from_file(input_audio)
-    audio.export(STYLE_WAV_PATH, format="wav")
-    return STYLE_WAV_PATH
+    audio.export(temp_wav_path, format="wav")
+    return temp_wav_path
 
 def run_demo_and_get_video(input_audio):
-    # Convert the input audio to .wav
+    # Convert the input audio to a temporary .wav file
     wav_path = convert_audio_to_wav(input_audio)
 
-    # Run the demo script with the specified arguments
-    command = [
-        "python", DEMO_SCRIPT_PATH,
-        "--config_file", CONFIG_FILE,
-        "--infer",
-        "--audio_file", wav_path,
-        "--body_model_name", "s2g_LS3DCG",
-        "--body_model_path", BODY_MODEL_PATH,
-        "--id", "0"
-    ]
-    subprocess.run(command, check=True)
-
-    # Check if the output file exists
-    if os.path.exists(OUTPUT_MP4_PATH):
-        return OUTPUT_MP4_PATH
-    else:
-        return "Error: Output video not generated."
+    try:
+        # Run the demo script with the specified arguments
+        command = [
+            "python", DEMO_SCRIPT_PATH,
+            "--config_file", CONFIG_FILE,
+            "--infer",
+            "--audio_file", wav_path,
+            "--body_model_name", "s2g_LS3DCG",
+            "--body_model_path", BODY_MODEL_PATH,
+            "--id", "0"
+        ]
+        subprocess.run(command, check=True)
+
+        # Check if the output file exists
+        if os.path.exists(OUTPUT_MP4_PATH):
+            return OUTPUT_MP4_PATH
+        else:
+            return "Error: Output video not generated."
+    finally:
+        # Clean up the temporary .wav file after processing
+        os.remove(wav_path)
 
 # Define the Gradio interface
 interface = gr.Interface(
@@ -46,4 +55,4 @@ interface = gr.Interface(
 
 # Launch the app
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()