Update app.py

app.py

@@ -49,20 +49,29 @@ def home_page():
     </body>
     </html>
     """
-
 @app.post("/upload/")
-def upload_file(file: UploadFile = File(...)):
-    contents = file.read()
-    img = Image.open(BytesIO(contents)).convert("RGB")
-    squared_img = fill_square_cropper(img)
+async def upload_file(file: UploadFile = File(...)):  # Make the function asynchronous
+    try:
+        # Await the read method
+        contents = await file.read()  
+        img = Image.open(BytesIO(contents)).convert("RGB")
+        squared_img = fill_square_cropper(img)
+
+        # Save the squared image
+        output = BytesIO()
+        squared_img.save(output, format="JPEG")
+        output.seek(0)
 
-    # Save the squared image
-    output = BytesIO()
-    squared_img.save(output, format="JPEG")
-    output.seek(0)
+        # Return base64-encoded image
+        import base64
+        encoded_img = base64.b64encode(output.getvalue()).decode("utf-8")
+        return HTMLResponse(
+            content=f"<h3>Image successfully squared!</h3><img src='data:image/jpeg;base64,{encoded_img}' />",
+            media_type="text/html"
+        )
+    except Exception as e:
+        return HTMLResponse(content=f"<h3>An error occurred: {e}</h3>", media_type="text/html")
 
-    return HTMLResponse(content=f"<h3>Image successfully squared!</h3><img src='data:image/jpeg;base64,{output.getvalue().hex()}' />", media_type="text/html")
-    
 
 if __name__ == "__main__":
     import uvicorn