Update app.py

app.py

@@ -115,15 +115,34 @@ async def upload_file(file: UploadFile = File(...)):
         img = Image.open(BytesIO(contents)).convert("RGB")
         squared_img = fill_square_cropper(img)
 
-        # Save the squared image
+        # Save the squared image (original size)
         output = BytesIO()
         squared_img.save(output, format="JPEG")
         output.seek(0)
 
-        # Return base64-encoded image
-        encoded_img = base64.b64encode(output.getvalue()).decode("utf-8")
+        # 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"<h3>Image successfully squared!</h3><img src='data:image/jpeg;base64,{encoded_img}' />",
+            content=f"""
+            <h3>Image successfully squared!</h3>
+            <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">
+                <button>Download Full-Size Image</button>
+            </a>
+            """,
             media_type="text/html"
         )
     except Exception as e: