Update app.py

app.py

@@ -19,6 +19,7 @@ def load_model_safely():
     print("Lade Modell...")
     
     try:
+        # Standard float32 für maximale CPU Kompatibilität
         pipe = I2VGenXLPipeline.from_pretrained(
             MODEL_ID, 
             torch_dtype=torch.float32, 
@@ -34,6 +35,7 @@ def load_model_safely():
         except Exception as e:
             log += f"⚠️ Turbo Fehler: {e}\n"
 
+        # Versuche aggressives Speichermanagement
         try:
             pipe.enable_model_cpu_offload()
             log += "✅ Model Offloading aktiv.\n"
@@ -52,8 +54,7 @@ def load_model_safely():
     except Exception as e:
         return None, f"Absturz beim Laden: {e}\n{traceback.format_exc()}"
 
-# HIER IST DIE MAGIE: Das Argument "progress=gr.Progress()"
-def generate_video(image_in, prompt, negative_prompt, progress=gr.Progress()):
+def generate_video(image_in, prompt, negative_prompt, resolution, progress=gr.Progress()):
     global pipe
     log_messages = ""
     
@@ -61,7 +62,7 @@ def generate_video(image_in, prompt, negative_prompt, progress=gr.Progress()):
         return None, "Kein Bild!"
     
     # Initialisierung des Balkens
-    progress(0, desc="Lade Modell (kann dauern)...")
+    progress(0, desc="Lade Modell...")
     
     if pipe is None:
         model, msg = load_model_safely()
@@ -74,23 +75,28 @@ def generate_video(image_in, prompt, negative_prompt, progress=gr.Progress()):
     gc.collect()
     
     try:
+        # EINSTELLUNGEN
         steps = 6         
         guidance = 1.2    
-        target_size = 448 
         
-        log_messages += f"Starte Generierung ({steps} Steps)...\n"
+        # Slider Wert übernehmen
+        target_size = int(resolution)
+        
+        log_messages += f"Skaliere Bild auf {target_size}x{target_size}...\n"
         
+        # Bild resizing (WICHTIG für Speed!)
         image_in = image_in.resize((target_size, target_size))
+        
+        log_messages += f"Starte Generierung ({steps} Steps)...\n"
+        
         generator = torch.manual_seed(42)
         
-        # --- DER FORTSCHRITTS-SPION ---
-        # Diese Funktion wird NACH JEDEM STEP aufgerufen
-        def callback_fn(pipe, step_index, timestep, callback_kwargs):
-            current = step_index + 1
-            # Aktualisiert den Balken oben im Bild
-            progress((current, steps), desc=f"Step {current} von {steps} fertig...")
-            return callback_kwargs
-        # ------------------------------
+        # --- FIX FÜR DEN FORTSCHRITTSBALKEN (Alte Methode) ---
+        def callback_fn(step, timestep, latents):
+            # step ist hier der Index (0, 1, 2...)
+            current = step + 1
+            progress((current, steps), desc=f"Step {current}/{steps} (Größe: {target_size}px)")
+        # -----------------------------------------------------
         
         output = pipe(
             prompt=prompt,
@@ -102,7 +108,8 @@ def generate_video(image_in, prompt, negative_prompt, progress=gr.Progress()):
             height=target_size, 
             width=target_size,  
             generator=generator,
-            callback_on_step_end=callback_fn  # Hier binden wir den Spion ein
+            callback=callback_fn, # Alte Methode nutzen!
+            callback_steps=1
         ).frames[0]
         
         video_path = "turbo_output.mp4"
@@ -112,23 +119,37 @@ def generate_video(image_in, prompt, negative_prompt, progress=gr.Progress()):
         return video_path, log_messages
         
     except Exception as e:
-        return None, log_messages + f"\n❌ Fehler: {e}"
+        # Fehler abfangen und anzeigen
+        err_msg = str(e)
+        return None, log_messages + f"\n❌ Fehler: {err_msg}"
 
+# UI
 with gr.Blocks() as demo:
-    gr.Markdown("# I2VGen-XL ⚡ LCM TURBO")
-    gr.Markdown("Jetzt mit Live-Fortschrittsanzeige!")
+    gr.Markdown("# I2VGen-XL ⚡ LCM TURBO (Repariert)")
+    gr.Markdown("Wähle die Auflösung passend zur Geduld. **384px oder 448px** empfohlen!")
     
     with gr.Row():
         with gr.Column():
             img = gr.Image(type="pil", label="Bild")
             txt = gr.Textbox(label="Prompt", value="fireworks in the sky")
             neg = gr.Textbox(value="distortion, blurry", label="Negative")
+            
+            # DER WICHTIGE REGLER FÜR SPEED
+            resolution_slider = gr.Slider(
+                minimum=320, 
+                maximum=640, 
+                value=448, 
+                step=64, 
+                label="Auflösung (Pixel)",
+                info="384 = Schnell | 448 = Mittel | 512+ = Langsam"
+            )
+            
             btn = gr.Button("Turbo Start")
     
     with gr.Row():
         vid = gr.Video(label="Video")
         logs = gr.Textbox(label="Status Log", lines=10)
         
-    btn.click(generate_video, [img, txt, neg], [vid, logs])
+    btn.click(generate_video, [img, txt, neg, resolution_slider], [vid, logs])
 
 demo.launch()