Update app.py

app.py

@@ -10,6 +10,7 @@ import io
 import os
 import warnings
 import math
+from math import sqrt
 from IPython.display import display
 from PIL import Image
 from stability_sdk import client
@@ -78,43 +79,75 @@ def img2img( path ,design,x_prompt,alt_prompt,strength,guidance_scale,steps):
 #####
 
     
-    max_pixels = 1048576
-    
-    # Open the image and retrieve its width and height
+    # max_pixels = 1048576
     img = Image.open(path)
     width, height = img.size
-    
-    # Calculate the number of pixels in the image
     num_pixels = width * height
     
-    # Check if the number of pixels is within the allowed range
-    if num_pixels > max_pixels:
-        # Calculate the maximum width and height based on the number of pixels in the original image
-        max_width = int(math.sqrt(max_pixels * width / height))
-        max_height = int(math.sqrt(max_pixels * height / width))
-    
-        # Calculate the new dimensions of the image, making sure that the width and height are within the allowed range
-        # and maintain the aspect ratio of the original image
-        if width > height:
-            new_width = max_width
-            new_height = int(new_width * height / width)
-        else:
-            new_height = max_height
-            new_width = int(new_height * width / height)
+    # Calculate the maximum number of pixels allowed
+    max_pixels = 1048576
+
+    # Calculate the new size of the image, making sure that the number of pixels does not exceed the maximum limit
+    if width * height > max_pixels:
+        # Calculate the new width and height of the image
+        ratio = width / height
+        new_width = int(math.sqrt(max_pixels * ratio))
+        new_height = int(math.sqrt(max_pixels / ratio))
     else:
-        # Set the new dimensions to the original dimensions of the image
         new_width = width
         new_height = height
+
+    # Make sure that either the width or the height of the resized image is a multiple of 64
+    if new_width % 64 != 0:
+        new_width = ((new_width + 63) // 64) * 64
+    if new_height % 64 != 0:
+        new_height = ((new_height + 63) // 64) * 64
+
+    # Resize the image
+    img = img.resize((new_width, new_height), resample=Image.BILINEAR)
+
+    # Check if the number of pixels in the resized image is within the maximum limit
+    # If not, adjust the width and height of the image to bring the number of pixels within the maximum limit
+    if new_width * new_height > max_pixels:
+        while new_width * new_height > max_pixels:
+            new_width -= 1
+            new_height = int(max_pixels / new_width)
+
+
+
     
-    # Calculate the new size of the image, making sure that the width and height are multiples of 64
-    new_width = ((new_width + 63) // 64) * 64
-    new_height = ((new_height + 63) // 64) * 64
+            # Calculate the closest multiple of 64 for each value
+            if new_width % 64 != 0:
+                new_width = (new_width // 64) * 64
+            if new_height % 64 != 0:
+                new_height = (new_height // 64) * 64
+
+            # Make sure that the final values are less than the original values
+            if new_width > 1407:
+                new_width -= 64
+            if new_height > 745:
+                new_height -= 64
+    
+    new_height ,new_width
+    # Initialize the values
+    widthz = new_width
+    heightz = new_height
+
+    # Calculate the closest multiple of 64 for each value
+    if widthz % 64 != 0:
+        widthz = (widthz // 64) * 64
+    if heightz % 64 != 0:
+        heightz = (heightz // 64) * 64
+
+    # Make sure that the final values are less than the original values
+    if widthz > 1407:
+        widthz -= 64
+    if heightz > 745:
+        heightz -= 64
+
     
-    # Resize the image
-    img = img.resize((new_width, new_height), resample=Image.Resampling.BILINEAR)  
-     
- 
 
+    img = img.resize((widthz, heightz), resample=Image.BILINEAR)