Update app.py

app.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, File, UploadFile
+from fastapi import FastAPI, File, UploadFilek, Form
 from fastapi.responses import HTMLResponse
 from fastapi.staticfiles import StaticFiles
 import numpy as np
@@ -7,14 +7,49 @@ from io import BytesIO
 import requests
 import base64
 import os
+from tkinter import Tk, Label, Button, Radiobutton, IntVar
 
 app = FastAPI()
 
 # Mount the static folder for CSS and other assets
 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 = np.average(avg_color_per_row, axis=0)
+
+    if padding_type == "top_bottom":
+        if img.width > img.height:
+            new_height = img.width
+            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))
+            return newimg
+        else:
+            return img  # No change for already square images
+
+    elif padding_type == "left_right":
+        if img.height > img.width:
+            new_width = img.height
+            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))
+            return newimg
+        else:
+            return img  # No change for already square images
+            
 # Function for cropping and filling the image
-def fill_rectangle_cropper(img):
+def fill_rectangle_cropper1(img,padding):
     imgsz = [img.height, img.width]
     avg_color_per_row = np.average(img, axis=0)
     avg_color = np.average(avg_color_per_row, axis=0)
@@ -42,7 +77,16 @@ def fill_rectangle_cropper(img):
         return newimg
     else:
         return img
-
+        
+def on_submit():
+        padding_choice = padding_option.get()
+        if padding_choice == 1:
+            result_img = fill_rectangle_cropper(img, "top_bottom")
+        elif padding_choice == 2:
+            result_img = fill_rectangle_cropper(img, "left_right")
+        # Here you can save or display result_img
+        print("Padding applied based on selection!")
+    
 # Home Page
 @app.get("/", response_class=HTMLResponse)
 def home_page():
@@ -111,7 +155,7 @@ def demo_page():
     # Process the first image
     response = requests.get(url1)
     img1 = Image.open(BytesIO(response.content)).convert("RGB")
-    rectangled_img1 = fill_rectangle_cropper(img1)
+    rectangled_img1 = fill_rectangle_cropper(img1, "top_bottom")
     output1 = BytesIO()
     rectangled_img1.save(output1, format="JPEG")
     encoded_img1 = base64.b64encode(output1.getvalue()).decode("utf-8")
@@ -119,7 +163,7 @@ def demo_page():
     # Process the second image
     response = requests.get(url2)
     img2 = Image.open(BytesIO(response.content)).convert("RGB")
-    rectangled_img2 = fill_rectangle_cropper(img2)
+    rectangled_img2 =  fill_rectangle_cropper(img2, "left_right")
     output2 = BytesIO()
     rectangled_img2.save(output2, format="JPEG")
     encoded_img2 = base64.b64encode(output2.getvalue()).decode("utf-8")
@@ -157,8 +201,16 @@ def application_page():
     <h2>Rectangle Image Application</h2>
     <p>Upload a JPG image to rectangle and fill with color filler.</p>
     <form action="/upload/" enctype="multipart/form-data" method="post">
-        <input name="file" type="file">
-        <input type="submit" value="Rectangle It">
+    <label for="file">Upload your image:</label>
+    <input name="file" type="file" required><br><br>
+
+    <label>Choose the padding direction:</label><br>
+    <input type="radio" id="top_bottom" name="padding_type" value="top_bottom" checked>
+    <label for="top_bottom">Top/Bottom</label><br>
+    <input type="radio" id="left_right" name="padding_type" value="left_right">
+    <label for="left_right">Left/Right</label><br><br>
+
+    <input type="submit" value="Rectangle It">
     </form>
     <a href="/">Back</a>
     </body>