Update app.py

app.py

@@ -12,17 +12,22 @@ generated_gifs = []
 def preprocess_image(image_data):
     # Extract the image and crop data from the ImageEditor output
     image = image_data["image"].convert('RGBA')
-    crop_data = image_data["crop"]
+    crop_data = image_data.get("crop")
 
     if crop_data:
         # Apply cropping if crop data is available
         left, top, width, height = crop_data
         image = image.crop((left, top, left + width, top + height))
 
-    # Ensure the image is square by padding it
-    size = max(image.size)
-    new_image = Image.new("RGBA", (size, size), (255, 255, 255, 0))
-    new_image.paste(image, ((size - image.width) // 2, (size - image.height) // 2))
+    # Resize image to fit within 256x256 while maintaining aspect ratio
+    image.thumbnail((256, 256), Image.LANCZOS)
+
+    # Create a new 256x256 image with a transparent background
+    new_image = Image.new("RGBA", (256, 256), (255, 255, 255, 0))
+    
+    # Paste the resized image at the center
+    offset = ((256 - image.width) // 2, (256 - image.height) // 2)
+    new_image.paste(image, offset)
     
     return new_image
 
@@ -36,10 +41,7 @@ def create_gif(editor1_output, editor2_output, transition_type):
         img1 = preprocess_image(editor1_output)
         img2 = preprocess_image(editor2_output)
 
-        # Set size for the GIF
         size = (256, 256)
-        img1 = img1.resize(size, Image.LANCZOS)
-        img2 = img2.resize(size, Image.LANCZOS)
 
         if transition_type == "slide":
             # Calculate step size for consistent speed
@@ -128,8 +130,8 @@ with gr.Blocks() as iface:
                 label="Edit Image 1",
                 brush=gr.Brush(colors=["#ff0000", "#00ff00", "#0000ff"]),
                 eraser=gr.Eraser(default_size=10),
-                height=400,
-                width="100%",
+                height=350,
+                width=350,
                 crop_size="1:1",
                 layers=True,
                 type="pil"
@@ -139,8 +141,8 @@ with gr.Blocks() as iface:
                 label="Edit Image 2",
                 brush=gr.Brush(colors=["#ff0000", "#00ff00", "#0000ff"]),
                 eraser=gr.Eraser(default_size=10),
-                height=400,
-                width="100%",
+                height=350,
+                width=350,
                 crop_size="1:1",
                 layers=True,
                 type="pil"
@@ -157,7 +159,7 @@ with gr.Blocks() as iface:
             elem_id="output_gallery",
             columns=3,
             rows=2,
-            height=400,
+            height=350,
             object_fit="contain"
         )