Update app.py

app.py

@@ -96,9 +96,11 @@ def generate_video(
     guidance_scale: float = 5.0,
     seed: int = -1,
     enabled_loras: list = None,
-    lora_strength_multiplier: float = 1.0
+    lora_strength_multiplier: float = 1.0,
+    progress=gr.Progress()
 ):
     try:
+
         pipeline = initialize_pipeline()
 
         # ── Hot-swap / enable only selected LoRAs ───────
@@ -153,7 +155,18 @@ def generate_video(
             gen_params["image"] = image
 
         print(f"Generating: {width}x{height}, {num_frames} frames, steps={num_inference_steps}")
-        output = pipeline(**gen_params).frames[0]
+        #output = pipeline(**gen_params).frames[0]
+
+        progress(0, desc="Starting generation...")
+
+        def step_callback(step, timestep, latents):
+            progress(step / num_inference_steps, desc=f"Step {step+1}/{num_inference_steps}")
+        
+        output = pipeline(
+            **gen_params,
+            callback=step_callback,
+            callback_steps=1,
+        ).frames[0] 
 
         output_path = "output.mp4"
         export_to_video(output, output_path, fps=24)