| from fastapi import FastAPI, File, UploadFile |
| from fastapi.responses import HTMLResponse |
| from PIL import Image |
| import numpy as np |
| from io import BytesIO |
| import os |
|
|
| app = FastAPI() |
|
|
| |
| def fill_square_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) |
|
|
| if img.height > img.width: |
| newimg = Image.new( |
| 'RGB', |
| (img.height, img.height), |
| (round(avg_color[0]), round(avg_color[1]), round(avg_color[2])) |
| ) |
| newpos = (img.height - img.width) // 2 |
| newimg.paste(img, (newpos, 0)) |
| return newimg |
|
|
| elif img.width > img.height: |
| newimg = Image.new( |
| 'RGB', |
| (img.width, img.width), |
| (round(avg_color[0]), round(avg_color[1]), round(avg_color[2])) |
| ) |
| newpos = (img.width - img.height) // 2 |
| newimg.paste(img, (0, newpos)) |
| return newimg |
| else: |
| return img |
|
|
| @app.get("/", response_class=HTMLResponse) |
| def home_page(): |
| return """ |
| <html> |
| <body> |
| <h2>Square and Fill Image App</h2> |
| <p>Upload a JPG image to square and fill with color filler.</p> |
| <form action="/upload/" enctype="multipart/form-data" method="post"> |
| <input name="file" type="file"> |
| <input type="submit"> |
| </form> |
| </body> |
| </html> |
| """ |
| @app.post("/upload/") |
| async def upload_file(file: UploadFile = File(...)): |
| try: |
| |
| contents = await file.read() |
| img = Image.open(BytesIO(contents)).convert("RGB") |
| squared_img = fill_square_cropper(img) |
|
|
| |
| output = BytesIO() |
| squared_img.save(output, format="JPEG") |
| output.seek(0) |
|
|
| |
| import base64 |
| encoded_img = base64.b64encode(output.getvalue()).decode("utf-8") |
| return HTMLResponse( |
| content=f"<h3>Image successfully squared!</h3><img src='data:image/jpeg;base64,{encoded_img}' />", |
| media_type="text/html" |
| ) |
| except Exception as e: |
| return HTMLResponse(content=f"<h3>An error occurred: {e}</h3>", media_type="text/html") |
|
|
|
|
| if __name__ == "__main__": |
| import uvicorn |
| uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", 7860))) |
|
|
