Improve error handling to show full error details and tracebacks in UI

app.py

@@ -18,6 +18,7 @@ import gradio as gr
 import torch
 import subprocess
 import pathlib
+import traceback
 from pathlib import Path
 
 # Suppress harmless warnings
@@ -249,9 +250,17 @@ def _generate_image_impl(prompt, aspect_ratio, cfg, seed, checkpoint_file, confi
             status_msg += f"\n📋 Full logs available at: {log_file}\n"
         
         if result.returncode != 0:
-            error_msg = f"Error during generation:\n{result.stderr}\n\nStdout:\n{result.stdout}"
+            error_msg = f"❌ Error during generation (return code: {result.returncode})\n\n"
+            error_msg += f"=== STDERR ===\n{result.stderr}\n\n"
+            error_msg += f"=== STDOUT ===\n{result.stdout}\n\n"
             if log_content:
-                error_msg += f"\n\n📋 Full log file ({log_file}):\n{log_content[-2000:]}"  # Last 2000 chars
+                # Show last 3000 chars of log for more context
+                error_msg += f"📋 Last 3000 characters from log file ({log_file}):\n"
+                error_msg += f"{'='*60}\n"
+                error_msg += log_content[-3000:]
+                error_msg += f"\n{'='*60}\n"
+            else:
+                error_msg += f"⚠️  Log file not found at: {log_file}\n"
             return None, error_msg
         
         status_msg += "Generation complete. Looking for output...\n"
@@ -304,7 +313,13 @@ def _generate_image_impl(prompt, aspect_ratio, cfg, seed, checkpoint_file, confi
             return None, error_msg
             
     except Exception as e:
-        return None, f"Error: {str(e)}"
+        # Get full traceback for debugging
+        error_traceback = traceback.format_exc()
+        error_msg = f"❌ Exception occurred during image generation:\n\n"
+        error_msg += f"Error Type: {type(e).__name__}\n"
+        error_msg += f"Error Message: {str(e)}\n\n"
+        error_msg += f"=== FULL TRACEBACK ===\n{error_traceback}\n"
+        return None, error_msg
 
 
 # Apply @spaces.GPU decorator if available (required for ZeroGPU)
@@ -384,7 +399,10 @@ def _generate_video_impl(prompt, aspect_ratio, cfg, seed, target_length, checkpo
         result = subprocess.run(cmd, capture_output=True, text=True, cwd=os.getcwd())
         
         if result.returncode != 0:
-            return None, f"Error: {result.stderr}"
+            error_msg = f"❌ Error during video generation (return code: {result.returncode})\n\n"
+            error_msg += f"=== STDERR ===\n{result.stderr}\n\n"
+            error_msg += f"=== STDOUT ===\n{result.stdout}\n"
+            return None, error_msg
         
         # Find the generated video
         output_files = list(output_dir.glob("**/*.mp4")) + list(output_dir.glob("**/*.gif"))
@@ -395,7 +413,13 @@ def _generate_video_impl(prompt, aspect_ratio, cfg, seed, target_length, checkpo
             return None, "Error: Generated video not found."
             
     except Exception as e:
-        return None, f"Error: {str(e)}"
+        # Get full traceback for debugging
+        error_traceback = traceback.format_exc()
+        error_msg = f"❌ Exception occurred during video generation:\n\n"
+        error_msg += f"Error Type: {type(e).__name__}\n"
+        error_msg += f"Error Message: {str(e)}\n\n"
+        error_msg += f"=== FULL TRACEBACK ===\n{error_traceback}\n"
+        return None, error_msg
 
 
 # Create Gradio interface