Update app.py

app.py

@@ -17,12 +17,13 @@ app.mount("/static", StaticFiles(directory="static"), name="static")
 # Function to add padding
 def fill_rectangle_cropper(img, padding_type):
     # Calculate the average color of the image
-    avg_color_per_row = np.average(img, axis=0)
+    avg_color_per_row = np.average(np.array(img), axis=0)
     avg_color = np.average(avg_color_per_row, axis=0)
 
     if padding_type == "top_bottom":
-        if img.width > img.height:
-            new_height = img.width
+        # Check if padding is needed for top/bottom
+        if img.height < img.width:
+            new_height = img.width  # Make the height equal to width
             newimg = Image.new(
                 'RGB',
                 (img.width, new_height),
@@ -32,11 +33,12 @@ def fill_rectangle_cropper(img, padding_type):
             newimg.paste(img, (0, padding_top))
             return newimg
         else:
-            return img  # No change for already square images
+            return img  # No padding needed if height >= width
 
     elif padding_type == "left_right":
-        if img.height > img.width:
-            new_width = img.height
+        # Check if padding is needed for left/right
+        if img.width < img.height:
+            new_width = img.height  # Make the width equal to height
             newimg = Image.new(
                 'RGB',
                 (new_width, img.height),
@@ -46,7 +48,7 @@ def fill_rectangle_cropper(img, padding_type):
             newimg.paste(img, (padding_left, 0))
             return newimg
         else:
-            return img  # No change for already square images
+            return img  # No padding needed if width >= height
             
 # Function for cropping and filling the image
 def fill_rectangle_cropper1(img):