Revert "Use simpler calculation"

This reverts commit 2df8b3be08e433b775ba07173cde88e050a3155d.

app.py

@@ -91,7 +91,31 @@ def get_duration(
     progress=None,
 ):
     """Calculate dynamic GPU duration based on parameters."""
-    return 10 + int(sd_steps) * 5
+    BASE_FRAMES_HEIGHT_WIDTH = 81 * 832 * 624
+    BASE_STEP_DURATION = 15
+
+    # Get dimensions from resolution or from first image
+    if flf2vid_image_first is not None:
+        width, height = flf2vid_image_first.size
+    else:
+        # Fallback to resolution string mapping
+        resolution_map = {
+            "720P": (1280, 720),
+            "1280x720": (1280, 720),
+            "480P": (832, 480),
+            "832x480": (832, 480),
+        }
+        width, height = resolution_map.get(resolution, (1280, 720))
+
+    # Use frame_num directly (already provided)
+    frames = int(frame_num) if frame_num else 81
+
+    # Calculate duration factor
+    factor = frames * width * height / BASE_FRAMES_HEIGHT_WIDTH
+    step_duration = BASE_STEP_DURATION * (factor**1.5)
+
+    # Return total duration in seconds
+    return 10 + int(sd_steps) * step_duration
 
 
 @spaces.GPU(duration=get_duration)