Upload 3 files

Dockerfile

@@ -0,0 +1,26 @@
+FROM python:3.10-slim
+
+# Install basic system dependencies
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    build-essential \
+    && rm -rf /var/lib/apt/clean/var/lib/apt/lists/*
+
+# Set up user permissions using Debian standard useradd
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
+WORKDIR $HOME/app
+
+# Copy python dependencies and install
+COPY --chown=user requirements.txt .
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+# Copy server code
+COPY --chown=user . .
+
+# Hugging Face Spaces routes traffic natively through port 7860
+EXPOSE 7860
+ENV PORT=7860
+
+CMD ["uvicorn", "server:app", "--host", "0.0.0.0", "--port", "7860"]

requirements.txt

@@ -0,0 +1,5 @@
+fastapi
+uvicorn
+rembg
+Pillow
+python-multipart

server.py

@@ -0,0 +1,38 @@
+from fastapi import FastAPI, UploadFile, File, Response
+from fastapi.middleware.cors import CORSMiddleware
+from PIL import Image
+from rembg import remove
+import io
+import os
+import uvicorn
+
+app = FastAPI()
+
+# Enable CORS configurations to allow the internal node proxy to make pipeline hops
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+@app.post("/remove-bg")
+async def remove_background(file: UploadFile = File(...)):
+    """
+    Removes the background from an uploaded image file.
+    Expects a multipart/form-data upload with an image file.
+    Returns a PNG image with a transparent background.
+    """
+    contents = await file.read()
+    input_image = Image.open(io.BytesIO(contents)).convert("RGBA")
+    output_image = remove(input_image)
+
+    output_buffer = io.BytesIO()
+    output_image.save(output_buffer, format="PNG")
+    return Response(content=output_buffer.getvalue(), media_type="image/png")
+
+
+if __name__ == "__main__":
+    port = int(os.environ.get("PORT", 8080))
+    uvicorn.run("server:app", host="0.0.0.0", port=port)