Update app.py

app.py

@@ -16,40 +16,46 @@ 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
+    # Ensure the image is correctly converted to a NumPy array for color calculations
     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":
-        # Check if padding is needed for top/bottom
+        # Ensure height is less than width before adding padding
         if img.height < img.width:
-            new_height = img.width  # Make the height equal to width
+            new_height = img.width  # Set new height equal to width
             newimg = Image.new(
                 'RGB',
                 (img.width, new_height),
-                (round(avg_color[0]), round(avg_color[1]), round(avg_color[2]))
+                (round(avg_color[0]), round(avg_color[1]), round(avg_color[2]))  # Fill with average color
             )
             padding_top = (new_height - img.height) // 2
-            newimg.paste(img, (0, padding_top))
+            newimg.paste(img, (0, padding_top))  # Center the image vertically
             return newimg
         else:
-            return img  # No padding needed if height >= width
+            print("No padding needed for top/bottom.")  # Debug message
+            return img
 
     elif padding_type == "left_right":
-        # Check if padding is needed for left/right
+        # Ensure width is less than height before adding padding
         if img.width < img.height:
-            new_width = img.height  # Make the width equal to height
+            new_width = img.height  # Set new width equal to height
             newimg = Image.new(
                 'RGB',
                 (new_width, img.height),
-                (round(avg_color[0]), round(avg_color[1]), round(avg_color[2]))
+                (round(avg_color[0]), round(avg_color[1]), round(avg_color[2]))  # Fill with average color
             )
             padding_left = (new_width - img.width) // 2
-            newimg.paste(img, (padding_left, 0))
+            newimg.paste(img, (padding_left, 0))  # Center the image horizontally
             return newimg
         else:
-            return img  # No padding needed if width >= height
-            
+            print("No padding needed for left/right.")  # Debug message
+            return img
+
+    else:
+        print("Invalid padding type.")  # Debug message
+        return img
+
 # Function for cropping and filling the image
 def fill_rectangle_cropper1(img):
     imgsz = [img.height, img.width]