Add application file

Dockerfile

@@ -0,0 +1,14 @@
+# read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
+# you will also find guides on how best to write your Dockerfile
+
+FROM python:3.9
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . .
+
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,68 @@
+from fastapi import FastAPI, UploadFile, File,BackgroundTasks
+from rembg import remove
+
+from fastapi.responses import FileResponse
+import os
+from starlette.middleware.cors import CORSMiddleware
+# from fastapi.staticfiles import StaticFiles
+
+
+app = FastAPI()
+
+
+
+
+origins = [
+    "http://localhost:3000",
+]
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=origins,
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Directory to store processed images
+output_dir = "output"
+os.makedirs(output_dir, exist_ok=True)
+# app.mount("/processed_images", StaticFiles(directory=output_dir), name="processed_images")
+
+@app.post("/remove-background/")
+async def remove_background(file: UploadFile):
+    # Ensure the uploaded file is an image (you can add more validation)
+    if not file.content_type.startswith("image/"):
+        return {"error": "Invalid file format"}
+
+    # Read the uploaded image file
+    image_bytes = await file.read()
+
+    # Process the image to remove the background using Rembg
+    output_image_bytes = remove(image_bytes)
+
+    # Save the processed image to a file
+    output_filename = f"{output_dir}/{file.filename}"
+
+    # output_filename  = f"processed_images/{file.filename.split('/')[-1]}"
+    with open(output_filename, "wb") as output_file:
+        output_file.write(output_image_bytes)
+
+    return {"result": "Background removed successfully", "output_filename": output_filename}
+
+    
+
+@app.get("/download/{filename}")
+async def download_processed_image(filename: str):
+    file_path = os.path.join(output_dir, filename)
+    # file_path = f"/output/{filename}"
+    if not os.path.exists(file_path):
+        return {"error": "File not found"}
+
+    return FileResponse(file_path)
+
+
+@app.get("/delete")
+def deleteAll():
+    os.rmdir("output")
+

requirements.txt

@@ -0,0 +1,4 @@
+rembg
+fastapi
+os
+uvicorn