Merge pull request #1 from syahkty/codex/explain-codebase-structure-and-essentials

README.md

@@ -7,4 +7,17 @@ sdk: docker
 pinned: false
 ---
 
+Service ini menjalankan FastAPI yang memanfaatkan model Hugging Face untuk
+menghapus background gambar.
+
+## Endpoint
+
+`POST /removebg` dengan field multipart `image` akan mengembalikan hasil dalam
+format `image/png` dengan background transparan.
+
+## Konfigurasi
+
+- `MODEL_ID` (opsional): ID model di Hub, default `briaai/RMBG-1.4`
+- `HF_TOKEN` (opsional): token akses jika model memerlukan autentikasi
+
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,12 +1,12 @@
+import io
 import os
 from fastapi import FastAPI, UploadFile, File, Response
-import httpx
+from huggingface_hub import AsyncInferenceClient
 
-# Endpoint Space tujuan (bisa dioverride via Space variable TARGET_URL)
-TARGET_URL = os.getenv(
-    "TARGET_URL",
-    "https://not-lain-background-removal.hf.space/image"
-)
+# Model Hugging Face untuk penghapusan background
+MODEL_ID = os.getenv("MODEL_ID", "briaai/RMBG-1.4")
+HF_TOKEN = os.getenv("HF_TOKEN")
+client = AsyncInferenceClient(model=MODEL_ID, token=HF_TOKEN)
 
 app = FastAPI(title="Background Removal Proxy")
 
@@ -15,7 +15,7 @@ async def root():
     return {
         "ok": True,
         "message": "POST /removebg with multipart field 'image' → returns image/png",
-        "target": TARGET_URL,
+        "model": MODEL_ID,
     }
 
 @app.get("/healthz")
@@ -24,20 +24,8 @@ async def health():
 
 @app.post("/removebg")
 async def removebg(image: UploadFile = File(...)):
-    timeout = httpx.Timeout(60.0, connect=20.0)
-    async with httpx.AsyncClient(timeout=timeout, follow_redirects=True) as client:
-        files = {
-            # Space target menerima field bernama 'image'
-            "image": (image.filename, await image.read(), image.content_type or "image/jpeg")
-        }
-
-        # retry ringan untuk cold start/gateway
-        for attempt in range(2):
-            r = await client.post(TARGET_URL, files=files)
-            if r.status_code == 200:
-                return Response(content=r.content, media_type="image/png")
-            if r.status_code in (502, 503, 504) and attempt == 0:
-                continue
-            return Response(content=r.text, media_type="application/json", status_code=r.status_code)
-
-    return Response(content='{"detail":"unknown error"}', media_type="application/json", status_code=500)
+    img_bytes = await image.read()
+    result = await client.image_to_image(img_bytes, model=MODEL_ID)
+    buf = io.BytesIO()
+    result.save(buf, format="PNG")
+    return Response(content=buf.getvalue(), media_type="image/png")

requirements.txt

@@ -2,3 +2,5 @@ fastapi==0.110.3
 uvicorn==0.30.1
 httpx==0.27.2
 python-multipart==0.0.9
+huggingface-hub==0.34.4
+Pillow==11.3.0