Fix: canonical /edit_video API endpoint (7 inputs) + input-scaled GPU duration (~68s worst case vs fixed 300)

app.py

@@ -74,8 +74,33 @@ def chunks_for_seconds(seconds: float) -> int:
     return max(1, 1 + (wanted - 1) // FRAMES_PER_CHUNK)
 
 
+def _estimate_duration(
+    video_path=None,
+    instruction="",
+    seconds=2.0,
+    seed=42,
+    randomize_seed=False,
+    num_inference_steps=2,
+    *args,
+    **kwargs,
+):
+    """Reserve GPU time scaled to the real per-chunk cost.
+
+    Measured on the live sm_120 slice: ~1.0 s per (chunk × step) plus fixed
+    overhead (condition encode + mux). Worst case (12 chunks × 4 steps) runs in
+    ~48 s of GPU compute, so 15 + units × 1.1 (≈68 s worst case) covers it with
+    a modest margin while keeping short runs cheap and high-priority.
+    """
+    try:
+        chunks = chunks_for_seconds(float(seconds))
+        steps = max(1, int(num_inference_steps))
+    except (TypeError, ValueError):
+        chunks, steps = 12, 4
+    return int(min(120, 15 + chunks * steps * 1.1))
+
+
 # ---------------------------------------------------------------------- handler
-@spaces.GPU(duration=300, size="large")
+@spaces.GPU(duration=_estimate_duration, size="large")
 def edit_video(
     video_path,
     instruction,
@@ -251,8 +276,13 @@ result streams back while it is still being generated (2 flow-matching steps per
 
     inputs = [video_in, instruction, seconds, seed, randomize_seed, steps, reference_image]
     outputs = [preview, video_out, status]
-    run_btn.click(edit_video, inputs=inputs, outputs=outputs, concurrency_limit=1)
-    instruction.submit(edit_video, inputs=inputs, outputs=outputs, concurrency_limit=1)
+    # Canonical API endpoint — the click handler owns `/edit_video` with the full
+    # 7-input signature. The submit handler gets a distinct name so it can't shadow
+    # the primary binding or register with a truncated signature.
+    run_btn.click(edit_video, inputs=inputs, outputs=outputs,
+                  concurrency_limit=1, api_name="edit_video")
+    instruction.submit(edit_video, inputs=inputs, outputs=outputs,
+                       concurrency_limit=1, api_name="edit_video_submit")
 
 if __name__ == "__main__":
     demo.queue(max_size=12).launch(theme=gr.themes.Citrus(), css=CSS)