Update app.py

app.py

@@ -2,97 +2,113 @@ import gradio as gr
 import os
 import subprocess
 import time
-import logging
 import traceback
+import sys
 
 def process_audio(audio_file):
-    # Configure detailed logging
-    logging.basicConfig(level=logging.DEBUG, 
-                        format='%(asctime)s - %(levelname)s - %(message)s')
-    logger = logging.getLogger(__name__)
-
     try:
-        # Detailed logging for input
-        logger.info(f"Received audio file: {audio_file}")
-        logger.info(f"Audio file exists: {os.path.exists(audio_file)}")
-        
+        # Ensure input is a valid filepath for Gradio 3.x
+        if isinstance(audio_file, dict):
+            audio_file = audio_file['name']
+
+        # Comprehensive error logging
+        print(f"Processing audio file: {audio_file}")
+        print(f"File exists: {os.path.exists(audio_file)}")
+        print(f"Current working directory: {os.getcwd()}")
+        print(f"Python path: {sys.path}")
+
         # Validate input file
         if not audio_file or not os.path.exists(audio_file):
-            raise ValueError(f"Invalid or non-existent audio file: {audio_file}")
+            error_msg = f"Invalid or non-existent audio file: {audio_file}"
+            print(error_msg)
+            return None, error_msg
 
-        # Ensure output directory exists
+        # Ensure output directories exist
         os.makedirs("visualise/video/body-pixel2", exist_ok=True)
-        
-        # Debugging: print current working directory and file details
-        logger.debug(f"Current working directory: {os.getcwd()}")
-        logger.debug(f"Audio file path: {os.path.abspath(audio_file)}")
-        logger.debug(f"Audio file size: {os.path.getsize(audio_file)} bytes")
 
-        # Construct command with full paths
+        # Detailed command execution
         cmd = [
             "python", 
-            os.path.abspath("scripts/demo.py"),
-            "--config_file", os.path.abspath("config/body_pixel.json"),
+            "scripts/demo.py",
+            "--config_file", "config/body_pixel.json",
             "--infer",
-            "--audio_file", os.path.abspath(audio_file),
+            "--audio_file", audio_file,
             "--id", "0",
             "--whole_body"
         ]
         
-        logger.info(f"Executing command: {' '.join(cmd)}")
-        
-        # Run with more detailed error capture
-        result = subprocess.run(
-            cmd, 
-            stdout=subprocess.PIPE, 
-            stderr=subprocess.PIPE,
-            text=True,
-            cwd=os.getcwd(),  # Ensure correct working directory
-            timeout=1800
-        )
-        
-        # Log full command output
-        logger.info(f"Command STDOUT: {result.stdout}")
-        logger.error(f"Command STDERR: {result.stderr}")
-        
+        print(f"Executing command: {' '.join(cmd)}")
+
+        # Run with error capture
+        try:
+            result = subprocess.run(
+                cmd, 
+                stdout=subprocess.PIPE, 
+                stderr=subprocess.PIPE,
+                text=True,
+                timeout=600  # 10-minute timeout
+            )
+            
+            # Log command output
+            print("Command STDOUT:", result.stdout)
+            print("Command STDERR:", result.stderr)
+
+            # Check for successful execution
+            if result.returncode != 0:
+                error_msg = f"Command failed. STDERR: {result.stderr}"
+                print(error_msg)
+                return None, error_msg
+
+        except subprocess.TimeoutExpired:
+            error_msg = "Inference process timed out"
+            print(error_msg)
+            return None, error_msg
+        except Exception as e:
+            error_msg = f"Subprocess error: {str(e)}"
+            print(error_msg)
+            print(traceback.format_exc())
+            return None, error_msg
+
         # Determine output video path
         audio_name = os.path.splitext(os.path.basename(audio_file))[0]
         output_dir = f"visualise/video/body-pixel2/{audio_name}"
         output_path = f"{output_dir}/1st-page.mp4"
         
-        logger.info(f"Expected output path: {output_path}")
-        
+        print(f"Expected output path: {output_path}")
+        print(f"Output directory contents: {os.listdir(output_dir) if os.path.exists(output_dir) else 'Directory not found'}")
+
         # Check output video
         if os.path.exists(output_path):
-            logger.info(f"Output video found: {output_path}")
+            print(f"Output video found: {output_path}")
             return output_path
         else:
-            logger.error("Output video not generated")
-            return None, f"Error: Output video not generated. STDERR: {result.stderr}"
-    
-    except subprocess.TimeoutExpired:
-        logger.error("Inference process timed out")
-        return None, "Error: Inference process took too long"
-    
+            error_msg = f"Output video not generated. Check script execution and paths."
+            print(error_msg)
+            return None, error_msg
+
     except Exception as e:
-        logger.error(f"Unexpected error: {str(e)}")
-        logger.error(traceback.format_exc())
-        return None, f"Unexpected error: {str(e)}"
+        error_msg = f"Unexpected error: {str(e)}"
+        print(error_msg)
+        print(traceback.format_exc())
+        return None, error_msg
 
-# Gradio Interface for 3.x compatibility
+# Gradio Interface
 demo = gr.Interface(
     fn=process_audio,
     inputs=gr.inputs.File(type="file", label="Upload Audio File"),
-    outputs=gr.outputs.Video(label="Generated Motion Video"),
+    outputs=[
+        gr.outputs.Video(label="Generated Motion Video"),
+        gr.outputs.Textbox(label="Error Messages")
+    ],
     title="TalkSHOW: Speech-to-Motion Translation System",
     description="Convert speech audio to realistic 3D human motion using the SMPL-X model.",
     examples=[["demo_audio/1st-page.wav"]]
 )
 
-# Launch with comprehensive logging
+# Launch the app
 if __name__ == "__main__":
     demo.launch(
         server_name="0.0.0.0", 
         server_port=7860,
-        debug=True  # Enable Gradio debug mode
+        debug=True
     )