Update app.py

app.py

@@ -79,37 +79,38 @@ def img2img( path ,design,x_prompt,alt_prompt,strength,guidance_scale,steps):
 
 
     
-    # Path to the image
-    # path = "path/to/image"
-    
-    # Maximum number of pixels allowed in the image
+ 
+ 
+
+ 
+
+# 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
     num_pixels = width * height
-    
     # Check if the number of pixels is within the allowed range
-    if num_pixels > max_pixels:
-        # Scale down the image to have the maximum allowed number of pixels, while maintaining its aspect ratio
-        factor = math.sqrt(max_pixels / (width * height))
-        new_width = int(width * factor)
-        new_height = int(height * factor)
-    else:
-        # Set the new dimensions to the original dimensions of the image
-        new_width = width
-        new_height = height
+    if num_pixels <= max_pixels:
+        # 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 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 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)
+       
     
-    # Resize the image
-    img = img.resize((new_width, new_height), resample=Image.Resampling.BILINEAR)
     
-
-
 
 ########   
     max_attempts = 5 # maximum number of attempts before giving up