RectangleIt

Sleeping

AItool commited on Mar 26

Commit

260965a

verified ·

1 Parent(s): ccf8eb8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -228,10 +228,10 @@ async def upload_file(file: UploadFile = File(...), padding_type: str = Form(...
         # Resize the image for display (512px by 512px)
         display_img = rectangled_img.copy()
-        if (padding_type=="top_bottom"):
-            display_img.thumbnail((512, *))  # Resize for display
-        else
-             display_img.thumbnail((*, 512))  # Resize for display
         display_output = BytesIO()
         display_img.save(display_output, format="JPEG")
         display_output.seek(0)
@@ -250,7 +250,8 @@ async def upload_file(file: UploadFile = File(...), padding_type: str = Form(...
             <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}'/>
             <p><a href="data:image/jpeg;base64,{full_size_encoded_img}" download="rectangled_image.jpg">
             Download Full-Size Image</a></p>
             <p><a href="/">Back</a></p>

         # 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)
+        display_img.thumbnail((desired_width, desired_height))
         display_output = BytesIO()
         display_img.save(display_output, format="JPEG")
         display_output.seek(0)
             <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_img}'/>
             <p><a href="data:image/jpeg;base64,{full_size_encoded_img}" download="rectangled_image.jpg">
             Download Full-Size Image</a></p>
             <p><a href="/">Back</a></p>