Update app.py

app.py

@@ -654,33 +654,16 @@ def calculate_frames(duration: float, frame_rate: float = DEFAULT_FRAME_RATE) ->
     return min(frames, MAX_FRAMES)
 
 
-def validate_resolution(height: int, width: int) -> tuple[int, int]:
-    height = round(height / STEP) * STEP
-    width = round(width / STEP) * STEP
-    height = max(MIN_DIM, min(height, MAX_DIM))
-    width = max(MIN_DIM, min(width, MAX_DIM))
-    return height, width
-
-
-def detect_aspect_ratio(image) -> str:
-    if image is None:
-        return "16:9"
-    if hasattr(image, "size"):
-        w, h = image.size
-    elif hasattr(image, "shape"):
-        h, w = image.shape[:2]
-    else:
-        return "16:9"
-    ratio = w / h
-    candidates = {"16:9": 16/9, "9:16": 9/16, "1:1": 1.0}
-    return min(candidates, key=lambda k: abs(ratio - candidates[k]))
+def on_image_upload(first_image, last_image, high_res):
+    ref_image = first_image if first_image is not None else last_image
+    aspect = detect_aspect_ratio(ref_image)
+    tier = "high" if high_res else "low"
+    w, h = RESOLUTIONS[tier][aspect]
+    return gr.update(value=w), gr.update(value=h)
 
 
-def on_image_upload(first_image, last_image, high_res):
-    # Always use first_image for resolution calculation
-    ref_image = first_image
-    if ref_image is None:
-        return gr.update(), gr.update()
+def on_highres_toggle(first_image, last_image, high_res):
+    ref_image = first_image if first_image is not None else last_image
     aspect = detect_aspect_ratio(ref_image)
     tier = "high" if high_res else "low"
     w, h = RESOLUTIONS[tier][aspect]