Update app.py

app.py

@@ -3,9 +3,9 @@ import subprocess
 import os
 
 def run_inference(audio_file):
-    # Define the output video filename
-    output_video = "output_video.mp4"  # Change this to your desired output filename
-
+    # Extract the audio filename to construct the video filename later
+    audio_filename = os.path.basename(audio_file)
+    
     # Define the command to run your script
     command = [
         "python", "scripts/demo.py",
@@ -14,7 +14,7 @@ def run_inference(audio_file):
         "--audio_file", audio_file,
         "--body_model_name", "s2g_LS3DCG",
         "--body_model_path", "experiments/2022-10-19-smplx_S2G-LS3DCG/ckpt-99.pth",
-        "--id", "0",
+        "--id", "0"
     ]
     
     # Run the command and capture the output
@@ -24,8 +24,16 @@ def run_inference(audio_file):
     if result.returncode != 0:
         return f"Error: {result.stderr}"
     
-    # Return the generated video file path
-    return output_video
+    # Construct the output video filename based on your logic
+    # Assuming config.Log.name is a fixed string or you may need to adjust this accordingly
+    config_log_name = "YourLogName"  # Replace with the actual log name used in your config
+    output_video_path = f'visualise/video/{config_log_name}/{audio_filename.rsplit(".", 1)[0]}.mp4'
+    
+    # Check if the video file exists and return the path
+    if os.path.exists(output_video_path):
+        return output_video_path
+    else:
+        return "Video file not found."
 
 # Create Gradio interface
 interface = gr.Interface(