Spaces:
Sleeping
Sleeping
File size: 5,301 Bytes
2974de7 68e9567 2974de7 c3f998c 2974de7 c3f998c 3607789 2974de7 68e9567 2974de7 c3f998c 2974de7 495811a 2974de7 68e9567 2974de7 68e9567 2974de7 b02f32f c3f998c cf08216 c3f998c 68e9567 c3f998c 2974de7 ce37540 2974de7 cf08216 2974de7 c3f998c 2974de7 c3f998c fe496f4 c3f998c fe496f4 5964a73 fe496f4 2974de7 5964a73 fe496f4 5964a73 68e9567 5964a73 358554d 5964a73 358554d b02f32f 5964a73 fe496f4 2974de7 495811a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 |
from fastapi import FastAPI, File, UploadFile
from fastapi.responses import HTMLResponse
from fastapi.staticfiles import StaticFiles
import numpy as np
from PIL import Image
from io import BytesIO
import requests
import base64
import os
app = FastAPI()
# Mount the static folder for CSS and other assets
app.mount("/static", StaticFiles(directory="static"), name="static")
# Function for cropping and filling the image
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
# Home Page
@app.get("/", response_class=HTMLResponse)
def home_page():
return """
<html>
<head>
<link rel="stylesheet" href="/static/styles/style.css">
</head>
<body>
<img src="/static/images/banner.jpg" alt="Banner" width="100%">
<h2>Square and Fill Image App</h2>
<p>Please select an option below:</p>
<ul>
<li><a href="/demo">Demo</a></li>
<li><a href="/application">Application</a></li>
</ul>
</body>
</html>
"""
# Demo Page
@app.get("/demo", response_class=HTMLResponse)
def demo_page():
# URLs for demo images
url1 = "https://raw.githubusercontent.com/webdevserv/images_video/main/cowportrait.jpg"
url2 = "https://raw.githubusercontent.com/webdevserv/images_video/main/cowlandscape.jpg"
# Process the first image
response = requests.get(url1)
img1 = Image.open(BytesIO(response.content)).convert("RGB")
squared_img1 = fill_square_cropper(img1)
output1 = BytesIO()
squared_img1.save(output1, format="JPEG")
encoded_img1 = base64.b64encode(output1.getvalue()).decode("utf-8")
# Process the second image
response = requests.get(url2)
img2 = Image.open(BytesIO(response.content)).convert("RGB")
squared_img2 = fill_square_cropper(img2)
output2 = BytesIO()
squared_img2.save(output2, format="JPEG")
encoded_img2 = base64.b64encode(output2.getvalue()).decode("utf-8")
return f"""
<html>
<body>
<h2>Square Image Demo</h2>
<p>Image will be squared with color filler where applicable.</p>
<h3>Result 1:</h3>
<img src="data:image/jpeg;base64,{encoded_img1}" />
<h3>Result 2:</h3>
<img src="data:image/jpeg;base64,{encoded_img2}" />
<p><a href="/">Back</a></p>
</body>
</html>
"""
# Application Page
@app.get("/application", response_class=HTMLResponse)
def application_page():
return """
<html>
<body>
<img src="/static/images/banner.jpg" alt="Banner" width="100%">
<h2>Square Image Application</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" value="Square It">
</form>
<a href="/">Back</a>
</body>
</html>
"""
@app.post("/upload/")
async def upload_file(file: UploadFile = File(...)):
try:
# Await file upload
contents = await file.read()
img = Image.open(BytesIO(contents)).convert("RGB")
squared_img = fill_square_cropper(img)
# Save the squared image (original size)
output = BytesIO()
squared_img.save(output, format="JPEG")
output.seek(0)
# Encode the full-size image for download
full_size_encoded_img = base64.b64encode(output.getvalue()).decode("utf-8")
# Resize the image for display (512px by 512px)
display_img = squared_img.copy()
display_img.thumbnail((512, 512)) # Resize for display
display_output = BytesIO()
display_img.save(display_output, format="JPEG")
display_output.seek(0)
# Encode the resized display image
display_encoded_img = base64.b64encode(display_output.getvalue()).decode("utf-8")
# Return the HTML response
return HTMLResponse(
content=f"""
<img src="/static/images/banner.jpg" alt="Banner" width="100%">
<h2>Image successfully squared!</h2>
<img src='data:image/jpeg;base64,{display_encoded_img}' width="512" height="512" />
<br/>
<a href="data:image/jpeg;base64,{full_size_encoded_img}" download="squared_image.jpg">
View Full-Size Image
</a>
""",
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))) |