initial code

initial code push

.dockerignore

@@ -0,0 +1,27 @@
+**/__pycache__
+**/.venv
+**/.classpath
+**/.dockerignore
+**/.env
+**/.git
+**/.gitignore
+**/.project
+**/.settings
+**/.toolstarget
+**/.vs
+**/.vscode
+**/*.*proj.user
+**/*.dbmdl
+**/*.jfm
+**/bin
+**/charts
+**/docker-compose*
+**/compose*
+**/Dockerfile*
+**/node_modules
+**/npm-debug.log
+**/obj
+**/secrets.dev.yaml
+**/values.dev.yaml
+LICENSE
+README.md

.gitignore

@@ -0,0 +1,5 @@
+.venv
+.env
+__pycache__/
+.vscode
+venv

Dockerfile

@@ -0,0 +1,26 @@
+# For more information, please refer to https://aka.ms/vscode-docker-python
+FROM python:3.11-slim
+ 
+EXPOSE 8000
+ 
+# Keeps Python from generating .pyc files in the container
+ENV PYTHONDONTWRITEBYTECODE=1
+ 
+# Turns off buffering for easier container logging
+ENV PYTHONUNBUFFERED=1
+ 
+# Install pip requirements
+COPY requirements.txt .
+RUN python -m pip install -r requirements.txt
+ 
+WORKDIR /app
+COPY . /app
+ 
+# Creates a non-root user with an explicit UID and adds permission to access the /app folder
+# For more info, please refer to https://aka.ms/vscode-docker-python-configure-containers
+RUN adduser -u 5678 --disabled-password --gecos "" appuser && chown -R appuser /app
+USER appuser
+ 
+# During debugging, this entry point will be overridden. For more information, please refer to https://aka.ms/vscode-docker-python-debug
+CMD ["gunicorn", "--bind", "0.0.0.0:8000", "-k", "uvicorn.workers.UvicornWorker", "-t", "3500" ,"--keep-alive" , "3500", "app.main:app"]
+

app/main.py

@@ -0,0 +1,37 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from fastapi.middleware.cors import CORSMiddleware
+import uvicorn
+from app.removal import remove_bg_base64
+from pydantic import BaseModel
+
+ 
+app = FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+class ImageData(BaseModel):
+    image_data: str  
+
+active_connections = []
+VERSION = "1.0.0"
+
+@app.get("/")
+async def health_check():
+    return {"status": "healthy"}
+
+@app.post("/BgRemoval")
+async def bg_removal(request: ImageData):
+    try:
+        image_data = request.image_data
+        output_image = remove_bg_base64(image_data)
+        return {"output_image": output_image}
+    except Exception as e:
+        print(f"Error Occured: {str(e)}")
+        return {"error": str(e)}

app/removal.py

@@ -0,0 +1,52 @@
+import base64
+from io import BytesIO
+from rembg import remove
+from PIL import Image
+import numpy as np
+import scipy.ndimage as ndi
+
+
+def keep_largest_object_alpha(alpha_mask_np, threshold=0.1):
+    binary_mask = (alpha_mask_np > threshold).astype(np.uint8)
+    labeled, n = ndi.label(binary_mask)
+    if n == 0:
+        return alpha_mask_np
+    sizes = ndi.sum(binary_mask, labeled, range(n + 1))
+    largest_label = sizes.argmax()
+    largest_mask = (labeled == largest_label).astype(np.float32)
+    return largest_mask * alpha_mask_np
+
+
+def remove_bg_base64(base64_image: str) -> str:
+    # Decode base64 image to PIL Image
+    image_data = base64.b64decode(base64_image)
+    input_image = Image.open(BytesIO(image_data)).convert("RGBA")
+
+    # Remove background
+    output = remove(input_image)
+
+    # Convert to RGBA and extract alpha
+    output = output.convert("RGBA")
+    r, g, b, a = output.split()
+    alpha_np = np.array(a).astype(np.float32) / 255.0
+
+    # Keep only largest subject in the alpha mask
+    filtered_alpha = keep_largest_object_alpha(alpha_np)
+
+    # Convert filtered alpha to image
+    filtered_alpha_img = Image.fromarray((filtered_alpha * 255).astype(np.uint8))
+    output.putalpha(filtered_alpha_img)
+
+    # Paste onto white background
+    white_bg = Image.new("RGBA", output.size, (255, 255, 255, 255))
+    white_bg.paste(output, (0, 0), mask=output.split()[3])
+
+    # Final image (no alpha)
+    final_image = white_bg.convert("RGB")
+
+    # Convert to base64 string
+    buffered = BytesIO()
+    final_image.save(buffered, format="JPEG")
+    final_base64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
+
+    return final_base64

requirements.txt

@@ -0,0 +1,9 @@
+fastapi
+gunicorn
+pydantic
+uvicorn
+numpy
+rembg
+Pillow
+scipy
+onnxruntime