Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -110,7 +110,7 @@ def demo_page():
|
|
| 110 |
# Process the first image
|
| 111 |
response = requests.get(url1)
|
| 112 |
img1 = Image.open(BytesIO(response.content)).convert("RGB")
|
| 113 |
-
rectangled_img1 =
|
| 114 |
output1 = BytesIO()
|
| 115 |
rectangled_img1.save(output1, format="JPEG")
|
| 116 |
encoded_img1 = base64.b64encode(output1.getvalue()).decode("utf-8")
|
|
@@ -118,7 +118,7 @@ def demo_page():
|
|
| 118 |
# Process the second image
|
| 119 |
response = requests.get(url2)
|
| 120 |
img2 = Image.open(BytesIO(response.content)).convert("RGB")
|
| 121 |
-
rectangled_img2 =
|
| 122 |
output2 = BytesIO()
|
| 123 |
rectangled_img2.save(output2, format="JPEG")
|
| 124 |
encoded_img2 = base64.b64encode(output2.getvalue()).decode("utf-8")
|
|
@@ -182,7 +182,7 @@ async def upload_file(file: UploadFile = File(...)):
|
|
| 182 |
|
| 183 |
# Resize the image for display (512px by 512px)
|
| 184 |
display_img = rectangled_img.copy()
|
| 185 |
-
display_img.thumbnail((512,
|
| 186 |
display_output = BytesIO()
|
| 187 |
display_img.save(display_output, format="JPEG")
|
| 188 |
display_output.seek(0)
|
|
@@ -201,7 +201,7 @@ async def upload_file(file: UploadFile = File(...)):
|
|
| 201 |
<body>
|
| 202 |
<img class="banner" src="/static/images/banner.jpg" alt="Banner" width="100%">
|
| 203 |
<h2>Image successfully rectangled!</h2>
|
| 204 |
-
<img src='data:image/jpeg;base64,{display_encoded_img}' width="512"
|
| 205 |
<p><a href="data:image/jpeg;base64,{full_size_encoded_img}" download="rectangled_image.jpg">
|
| 206 |
Download Full-Size Image</a></p>
|
| 207 |
<p><a href="/">Back</a></p>
|
|
|
|
| 110 |
# Process the first image
|
| 111 |
response = requests.get(url1)
|
| 112 |
img1 = Image.open(BytesIO(response.content)).convert("RGB")
|
| 113 |
+
rectangled_img1 = fill_rectangle_cropper(img1)
|
| 114 |
output1 = BytesIO()
|
| 115 |
rectangled_img1.save(output1, format="JPEG")
|
| 116 |
encoded_img1 = base64.b64encode(output1.getvalue()).decode("utf-8")
|
|
|
|
| 118 |
# Process the second image
|
| 119 |
response = requests.get(url2)
|
| 120 |
img2 = Image.open(BytesIO(response.content)).convert("RGB")
|
| 121 |
+
rectangled_img2 = fill_rectangle_cropper(img2)
|
| 122 |
output2 = BytesIO()
|
| 123 |
rectangled_img2.save(output2, format="JPEG")
|
| 124 |
encoded_img2 = base64.b64encode(output2.getvalue()).decode("utf-8")
|
|
|
|
| 182 |
|
| 183 |
# Resize the image for display (512px by 512px)
|
| 184 |
display_img = rectangled_img.copy()
|
| 185 |
+
display_img.thumbnail((512, 206)) # Resize for display
|
| 186 |
display_output = BytesIO()
|
| 187 |
display_img.save(display_output, format="JPEG")
|
| 188 |
display_output.seek(0)
|
|
|
|
| 201 |
<body>
|
| 202 |
<img class="banner" src="/static/images/banner.jpg" alt="Banner" width="100%">
|
| 203 |
<h2>Image successfully rectangled!</h2>
|
| 204 |
+
<img src='data:image/jpeg;base64,{display_encoded_img}' width="512" />
|
| 205 |
<p><a href="data:image/jpeg;base64,{full_size_encoded_img}" download="rectangled_image.jpg">
|
| 206 |
Download Full-Size Image</a></p>
|
| 207 |
<p><a href="/">Back</a></p>
|