Update app.py

app.py

@@ -4,38 +4,38 @@ from fastapi import FastAPI, UploadFile, File, HTTPException
 from fastapi.responses import Response
 import os
 
+# Create background removal function
+def remove_background(image_bytes):
+    try:
+        return remove(image_bytes)
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+# Create Gradio interface
+def gradio_interface(input_image):
+    if input_image is None:
+        return None
+    return remove_background(input_image)
+
 # Create FastAPI app
 app = FastAPI()
 
-# Background removal function
-def remove_bg(image_path):
-    with open(image_path, "rb") as f:
-        input_image = f.read()
-    output_image = remove(input_image)
-    output_path = "/tmp/output.png"
-    with open(output_path, "wb") as f:
-        f.write(output_image)
-    return output_path
-
-# Create Gradio interface
+# Create and mount Gradio app
 gradio_app = gr.Interface(
-    fn=remove_bg,
-    inputs=gr.Image(type="filepath", label="Input Image"),
-    outputs=gr.Image(type="filepath", label="Output Image"),
+    fn=gradio_interface,
+    inputs=gr.Image(type="filepath"),
+    outputs=gr.Image(type="filepath"),
     title="Background Remover",
     description="Upload an image to remove the background",
     allow_flagging="never"
 )
 
-# Mount Gradio app
-app.mount("/", gradio_app)
-
 # API endpoint
 @app.post("/api/remove-bg")
 async def api_remove_bg(file: UploadFile = File(...)):
-    try:
-        input_image = await file.read()
-        output_bytes = remove(input_image)
-        return Response(content=output_bytes, media_type="image/png")
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+    image_bytes = await file.read()
+    output_bytes = remove_background(image_bytes)
+    return Response(content=output_bytes, media_type="image/png")
+
+# Mount Gradio app
+app = gr.mount_gradio_app(app, gradio_app, path="/")