Resize input image

app_gradio.py

@@ -163,6 +163,24 @@ def generate_output(image, apply_filter, prompt: str, num_seeds: int = 3, lambda
     
     # Save the input image temporarily
     temp_image_path = os.path.join(exp_dir, "temp_input.png")
+
+    # Get original dimensions
+    width, height = image.size
+
+    # Step 1: Make the image square if it's not already
+    if width != height:
+        # Use the smaller dimension to crop to a square
+        min_dim = min(width, height)
+        left = (width - min_dim) // 2
+        top = (height - min_dim) // 2
+        right = left + min_dim
+        bottom = top + min_dim
+        image = image.crop((left, top, right, bottom))
+
+    # Step 2: Resize to 512x512 if the image is larger
+    if image.size[0] > 512:  # Image is square at this point
+        image = image.resize((512, 512), Image.LANCZOS)
+
     image.save(temp_image_path)
     
     # Generate the GIFs