Update simple_app.py

simple_app.py

@@ -27,7 +27,7 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
     # Run the model inference in a subprocess
     process = subprocess.Popen(command, 
                                stdout=subprocess.PIPE, 
-                               stderr=subprocess.PIPE, 
+                               stderr=subprocess.PIPE,  # Capture stderr for error messages
                                text=True, 
                                bufsize=1)
 
@@ -67,16 +67,22 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
         if process.poll() is not None:
             break
 
+    # Drain any remaining output from stderr
+    stderr_output = process.stderr.read().strip()
+    if stderr_output:
+        print(f"Error output:\n{stderr_output}")
+
     # Clean up and finalize the progress bar
     process.wait()
     if video_progress_bar:
         video_progress_bar.close()
 
-    # Return the video file path if successful
+    # Check if the process finished successfully
     if process.returncode == 0:
         return "generated_video.mp4"
     else:
-        raise Exception("Error executing command")
+        print(f"Process failed with return code {process.returncode}")
+        raise Exception(f"Error executing command: {stderr_output}")
 
 # Gradio UI
 with gr.Blocks() as demo: