Update app.py

app.py

@@ -1,29 +1,79 @@
 import gradio as gr
 from rembg import remove
-from fastapi import FastAPI, UploadFile, File, HTTPException
-from fastapi.responses import Response
+from fastapi import FastAPI
+from fastapi.staticfiles import StaticFiles
+import os
 
-# Create apps
+# 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
 gradio_app = gr.Interface(
-    fn=lambda img: remove(img),
-    inputs=gr.Image(type="filepath"),
-    outputs=gr.Image(type="filepath"),
+    fn=remove_bg,
+    inputs=gr.Image(type="filepath", label="Input Image"),
+    outputs=gr.Image(type="filepath", label="Output Image"),
     title="Background Remover",
     description="Upload an image to remove the background",
-    allow_flagging="never"  # Disable flagging feature
+    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(...)):
-    if not file.filename.lower().endswith(('.png', '.jpg', '.jpeg')):
-        raise HTTPException(400, "Only PNG/JPG images allowed")
-    
-    image_bytes = await file.read()
+async def api_remove_bg(file: bytes = File(...)):
     try:
-        output = remove(image_bytes)
-        return Response(content=output, media_type="image/png")
+        output_bytes = remove(file)
+        return Response(content=output_bytes, media_type="image/png")
     except Exception as e:
-        raise HTTPException(500, f"Error processing image: {str(e)}")
+        raise HTTPException(status_code=500, detail=str(e))import gradio as gr
+from rembg import remove
+from fastapi import FastAPI
+from fastapi.staticfiles import StaticFiles
+import os
+
+# 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
+gradio_app = gr.Interface(
+    fn=remove_bg,
+    inputs=gr.Image(type="filepath", label="Input Image"),
+    outputs=gr.Image(type="filepath", label="Output Image"),
+    title="Background Remover",
+    description="Upload an image to remove the background",
+    allow_flagging="never"
+)
 
-app = gr.mount_gradio_app(app, gradio_app, path="/")
+# Mount Gradio app
+app.mount("/", gradio_app)
+
+# API endpoint
+@app.post("/api/remove-bg")
+async def api_remove_bg(file: bytes = File(...)):
+    try:
+        output_bytes = remove(file)
+        return Response(content=output_bytes, media_type="image/png")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))