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]