Update app.py

app.py

@@ -1,24 +1,30 @@
-
-import gradio as gr
+from fastapi import FastAPI, File, UploadFile, HTTPException, Header
+from fastapi.responses import StreamingResponse
 from rembg import remove
 from PIL import Image
 import io
 
-def remove_background(input_image):
-    if isinstance(input_image, str):
-        input_image = Image.open(input_image)
+app = FastAPI()
+
+# Replace this with your secret
+API_KEY = "your_super_secret_key"
+
+@app.post("/remove-bg")
+async def remove_bg(file: UploadFile = File(...), authorization: str = Header(None)):
+    # Check API key in Authorization header
+    if authorization != f"Bearer {API_KEY}":
+        raise HTTPException(status_code=401, detail="Unauthorized")
+
+    # Read image
+    contents = await file.read()
+    input_image = Image.open(io.BytesIO(contents))
+
+    # Remove background
     output_image = remove(input_image)
-    return output_image
 
-demo = gr.Interface(
-    fn=remove_background,
-    inputs=gr.Image(type="pil"),
-    outputs=gr.Image(type="pil"),
-    title="Background Remover",
-    description="Upload an image to remove its background",
-    examples=[],
-    theme=gr.themes.Soft()
-)
+    # Prepare image response
+    img_byte_arr = io.BytesIO()
+    output_image.save(img_byte_arr, format="PNG")
+    img_byte_arr.seek(0)
 
-if __name__ == "__main__":
-    demo.launch()
+    return StreamingResponse(img_byte_arr, media_type="image/png")