Update app.py

app.py

@@ -68,19 +68,45 @@ def process(data, api, api_key):
  
  return image
 
-def process_generate(fore, prompt, image_width, image_height, intensity, mode, refprompt, bg): 
+def resize_to_fit(max_size, original_size):
+ """
+ Calculate the new size for an image to fit within max_size while maintaining the aspect ratio.
+
+ :param max_size: Maximum allowed size as a tuple (width, height).
+ :param original_size: Original size of the image as a tuple (width, height).
+ :return: New size as a tuple (new_width, new_height) that fits within max_size while maintaining the aspect ratio.
+ """
+ original_width, original_height = original_size
+ max_width, max_height = max_size
+
+ # Calculate the scaling factor to maintain aspect ratio
+ width_ratio = max_width / original_width
+ height_ratio = max_height / original_height
+ scaling_factor = min(width_ratio, height_ratio)
+
+ # Calculate the new size while maintaining the aspect ratio
+ new_width = int(original_width * scaling_factor)
+ new_height = int(original_height * scaling_factor)
+
+ return new_width, new_height
+
+ 
+def process_generate(fore, prompt, intensity, mode, refprompt, bg): 
  
  size = fore.size
+ image_width = size[0]
+ image_height = size[1]
+ 
  if size[0]*size[1]<=(768*768):
  gr.Warning("ℹ️ The input image resolution is low, it might lead to some deformation!")
  
  if size[0]*size[1]>(1500*1500):
  gr.Warning("ℹ️ The input image size is too big, I will lower it!")
- 
- forestr = image_to_base64(fore.convert("RGBA"))
+ image_width, image_height = resize_to_fit((1500,1500), (image_width, image_height))
+
 
  
- 
+ forestr = image_to_base64(fore.convert("RGBA")) 
  data = {
  "foreground_image64": forestr,
  "prompt" : prompt,
@@ -182,7 +208,7 @@ with block:
 
 
 
- ips = [fore, prompt, image_width, image_height, intensity, mode, refprompt, bg]
+ ips = [fore, prompt, intensity, mode, refprompt, bg]
  generate_button.click(fn=process_generate, inputs=ips, outputs=[result_gallery])