Update app.py

app.py

@@ -1,12 +1,17 @@
-import io, requests, json
-import gradio as gr
+import io
+import requests
+from fastapi import FastAPI, UploadFile, File, Query
+from fastapi.responses import JSONResponse
 from rembg import remove
 from PIL import Image
+import uvicorn
 
-# -----------------
-# Core background remover
-# -----------------
-def process_image_bytes(img_bytes: bytes):
+app = FastAPI(title="AI ClearCut - Background Remover")
+
+# --------------------------
+# Helper: process image bytes
+# --------------------------
+def process_image(img_bytes: bytes):
     # Remove background
     output_bytes = remove(
         img_bytes,
@@ -16,77 +21,43 @@ def process_image_bytes(img_bytes: bytes):
         alpha_matting_erode_size=10
     )
 
-    # Upload to ar-media hosting
+    # Upload to ar-hosting
     files = {"file": ("output.png", output_bytes, "image/png")}
     r = requests.post("https://ar-hosting.pages.dev/upload", files=files)
     r.raise_for_status()
     data = r.json()
 
     if "url" in data:
-        return output_bytes, data["url"]
-
-    raise Exception("Upload failed")
-
-# -----------------
-# Gradio function (UI)
-# -----------------
-def remove_and_upload(image: Image.Image):
+        return {"success": True, "url": data["url"]}
+    return {"error": "Upload failed"}
+
+# --------------------------
+# Endpoint 1: Upload file
+# --------------------------
+@app.post("/upload")
+async def upload_file(file: UploadFile = File(...)):
     try:
-        with io.BytesIO() as buffered:
-            image.save(buffered, format="PNG", optimize=True)
-            img_bytes = buffered.getvalue()
-
-        output_bytes, hosted_url = process_image_bytes(img_bytes)
-
-        return (
-            Image.open(io.BytesIO(output_bytes)).convert("RGBA"),
-            json.dumps({"success": True, "url": hosted_url}, indent=2)
-        )
-
+        img_bytes = await file.read()
+        result = process_image(img_bytes)
+        return JSONResponse(content=result)
     except Exception as e:
-        return None, json.dumps({"error": str(e)}, indent=2)
+        return JSONResponse(content={"error": str(e)})
 
-# -----------------
-# API-friendly function (?url=)
-# -----------------
-def api_json(url: str):
+# --------------------------
+# Endpoint 2: From image URL
+# --------------------------
+@app.get("/json")
+async def from_url(url: str = Query(..., description="Image URL")):
     try:
-        resp = requests.get(url, timeout=10)
+        resp = requests.get(url, timeout=15)
         resp.raise_for_status()
-        img_bytes = resp.content
-
-        _, hosted_url = process_image_bytes(img_bytes)
-
-        return {"success": True, "url": hosted_url}
+        result = process_image(resp.content)
+        return JSONResponse(content=result)
     except Exception as e:
-        return {"error": str(e)}
-
-# -----------------
-# Gradio UI
-# -----------------
-examples = [
-    ["https://upload.wikimedia.org/wikipedia/commons/1/15/Cat_August_2010-4.jpg"],
-    ["https://upload.wikimedia.org/wikipedia/commons/4/4d/Cat_November_2010-1a.jpg"]
-]
-
-with gr.Blocks(title="AI ClearCut - Background Remover") as demo:
-    gr.Markdown("# <center>AI ClearCut</center>\n### <center>Background Removal + JSON API Output</center>")
-
-    with gr.Row():
-        with gr.Column():
-            input_image = gr.Image(label="Upload Image", type="pil")
-            submit_btn = gr.Button("Remove Background", variant="primary")
-            gr.Examples(examples=examples, inputs=[input_image], label="Examples")
-
-        with gr.Column():
-            output_image = gr.Image(label="Result Image", type="pil")
-            output_json = gr.Textbox(label="JSON Output", lines=6, interactive=False)
-
-    # UI button → normal background removal
-    submit_btn.click(fn=remove_and_upload, inputs=[input_image], outputs=[output_image, output_json])
-
-    # API endpoint for ?url=
-    demo.load(fn=lambda url: api_json(url), inputs=[gr.Textbox(label="Image URL")], outputs="json", api_name="json")
+        return JSONResponse(content={"error": str(e)})
 
-# ✅ Launch for Hugging Face
-demo.launch()
+# --------------------------
+# HF Entry point
+# --------------------------
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)