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])