Fix: auto-detect portrait/landscape orientation for proper aspect ratio

app.py

@@ -40,8 +40,22 @@ default_negative_prompt = "low quality, worst quality, blurry, distorted, deform
 # IMAGE RESIZING LOGIC
 # =========================================================
 
-def resize_image(image: Image.Image, width: int = 1024, height: int = 576) -> Image.Image:
-    return image.resize((width, height), Image.LANCZOS)
+def resize_image(image: Image.Image) -> Image.Image:
+    width, height = image.size
+    aspect_ratio = width / height
+
+    # SVD works best with 1024x576 or 576x1024
+    if aspect_ratio > 1:  # Landscape
+        new_width = 1024
+        new_height = 576
+    elif aspect_ratio < 1:  # Portrait
+        new_width = 576
+        new_height = 1024
+    else:  # Square
+        new_width = 768
+        new_height = 768
+
+    return image.resize((new_width, new_height), Image.LANCZOS)
 
 # =========================================================
 # MAIN GENERATION FUNCTION