Update app.py

app.py

@@ -77,22 +77,22 @@ def img2img( path ,design,x_prompt,alt_prompt,strength,guidance_scale,steps):
 
 #####
 
-
     
- 
- 
-
- 
-
-# Maximum number of pixels allowed in the image
     max_pixels = 1048576
     
     # Open the image and retrieve its width and height
     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:
+    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:
@@ -101,15 +101,23 @@ def img2img( path ,design,x_prompt,alt_prompt,strength,guidance_scale,steps):
         else:
             new_height = max_height
             new_width = int(new_height * width / height)
+    else:
+        # Set the new dimensions to the original dimensions of the image
+        new_width = width
+        new_height = height
     
-        # 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
-    
-        # Resize the image
-        img = img.resize((new_width, new_height), resample=Image.Resampling.BILINEAR)
-       
+    # 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
     
+    # Resize the image
+    img = img.resize((new_width, new_height), resample=Image.Resampling.BILINEAR)  
+     
+ 
+
+ 
+
+
     
 
 ########