Update app.py

app.py

@@ -13,7 +13,7 @@ vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=to
 pipe = LucyEditPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
 pipe.to("cuda")
 
-def calculate_resolution(input_width, input_height, max_dimension=832):
+def calculate_resolution(input_width, input_height, min_dimension=480, max_dimension=832):
     """Calculate optimal resolution preserving aspect ratio"""
     # Ensure dimensions are multiples of 16
     def round_to_16(x):
@@ -24,40 +24,45 @@ def calculate_resolution(input_width, input_height, max_dimension=832):
     
     # Square videos
     if 0.95 <= aspect_ratio <= 1.05:
-        return 512, 512
+        return 640, 640
     
     # Landscape videos (width > height)
     elif aspect_ratio > 1:
-        if input_width > input_height:
-            # Fit to max width
-            new_width = min(max_dimension, input_width)
-            new_height = int(new_width / aspect_ratio)
+        # Start with max width
+        new_width = max_dimension
+        new_height = int(new_width / aspect_ratio)
+        
+        # If height is too small, use min height instead
+        if new_height < min_dimension:
+            new_height = min_dimension
+            new_width = int(new_height * aspect_ratio)
             
-            # Ensure height doesn't exceed max
-            if new_height > 480:
-                new_height = 480
-                new_width = int(new_height * aspect_ratio)
-        else:
-            new_width = max_dimension
-            new_height = int(new_width / aspect_ratio)
+            # Clamp width if needed
+            if new_width > max_dimension:
+                new_width = max_dimension
+                new_height = int(new_width / aspect_ratio)
     
-    # Portrait videos (height > width)
+    # Portrait videos (height > width)  
     else:
-        if input_height > input_width:
-            # Fit to max height
-            new_height = min(max_dimension, input_height)
-            new_width = int(new_height * aspect_ratio)
+        # Start with max height
+        new_height = max_dimension
+        new_width = int(new_height * aspect_ratio)
+        
+        # If width is too small, use min width instead
+        if new_width < min_dimension:
+            new_width = min_dimension
+            new_height = int(new_width / aspect_ratio)
             
-            # Ensure width doesn't exceed max
-            if new_width > 480:
-                new_width = 480
-                new_height = int(new_width / aspect_ratio)
-        else:
-            new_height = max_dimension
-            new_width = int(new_height * aspect_ratio)
+            # Clamp height if needed
+            if new_height > max_dimension:
+                new_height = max_dimension
+                new_width = int(new_height / aspect_ratio)
+    
+    # Round to multiples of 16 and ensure within bounds
+    final_width = round_to_16(max(min_dimension, min(max_dimension, new_width)))
+    final_height = round_to_16(max(min_dimension, min(max_dimension, new_height)))
     
-    # Round to multiples of 16
-    return round_to_16(new_width), round_to_16(new_height)
+    return final_width, final_height
 
 @spaces.GPU(duration=90)
 def process_video(