ReSizeImage

Sleeping

App Files Files Community

AItool commited on Oct 15

Commit

881f90b

verified ·

1 Parent(s): af17c05

Update app.py

Browse files

Files changed (1) hide show

app.py +16 -22

app.py CHANGED Viewed

@@ -183,26 +183,30 @@ def application_page():
     </body>
     </html>
     """
 @app.post("/upload/")
-async def upload_file(file: UploadFile = File(...), padding_type: str = Form(...)):
     try:
         # Await file upload
         contents = await file.read()
         img = Image.open(BytesIO(contents)).convert("RGB")
-        # Apply padding based on user's choice
-        rectangled_img = cropper(img,padding_type)
-        # Save the rectangle image (original size)
         output = BytesIO()
-        rectangled_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 = rectangled_img.copy()
         desired_width = 512
         aspect_ratio = display_img.height / display_img.width
         desired_height = int(desired_width * aspect_ratio)
@@ -224,21 +228,11 @@ async def upload_file(file: UploadFile = File(...), padding_type: str = Form(...
             </head>
             <body>
             <img class="banner" src="/static/images/banner.jpg" alt="Banner" width="100%">
-            <h2>Image successfully rectangled!</h2>
-            <!--<img src='data:image/jpeg;base64,display_encoded_img'/>-->
             <img src='data:image/jpeg;base64,{display_encoded_img}' width="512"/>
-            <p><a href="data:image/jpeg;base64,{full_size_encoded_img}" download="rectangled_image.jpg">
             Download Full-Size Image</a></p>
-            <div style="margin: 0.75em 0;"><a href="https://www.buymeacoffee.com/Artgen" target="_blank"><img src="https://cdn.buymeacoffee.com/buttons/default-orange.png" alt="Buy Me A Coffee" height="41" width="174"></a></div>
-            <div style="margin: 0.75em 0;">But what would really help me is a <strong>PRO subscription</strong> to Google Colab, Kaggle or Hugging Face. Many thanks.</div>
             <p><a href="/">Back</a></p>
-            <div id="credit">Image credit
-            <a href="https://stock.adobe.com/es/contributor/212598146/UMAMI%20LAB" target="_blank">Adobe Stock User Umami Lab</a>
-            and
-            <a href="https://www.shutterstock.com/g/Idoia+Lerchundi?rid=430751957" target="_blank">Shutterstock User PhoArt101</a>.
-            </div>
-            <div style="margin: 0.75em 0;"><a href="https://www.buymeacoffee.com/Artgen" target="_blank"><img src="https://cdn.buymeacoffee.com/buttons/default-orange.png" alt="Buy Me A Coffee" height="41" width="174"></a></div>
-            <div style="margin: 0.75em 0;">But what would really help me is a <strong>PRO subscription</strong> to Google Colab, Kaggle or Hugging Face. Many thanks.</div>
             </body>
             </html>
             """,
@@ -248,5 +242,5 @@ async def upload_file(file: UploadFile = File(...), padding_type: str = Form(...
         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)))

     </body>
     </html>
     """
 @app.post("/upload/")
+async def upload_file(
+    file: UploadFile = File(...),
+    target_width: int = Form(...),
+    target_height: int = Form(...)
+):
     try:
         # Await file upload
         contents = await file.read()
         img = Image.open(BytesIO(contents)).convert("RGB")
+        # Crop to the user’s chosen dimensions (cropper clamps to max 1024px)
+        cropped_img = cropper(img, target_width, target_height)
+        # Save the cropped image (original size)
         output = BytesIO()
+        cropped_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 wide, keep aspect ratio)
+        display_img = cropped_img.copy()
         desired_width = 512
         aspect_ratio = display_img.height / display_img.width
         desired_height = int(desired_width * aspect_ratio)
             </head>
             <body>
             <img class="banner" src="/static/images/banner.jpg" alt="Banner" width="100%">
+            <h2>Image successfully cropped!</h2>
             <img src='data:image/jpeg;base64,{display_encoded_img}' width="512"/>
+            <p><a href="data:image/jpeg;base64,{full_size_encoded_img}" download="cropped_image.jpg">
             Download Full-Size Image</a></p>
             <p><a href="/">Back</a></p>
             </body>
             </html>
             """,
         return HTMLResponse(content=f"<h3>An error occurred: {e}</h3>", media_type="text/html")
 if __name__ == "__main__":
+    import uvicorn, os
+    uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))