Update app.py

app.py

@@ -16,45 +16,33 @@ app.mount("/static", StaticFiles(directory="static"), name="static")
 
 # Function to add padding
 def fill_rectangle_cropper(img, padding_type):
-    # Ensure the image is correctly converted to a NumPy array for color calculations
+    # Calculate the average color of the image
     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":
-        # Ensure height is less than width before adding padding
-        if img.height < img.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]))  # Fill with average color
-            )
-            padding_top = (new_height - img.height) // 2
-            newimg.paste(img, (0, padding_top))  # Center the image vertically
-            return newimg
-        else:
-            print("No padding needed for top/bottom.")  # Debug message
-            return img
+        # Increase height to create a rectangle
+        new_height = int(img.width * (4/3))  # Example: height = width * 4/3
+        newimg = Image.new(
+            'RGB',
+            (img.width, new_height),
+            (round(avg_color[0]), round(avg_color[1]), round(avg_color[2]))
+        )
+        padding_top = (new_height - img.height) // 2
+        newimg.paste(img, (0, padding_top))  # Center the image vertically
+        return newimg
 
     elif padding_type == "left_right":
-        # Ensure width is less than height before adding padding
-        if img.width < img.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]))  # Fill with average color
-            )
-            padding_left = (new_width - img.width) // 2
-            newimg.paste(img, (padding_left, 0))  # Center the image horizontally
-            return newimg
-        else:
-            print("No padding needed for left/right.")  # Debug message
-            return img
-
-    else:
-        print("Invalid padding type.")  # Debug message
-        return img
+        # Increase width to create a rectangle
+        new_width = int(img.height * (4/3))  # Example: width = height * 4/3
+        newimg = Image.new(
+            'RGB',
+            (new_width, img.height),
+            (round(avg_color[0]), round(avg_color[1]), round(avg_color[2]))
+        )
+        padding_left = (new_width - img.width) // 2
+        newimg.paste(img, (padding_left, 0))  # Center the image horizontally
+        return newimg
 
 # Function for cropping and filling the image
 def fill_rectangle_cropper1(img):