Update app.py

app.py

@@ -28,7 +28,29 @@ def image_to_base64(image):
         # Encode the binary data to a base64 string
         base64_image = base64.b64encode(binary_image_data).decode("utf-8")
         return base64_image
-    
+
+def create_square_image(image):
+    """
+    Create a new square image with the side length equal to the largest dimension
+    of the original image and paste the original image at the center on a transparent canvas.
+
+    :param image: A PIL image.
+    :return: A new square PIL image.
+    """
+    original_width, original_height = image.size
+    new_side_length = max(original_width, original_height)
+
+    # Create a new square image with a transparent background
+    new_image = Image.new("RGBA", (new_side_length, new_side_length), (255, 255, 255, 0))
+
+    # Calculate the position to paste the original image on the new square canvas
+    paste_x = (new_side_length - original_width) // 2
+    paste_y = (new_side_length - original_height) // 2
+
+    # Paste the original image onto the new square canvas using the alpha channel as a mask
+    new_image.paste(image, (paste_x, paste_y), image)
+
+    return new_image
      
 def process(data, api, api_key):
     
@@ -91,8 +113,9 @@ def resize_to_fit(max_size, original_size):
     return new_width, new_height
 
     
-def process_generate(fore, prompt, intensity, mode, refprompt, bg):    
-        
+def process_generate(fore, prompt, intensity, mode, refprompt, bg): 
+
+    fore = create_square_image(fore)
     size = fore.size
     image_width = size[0]
     image_height = size[1]