Update app.py

app.py

@@ -21,34 +21,37 @@ ASSETS = {
     "logo": "logo.png"
 }
 
+def needs_resizing(image_path, max_short_edge=1280):
+    # Check if image needs resizing (returns True if short edge > max_short_edge)
+    with Image.open(image_path) as img:
+        width, height = img.size
+        return min(width, height) > max_short_edge
+
 def resize_image_to_short_edge(image_path, max_short_edge=1280):
-    """Resize image so the short edge is at most max_short_edge pixels"""
-    img = Image.open(image_path)
-    width, height = img.size
-    
-    # Determine scaling factor
-    if width < height:
-        if width <= max_short_edge:
-            return image_path  # No resizing needed
-        scaling_factor = max_short_edge / width
-    else:
-        if height <= max_short_edge:
-            return image_path  # No resizing needed
-        scaling_factor = max_short_edge / height
-    
-    # Calculate new dimensions and resize
-    new_width = int(width * scaling_factor)
-    new_height = int(height * scaling_factor)
-    resized_img = img.resize((new_width, new_height), Image.LANCZOS)
-    
-    # Save to bytes buffer
-    buffer = io.BytesIO()
-    resized_img.save(buffer, format=img.format)
-    buffer.seek(0)
-    return buffer
+    # Resize image so the short edge is at most max_short_edge pixels
+    with Image.open(image_path) as img:
+        width, height = img.size
+        format = img.format
+        
+        # Determine scaling factor
+        if width < height:
+            scaling_factor = max_short_edge / width
+        else:
+            scaling_factor = max_short_edge / height
+        
+        # Calculate new dimensions and resize
+        new_width = int(width * scaling_factor)
+        new_height = int(height * scaling_factor)
+        resized_img = img.resize((new_width, new_height), Image.LANCZOS)
+        
+        # Save to bytes buffer
+        buffer = io.BytesIO()
+        resized_img.save(buffer, format=format)
+        buffer.seek(0)
+        return buffer
 
 def validate_assets():
-    """Check if all required asset files exist"""
+    # Check if all required asset files exist
     missing = []
     for asset_type in ["top", "bottom"]:
         if not os.path.exists(ASSETS[asset_type]["reference_image"]):
@@ -64,7 +67,7 @@ def generate_and_wait_for_image(
     garment_type: str,
     progress=gr.Progress()
 ):
-    """Make POST request and automatically poll for the result image"""
+    # Make POST request and automatically poll for the result image
     # Validate assets first
     try:
         validate_assets()
@@ -90,10 +93,16 @@ def generate_and_wait_for_image(
             
             # Prepare all required files
             files = {
-                'input_image': resize_image_to_short_edge(input_image),
                 'logo_image': open(ASSETS["logo"], 'rb'),
                 'reference_image': open(ASSETS[garment_type]["reference_image"], 'rb')
             }
+            
+            # Only resize input_image if needed
+            if needs_resizing(input_image):
+                files['input_image'] = resize_image_to_short_edge(input_image)
+            else:
+                files['input_image'] = open(input_image, 'rb')
+
             data = {
                 'reference_image_type': (f'{garment_type}'),
                 'output_format': (None, "png")  # Hardcoded to PNG