Update app.py

app.py

@@ -1,21 +1,31 @@
 import os
+# Must be at VERY TOP before any imports
+os.environ['MPLCONFIGDIR'] = '/tmp/matplotlib'
+os.environ['FONTCONFIG_PATH'] = '/tmp/fontconfig'
+
+import torch  # Import torch immediately after environment vars
 import gradio as gr
 import tempfile
+import logging
 from models import load_models
 from video_generator import generate_video_pipeline
 
-# Configure environment variables first
-os.environ['MPLCONFIGDIR'] = '/tmp/matplotlib'
-os.environ['FONTCONFIG_PATH'] = '/tmp/fontconfig'
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
-# Force CPU mode if no GPU available
+# Early hardware detection
 if not torch.cuda.is_available():
+    logger.warning("CUDA not available - Using CPU with reduced performance")
     torch.set_num_threads(4)
     os.environ['CUDA_VISIBLE_DEVICES'] = ''
-    logger.warning("Running in CPU-only mode - expect slower generation")
 
-# Load models at startup
-text_to_image, image_to_video, tts_model = load_models()
+# Load models after hardware config
+try:
+    text_to_image, image_to_video, tts_model = load_models()
+except Exception as e:
+    logger.error(f"Model loading failed: {str(e)}")
+    raise
 
 def generate_video(prompt, duration=5, fps=24):
     with tempfile.TemporaryDirectory() as tmpdir:
@@ -31,7 +41,10 @@ def generate_video(prompt, duration=5, fps=24):
             )
             return video_path
         except Exception as e:
-            raise gr.Error(f"Video generation failed: {str(e)}")
+            logger.error(f"Generation error: {str(e)}")
+            raise gr.Error("Video generation failed - check logs")
+
+# Rest of Gradio interface remains unchanged
 
 with gr.Blocks(title="AI Video Generator") as app:
     gr.Markdown("# 🎥 AI Video Generator")