Update app.py

app.py

@@ -21,6 +21,32 @@ ASSETS = {
     "logo": "logo.png"
 }
 
+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
+
 def validate_assets():
     """Check if all required asset files exist"""
     missing = []
@@ -64,7 +90,7 @@ def generate_and_wait_for_image(
             
             # Prepare all required files
             files = {
-                'input_image': open(input_image, 'rb'),
+                'input_image': resize_image_to_short_edge(input_image),
                 'logo_image': open(ASSETS["logo"], 'rb'),
                 'reference_image': open(ASSETS[garment_type]["reference_image"], 'rb')
             }
@@ -208,11 +234,8 @@ with gr.Blocks(title="Virtual Try-On Demo") as demo:
         queue=False
     )
     gr.Markdown("""
-    # Input image requirements:\n
-    - Supported formats: jpeg, png, webp
-    - Minimum Resolution: at least 64 on each side
-    - Total pixel count: 4,096 to 9,437,184 pixels
-    - Aspect ratio: 1:2.5 to 2.5:1
+    # Note:\n
+    The image will be resized to 1280px on the short edge.
     """)
 
 if __name__ == "__main__":