Update app.py

app.py

@@ -14,55 +14,50 @@ pipe = LucyEditPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfl
 pipe.to("cuda")
 
 def calculate_resolution(input_width, input_height, min_dimension=480, max_dimension=832):
-    """Calculate optimal resolution preserving aspect ratio"""
+    """Calculate optimal resolution preserving aspect ratio within bounds"""
     # Ensure dimensions are multiples of 16
     def round_to_16(x):
-        return int(round(x / 16.0) * 16)
+        return max(min_dimension, min(max_dimension, int(round(x / 16.0) * 16)))
     
     # Get aspect ratio
     aspect_ratio = input_width / input_height
     
-    # Square videos
-    if 0.95 <= aspect_ratio <= 1.05:
+    # Square videos (aspect ratio close to 1:1)
+    if 0.98 <= aspect_ratio <= 1.02:
         return 640, 640
     
     # Landscape videos (width > height)
     elif aspect_ratio > 1:
-        # Start with max width
+        # Try to use max width
         new_width = max_dimension
-        new_height = int(new_width / aspect_ratio)
+        new_height = new_width / aspect_ratio
         
-        # If height is too small, use min height instead
+        # If height would be too small, use min height
         if new_height < min_dimension:
             new_height = min_dimension
-            new_width = int(new_height * aspect_ratio)
-            
-            # Clamp width if needed
+            new_width = new_height * aspect_ratio
+            # If width exceeds max, clamp it
             if new_width > max_dimension:
                 new_width = max_dimension
-                new_height = int(new_width / aspect_ratio)
+        
+        return round_to_16(new_width), round_to_16(new_height)
     
-    # Portrait videos (height > width)  
+    # Portrait videos (height > width)
     else:
-        # Start with max height
+        # Try to use max height
         new_height = max_dimension
-        new_width = int(new_height * aspect_ratio)
+        new_width = new_height * aspect_ratio
         
-        # If width is too small, use min width instead
+        # If width would be too small, use min width
         if new_width < min_dimension:
             new_width = min_dimension
-            new_height = int(new_width / aspect_ratio)
-            
-            # Clamp height if needed
+            new_height = new_width / aspect_ratio
+            # If height exceeds max, clamp it
             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)))
-    
-    return final_width, final_height
+        
+        return round_to_16(new_width), round_to_16(new_height)
+
 
 @spaces.GPU(duration=90)
 def process_video(