Update app.py

app.py

@@ -58,44 +58,44 @@ async def root():
 
 # FIXED: Accept both UploadFile and raw bytes
 @app.post("/remove-bg")
-async def remove_background(request: Request, file: UploadFile = File(None)):
+async def remove_background(
+    file: UploadFile = File(None),
+    request: Request = None
+):
     try:
-        # Handle different content types
-        if file is None:
-            # Fallback: read raw body if no file param
-            body = await request.body()
-            if not body:
-                raise HTTPException(400, detail="No image data received")
-            image = Image.open(io.BytesIO(body))
-        else:
+        contents = None
+        
+        # Try UploadFile first
+        if file:
             contents = await file.read()
-            if not contents:
-                raise HTTPException(400, detail="Empty file")
-            image = Image.open(io.BytesIO(contents))
+            print(f"✅ File received: {file.filename}, {len(contents)} bytes")
+        else:
+            # Fallback: check raw body
+            body = await request.body()
+            contents = body
+            print(f"✅ Raw body received: {len(contents)} bytes")
+        
+        if not contents or len(contents) == 0:
+            raise HTTPException(400, detail="No file uploaded")
         
-        # Validate image
-        if not image.format or image.format not in ["JPEG", "PNG", "JPG"]:
-            raise HTTPException(400, detail="Invalid image format")
+        image = Image.open(io.BytesIO(contents))
+        print(f"✅ Image loaded: {image.size}, format: {image.format}")
         
-        # Process
         result = remove_bg(image)
         
-        # Return PNG
         img_byte_arr = io.BytesIO()
         result.save(img_byte_arr, format="PNG")
         img_byte_arr.seek(0)
         
+        print("✅ Processing complete")
         return StreamingResponse(
             img_byte_arr,
             media_type="image/png",
-            headers={
-                "Content-Disposition": "inline; filename=removed-bg.png",
-                "Cache-Control": "public, max-age=3600"
-            }
+            headers={"Content-Disposition": "inline; filename=removed-bg.png"}
         )
     except Exception as e:
-        print(f"❌ Error: {str(e)}")
-        raise HTTPException(500, detail=f"Processing failed: {str(e)}")
+        print(f"❌ Detailed error: {str(e)}")
+        raise HTTPException(400, detail=f"Processing failed: {str(e)}")
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)