Update app.py

app.py

@@ -1,24 +1,29 @@
 import gradio as gr
 from rembg import remove
-import numpy as np
-from io import BytesIO
-from fastapi import FastAPI, Response
+from fastapi import FastAPI, UploadFile, File, HTTPException
+from fastapi.responses import Response
+import os
 
-# Create both Gradio and FastAPI apps
+# Create apps
+app = FastAPI()
 gradio_app = gr.Interface(
     fn=lambda img: remove(img),
-    inputs=gr.Image(type="numpy"),
-    outputs=gr.Image(type="numpy"),
+    inputs=gr.Image(type="filepath"),
+    outputs=gr.Image(type="filepath"),
     title="Background Remover",
     description="Upload an image to remove the background"
 )
 
-app = FastAPI()
-
 @app.post("/api/remove-bg")
-async def api_remove_bg(file: bytes = File(...)):
-    output_bytes = remove(file)
-    return Response(content=output_bytes, media_type="image/png")
+async def api_remove_bg(file: UploadFile = File(...)):
+    if not file.filename.lower().endswith(('.png', '.jpg', '.jpeg')):
+        raise HTTPException(400, "Only PNG/JPG images allowed")
+    
+    image_bytes = await file.read()
+    try:
+        output = remove(image_bytes)
+        return Response(content=output, media_type="image/png")
+    except Exception as e:
+        raise HTTPException(500, f"Error processing image: {str(e)}")
 
-# Mount Gradio to the root path
 app = gr.mount_gradio_app(app, gradio_app, path="/")