Update app.py

app.py

@@ -6,7 +6,7 @@ from PIL import Image
 import tempfile
 import subprocess
 import sys
-from huggingface_hub import snapshot_download, hf_hub_download
+from huggingface_hub import snapshot_download
 import shutil
 
 # Configuration
@@ -63,20 +63,17 @@ def download_and_setup_model():
         print(f"❌ Setup failed: {e}")
         return False
 
-@spaces.GPU(duration=120)  # Allocate GPU for 2 minutes max
-def generate_video(input_image, num_frames, seed, progress=gr.Progress()):
+@spaces.GPU(duration=120)
+def generate_video(input_image, num_frames, seed):
     """Generate video using Matrix-Game-2.0"""
     
     if input_image is None:
         return None, "❌ Please upload an input image first"
     
     # Setup model if not already done
-    progress(0.1, desc="🔧 Setting up model...")
     if not download_and_setup_model():
         return None, "❌ Failed to setup model"
     
-    progress(0.2, desc="📷 Processing input image...")
-    
     try:
         # Create temporary directories
         temp_dir = tempfile.mkdtemp(prefix="matrix_gen_")
@@ -84,7 +81,7 @@ def generate_video(input_image, num_frames, seed, progress=gr.Progress()):
         os.makedirs(output_dir, exist_ok=True)
         
         # Prepare input image
-        if max(input_image.size) > 512:  # Resize for faster processing
+        if max(input_image.size) > 512:
             ratio = 512 / max(input_image.size)
             new_size = (int(input_image.size[0] * ratio), int(input_image.size[1] * ratio))
             input_image = input_image.resize(new_size, Image.Resampling.LANCZOS)
@@ -92,18 +89,16 @@ def generate_video(input_image, num_frames, seed, progress=gr.Progress()):
         input_path = os.path.join(temp_dir, "input.jpg")
         input_image.save(input_path, "JPEG", quality=95)
         
-        progress(0.4, desc="🚀 Generating video...")
-        
         # Find the inference script and config
         matrix_dir = os.path.join("Matrix-Game", "Matrix-Game-2")
         
-        # Basic inference command (simplified)
+        # Basic inference command
         cmd = [
             sys.executable, 
             os.path.join(matrix_dir, "inference.py"),
             "--img_path", input_path,
             "--output_folder", output_dir,
-            "--num_output_frames", str(min(num_frames, 100)),  # Limit frames for HF Spaces
+            "--num_output_frames", str(min(num_frames, 100)),
             "--seed", str(seed)
         ]
         
@@ -120,19 +115,15 @@ def generate_video(input_image, num_frames, seed, progress=gr.Progress()):
         if model_path:
             cmd.extend(["--pretrained_model_path", model_path])
         
-        progress(0.6, desc="🎬 Running inference...")
-        
         # Execute with timeout
         process = subprocess.run(
             cmd, 
             capture_output=True, 
             text=True, 
-            timeout=300,  # 5 minute timeout
+            timeout=300,
             cwd=matrix_dir
         )
         
-        progress(0.9, desc="📹 Finalizing video...")
-        
         # Find output video
         video_files = []
         for root, dirs, files in os.walk(output_dir):
@@ -145,22 +136,11 @@ def generate_video(input_image, num_frames, seed, progress=gr.Progress()):
             final_output = f"output_{seed}.mp4"
             shutil.copy(video_files[0], final_output)
             
-            log = f"""
-✅ **Generation Successful!**
-📊 Input: {input_image.size}
-🎬 Frames: {num_frames}
-🎲 Seed: {seed}
-📁 Output: {final_output}
-            """
+            log = f"✅ Generation Successful!\n📊 Input: {input_image.size}\n🎬 Frames: {num_frames}\n🎲 Seed: {seed}\n📁 Output: {final_output}"
             
-            progress(1.0, desc="✅ Complete!")
             return final_output, log
         else:
-            error_log = f"""
-❌ **Generation Failed**
-📝 Error output: {process.stderr[:500] if process.stderr else 'No error details'}
-💭 Try adjusting parameters or using a different input image
-            """
+            error_log = f"❌ Generation Failed\n📝 Error output: {process.stderr[:500] if process.stderr else 'No error details'}\n💭 Try adjusting parameters or using a different input image"
             return None, error_log
             
     except subprocess.TimeoutExpired:
@@ -172,7 +152,7 @@ def generate_video(input_image, num_frames, seed, progress=gr.Progress()):
         if 'temp_dir' in locals() and os.path.exists(temp_dir):
             shutil.rmtree(temp_dir, ignore_errors=True)
 
-# Gradio Interface
+# Simple Gradio Interface
 def create_interface():
     with gr.Blocks(title="Matrix-Game-2.0") as interface:
         
@@ -185,49 +165,21 @@ def create_interface():
         """)
         
         with gr.Row():
-            with gr.Column(scale=1):
+            with gr.Column():
                 gr.Markdown("### 📸 Input")
-                input_image = gr.Image(
-                    label="Input Image", 
-                    type="pil"
-                )
+                input_image = gr.Image(label="Input Image", type="pil")
                 
                 gr.Markdown("### ⚙️ Settings")
                 with gr.Row():
-                    num_frames = gr.Slider(
-                        minimum=25, 
-                        maximum=100, 
-                        value=50, 
-                        step=25, 
-                        label="Number of Frames"
-                    )
-                    seed = gr.Number(
-                        value=42, 
-                        label="Seed", 
-                        precision=0
-                    )
+                    num_frames = gr.Slider(25, 100, 50, step=25, label="Number of Frames")
+                    seed = gr.Number(value=42, label="Seed", precision=0)
                 
-                generate_btn = gr.Button(
-                    "🚀 Generate Video", 
-                    variant="primary"
-                )
+                generate_btn = gr.Button("🚀 Generate Video", variant="primary")
                 
-                gr.Markdown("""
-                ### 💡 Tips
-                - Use clear, well-lit images
-                - Landscapes and scenes work best
-                - Lower frame counts = faster generation
-                - Try different seeds for variety
-                """)
-            
-            with gr.Column(scale=1):
+            with gr.Column():
                 gr.Markdown("### 🎬 Generated Video")
                 output_video = gr.Video(label="Result")
-                
-                status_log = gr.Textbox(
-                    label="Status Log", 
-                    lines=8
-                )
+                status_log = gr.Textbox(label="Status Log", lines=8)
         
         # Event handlers
         generate_btn.click(
@@ -241,4 +193,4 @@ def create_interface():
 # Launch the app
 if __name__ == "__main__":
     demo = create_interface()
-    demo.launch()
+    demo.launch(share=True)