Update app_lora.py

app_lora.py

@@ -48,8 +48,8 @@ SLIDER_MIN_W, SLIDER_MAX_W = 128, 1024
 MAX_SEED = np.iinfo(np.int32).max
 
 FIXED_FPS = 24
-MIN_FRAMES_MODEL = 8
-MAX_FRAMES_MODEL = 81 
+MIN_FRAMES_MODEL = 8  # Minimum 8 frames (~0.33s)
+MAX_FRAMES_MODEL = 240  # Changed from 81 to 240 (10 seconds at 24fps)
 
 default_prompt_i2v = "make this image come alive, cinematic motion, smooth animation"
 default_negative_prompt = "Bright tones, overexposed, static, blurred details, subtitles, style, works, paintings, images, static, overall gray, worst quality, low quality, JPEG compression residue, ugly, incomplete, extra fingers, poorly drawn hands, poorly drawn faces, deformed, disfigured, misshapen limbs, fused fingers, still picture, messy background, three legs, many people in the background, walking backwards, watermark, text, signature"
@@ -91,14 +91,15 @@ def handle_image_upload_for_dims_wan(uploaded_pil_image, current_h_val, current_
         return gr.update(value=DEFAULT_H_SLIDER_VALUE), gr.update(value=DEFAULT_W_SLIDER_VALUE)
 
 def get_duration(input_image, prompt, height, width, 
-                   negative_prompt, duration_seconds,
-                   guidance_scale, steps,
-                   seed, randomize_seed, 
-                   progress):
-    if steps > 4 and duration_seconds > 2:
+               negative_prompt, duration_seconds,
+               guidance_scale, steps,
+               seed, randomize_seed, 
+               progress):
+    # Longer timeout for longer videos
+    if steps > 4 and duration_seconds > 5:  # Changed from 2 to 5
+        return 120  # Increased timeout for long generations
+    elif steps > 4 or duration_seconds > 5:  # Changed from 2 to 5
         return 90
-    elif steps > 4 or duration_seconds > 2:
-        return 75
     else:
         return 60
 
@@ -141,8 +142,14 @@ with gr.Blocks() as demo:
         with gr.Column():
             input_image_component = gr.Image(type="pil", label="Input Image (auto-resized to target H/W)")
             prompt_input = gr.Textbox(label="Prompt", value=default_prompt_i2v)
-            duration_seconds_input = gr.Slider(minimum=round(MIN_FRAMES_MODEL/FIXED_FPS,1), maximum=round(MAX_FRAMES_MODEL/FIXED_FPS,1), step=0.1, value=2, label="Duration (seconds)", info=f"Clamped to model's {MIN_FRAMES_MODEL}-{MAX_FRAMES_MODEL} frames at {FIXED_FPS}fps.")
-            
+            duration_seconds_input = gr.Slider(
+                minimum=round(MIN_FRAMES_MODEL/FIXED_FPS,1),  # 0.3s (8 frames)
+                maximum=round(MAX_FRAMES_MODEL/FIXED_FPS,1),  # Now 10.0s (240 frames)
+                step=0.1, 
+                value=3,  # Changed default from 2 to 3 seconds
+                label="Duration (seconds)", 
+                info=f"Clamped to {MIN_FRAMES_MODEL}-{MAX_FRAMES_MODEL} frames at {FIXED_FPS}fps (~{MIN_FRAMES_MODEL/FIXED_FPS:.1f}-{MAX_FRAMES_MODEL/FIXED_FPS:.1f}s)"
+            )            
             with gr.Accordion("Advanced Settings", open=False):
                 negative_prompt_input = gr.Textbox(label="Negative Prompt", value=default_negative_prompt, lines=3)
                 seed_input = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=42, interactive=True)