duration max 25, steps_per_chunk param

app.py

@@ -921,6 +921,7 @@ def generate_chunk_handler(
     ipnc_svicamera_l: float = 100.0,
     # chunking params
     session_id: str = "",
+    steps_per_chunk: str = "2",
     chunk_step: str = "0",
     progress=gr.Progress(track_tqdm=True),
 ):
@@ -952,7 +953,7 @@ def generate_chunk_handler(
         gen_w, gen_h = _compute_dims(str(aspect_ratio), image_paths[0], bool(use_ref_aspect))
         n_frames       = max(1, round(float(duration_secs) * int(base_fps)))
         steps          = int(num_steps)
-        STEPS_PER_CHUNK = 2
+        STEPS_PER_CHUNK = max(1, int(float(steps_per_chunk or 2)))
         chunk_idx       = int(float(chunk_step or 0))
         step_start      = chunk_idx * STEPS_PER_CHUNK
         step_end        = min(step_start + STEPS_PER_CHUNK, steps)
@@ -992,7 +993,7 @@ def generate_chunk_handler(
         sampler_id = "57" if use_high else "58"
 
         # Inject chunk nodes: save/load latent, save/load conditioning
-        # Batch multiple steps per chunk (STEPS_PER_CHUNK=2 for short videos)
+        # Batch multiple steps per chunk (set by the caller)
         api_wf[sampler_id]["inputs"]["steps"]         = steps
         api_wf[sampler_id]["inputs"]["start_at_step"]  = step_start
         api_wf[sampler_id]["inputs"]["end_at_step"]    = step_end
@@ -1258,7 +1259,7 @@ with gr.Blocks(title="Bernini-R Wan 2.2 R2V Lightning") as demo:
                 gpu_budget = gr.Slider(0, 540, value=0, step=10, label="ZeroGPU budget (0=auto)")
             with gr.Row():
                 num_steps = gr.Slider(4, 20, value=6, step=1, label="Steps")
-                duration_secs = gr.Slider(1, 15, value=5, step=0.5, label="Duration (s)")
+                duration_secs = gr.Slider(1, 25, value=5, step=0.5, label="Duration (s)")
                 sampler_name = gr.Dropdown(choices=["uni_pc", "lcm"], value="lcm", label="Sampler")
                 base_fps = gr.Number(value=15, precision=0, label="Combine FPS")
                 rife_mode = gr.Dropdown(
@@ -1293,6 +1294,7 @@ with gr.Blocks(title="Bernini-R Wan 2.2 R2V Lightning") as demo:
     )
 
     _chunk_session_id = gr.Textbox(visible=False, value="")
+    _chunk_spc_txt    = gr.Textbox(visible=False, value="2")
     _chunk_step_txt   = gr.Textbox(visible=False, value="0")
     _chunk_btn        = gr.Button(visible=False)
     _chunk_btn.click(
@@ -1309,7 +1311,7 @@ with gr.Blocks(title="Bernini-R Wan 2.2 R2V Lightning") as demo:
             ipnc_wamu_l, ipnc_dreamly_l,
             s_svicamera_h, s_svicamera_l,
             ipnc_svicamera_h, ipnc_svicamera_l,
-            _chunk_session_id, _chunk_step_txt,
+            _chunk_session_id, _chunk_spc_txt, _chunk_step_txt,
         ],
         outputs=[output_video, output_status],
         api_name="generate_chunk",