Update app.py

app.py

@@ -14,7 +14,7 @@ app = FastAPI()
 app.mount("/static", StaticFiles(directory="static"), name="static")
 
 # Function for cropping and filling the image
-def fill_square_cropper(img):
+def fill_rectangle_cropper(img):
     imgsz = [img.height, img.width]
     avg_color_per_row = np.average(img, axis=0)
     avg_color = np.average(avg_color_per_row, axis=0)
@@ -25,7 +25,8 @@ def fill_square_cropper(img):
             (img.height, img.height),
             (round(avg_color[0]), round(avg_color[1]), round(avg_color[2]))
         )
-        newpos = (img.height - img.width) // 2
+        newpos = (img.width - img.height) // 2
+       
         newimg.paste(img, (newpos, 0))
         return newimg
 
@@ -35,7 +36,7 @@ def fill_square_cropper(img):
             (img.width, img.width),
             (round(avg_color[0]), round(avg_color[1]), round(avg_color[2]))
         )
-        newpos = (img.width - img.height) // 2
+        newpos = (img.height - img.width) // 2
         newimg.paste(img, (0, newpos))
         return newimg
     else:
@@ -109,7 +110,7 @@ def demo_page():
     # Process the first image
     response = requests.get(url1)
     img1 = Image.open(BytesIO(response.content)).convert("RGB")
-    squared_img1 = fill_square_cropper(img1)
+    squared_img1 = fill_rectangle_cropper_cropper(img1)
     output1 = BytesIO()
     squared_img1.save(output1, format="JPEG")
     encoded_img1 = base64.b64encode(output1.getvalue()).decode("utf-8")
@@ -117,7 +118,7 @@ def demo_page():
     # Process the second image
     response = requests.get(url2)
     img2 = Image.open(BytesIO(response.content)).convert("RGB")
-    squared_img2 = fill_square_cropper(img2)
+    squared_img2 = fill_rectangle_cropper_cropper(img2)
     output2 = BytesIO()
     squared_img2.save(output2, format="JPEG")
     encoded_img2 = base64.b64encode(output2.getvalue()).decode("utf-8")
@@ -169,7 +170,7 @@ async def upload_file(file: UploadFile = File(...)):
         # Await file upload
         contents = await file.read()
         img = Image.open(BytesIO(contents)).convert("RGB")
-        squared_img = fill_square_cropper(img)
+        squared_img = fill_rectangle_cropper_cropper(img)
 
         # Save the squared image (original size)
         output = BytesIO()