Update app.py

app.py

@@ -1,15 +1,26 @@
 import gradio as gr
 import subprocess
 import os
+import json
+
+def get_log_name(config_file):
+    """Load the JSON configuration file to extract the log name."""
+    with open(config_file, 'r') as f:
+        config = json.load(f)
+    return config['Log']['name']  # Adjust this if the structure is different
 
 def run_inference(audio_file):
-    # Extract the audio filename to construct the video filename later
+    # Extract the audio filename
     audio_filename = os.path.basename(audio_file)
-    
+
+    # Load the configuration to get the log name
+    config_file = './config/LS3DCG.json'
+    config_log_name = get_log_name(config_file)
+
     # Define the command to run your script
     command = [
         "python", "scripts/demo.py",
-        "--config_file", "./config/LS3DCG.json",
+        "--config_file", config_file,
         "--infer",
         "--audio_file", audio_file,
         "--body_model_name", "s2g_LS3DCG",
@@ -23,10 +34,8 @@ def run_inference(audio_file):
     # Check for errors
     if result.returncode != 0:
         return f"Error: {result.stderr}"
-    
-    # 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
+
+    # Construct the output video filename based on the log name
     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