Update app.py

app.py

@@ -1,4 +1,3 @@
-# Environment config at VERY TOP
 import os
 os.environ['MPLCONFIGDIR'] = '/tmp/matplotlib'
 os.environ['FONTCONFIG_PATH'] = '/tmp/fontconfig'
@@ -10,26 +9,24 @@ import logging
 from models import load_models
 from video_generator import generate_video_pipeline
 
-# Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 # Hardware configuration
-if not torch.cuda.is_available():
-    logger.warning("Using CPU-only mode")
-    torch.set_num_threads(4)
+device = "cuda" if torch.cuda.is_available() else "cpu"
+logger.info(f"Running on {device.upper()}")
 
 # Load models
 try:
     text_to_image, image_to_video, tts_model = load_models()
 except Exception as e:
-    logger.error(f"Model load failed: {str(e)}")
+    logger.error(f"Initialization failed: {str(e)}")
     raise
 
 def generate_video(prompt, duration=5, fps=24):
     with tempfile.TemporaryDirectory() as tmpdir:
         try:
-            video_path = generate_video_pipeline(
+            return generate_video_pipeline(
                 prompt=prompt,
                 text_to_image_model=text_to_image,
                 image_to_video_model=image_to_video,
@@ -38,36 +35,31 @@ def generate_video(prompt, duration=5, fps=24):
                 duration=duration,
                 fps=fps
             )
-            return video_path
         except Exception as e:
             logger.error(f"Generation failed: {str(e)}")
-            raise gr.Error(f"Video creation failed: {str(e)}")
+            raise gr.Error(f"Error: {str(e)}")
 
-# Gradio interface
 with gr.Blocks(title="AI Video Generator") as app:
     gr.Markdown("# 🎥 AI Video Generator")
     
     with gr.Row():
-        prompt_input = gr.Textbox(label="Input Prompt", 
-                                placeholder="A cat walking through a forest...")
+        prompt_input = gr.Textbox(label="Prompt", placeholder="A cat in space...")
     
     with gr.Row():
-        duration = gr.Slider(2, 60, value=5, label="Duration (seconds)")
-        fps = gr.Slider(12, 60, value=24, step=1, label="FPS")
+        duration = gr.Slider(2, 30, 5, label="Duration (s)")
+        fps = gr.Slider(12, 60, 24, label="FPS")
     
-    generate_btn = gr.Button("Generate Video", variant="primary")
+    generate_btn = gr.Button("Generate", variant="primary")
     
     with gr.Row():
         video_output = gr.Video(label="Result", format="mp4")
-        download_btn = gr.File(label="Download Video", type="file", interactive=False)
-    
-    progress = gr.Progress()
+        download_btn = gr.File(label="Download", type="file")
     
     generate_btn.click(
-        fn=generate_video,
+        generate_video,
         inputs=[prompt_input, duration, fps],
-        outputs=[video_output, download_btn],
+        outputs=[video_output, download_btn]
     )
 
 if __name__ == "__main__":
-    app.launch(server_name="0.0.0.0", share=False)
+    app.launch(server_name="0.0.0.0")