feat: update remove-image-background route

app.py

@@ -294,7 +294,7 @@ def split_image(image: Image.Image) -> List[Image.Image]:
     return [preprocess_image(image) for image in images]
 
 from fastapi import FastAPI, File, UploadFile, HTTPException
-from fastapi.responses import JSONResponse, FileResponse
+from fastapi.responses import JSONResponse, FileResponse, StreamingResponse
 import tempfile
 
 app = FastAPI()
@@ -338,7 +338,7 @@ async def generate_3d(image: UploadFile = File(...)):
         raise HTTPException(status_code=500, detail=str(e))
 
 @app.post("/remove-image-background")
-async def remove_background_image(image: UploadFile = File(...)):
+async def remove_image_background(image: UploadFile = File(...)):
     if not image:
         raise HTTPException(status_code=400, detail="No image provided")
 
@@ -346,14 +346,18 @@ async def remove_background_image(image: UploadFile = File(...)):
         image_data = Image.open(image.file)
         image_data = image_data.convert("RGBA")
         image_data = preprocess_image(image_data)
-
         buffer = io.BytesIO()
-        image.save(buffer, format="PNG")
+        image_data.save(buffer, format="PNG")
         buffer.seek(0)
 
-        return FileResponse(buffer, media_type='application/octet-stream', filename="image.png")
+        return StreamingResponse(
+            content=buffer,
+            media_type='image/png',
+            headers={"Content-Disposition": "inline; filename=image.png"}
+        )
 
     except Exception as e:
+        print(e)
         raise HTTPException(status_code=500, detail=str(e))