Upload folder using huggingface_hub

app/main.py

@@ -10,6 +10,7 @@ from app.routers.pdf import pdf_tools
 # 📌 Image Module
 # ==========================
 from app.routers.image import jpgcompressor
+from app.routers.image import bgremover
 
 # ==========================
 # 🔒 Security Module
@@ -52,6 +53,7 @@ app.include_router(calculator.router)                 # Mini calculator service
 app.include_router(auth_system.router)                # <-- NEW AUTH SYSTEM
 app.include_router(storage.router)                    # new cloud drive support
 app.include_router(server_status.router)              # function to check the system
+app.include_router(bgremover.router)                  # Background Remover
 
 
 
@@ -63,4 +65,5 @@ def greet_json():
     return """
     <h2>Go to the Swagger docs 👇</h2>
     <a href="/docs">Click here for API Docs</a>
+    <h3>created by sameer banchhor</h3>
     """

app/routers/image/bgremover.py

@@ -0,0 +1,46 @@
+from fastapi import APIRouter, UploadFile, File, Depends, HTTPException
+from fastapi.responses import Response
+from rembg import remove
+from PIL import Image
+import io
+
+# Import the auth dependency from your existing system
+from app.routers.auth.system import get_current_user
+
+router = APIRouter(
+    prefix="/image",
+    tags=["Image Processing"]
+)
+
+# ==========================================
+# 🔐 Auth Toggle
+# ==========================================
+AUTH_ENABLED = False  # <<< set False to disable authentication
+
+# Logic to conditionally apply authentication
+# If True, the endpoint will require a valid token.
+# If False, the dependency list is empty.
+security_dependencies = [Depends(get_current_user)] if AUTH_ENABLED else []
+
+@router.post("/remove-bg", dependencies=security_dependencies)
+async def remove_background(file: UploadFile = File(...)):
+    """
+    Upload an image and receive it back with the background removed (PNG).
+    """
+    # 1. Validate file type
+    if not file.content_type.startswith("image/"):
+        raise HTTPException(status_code=400, detail="Invalid file type. Please upload an image.")
+
+    try:
+        # 2. Read image data
+        image_data = await file.read()
+        
+        # 3. Process image using rembg
+        # We pass the raw bytes to remove() and it returns bytes (PNG)
+        output_data = remove(image_data)
+
+        # 4. Return the result as a raw image response
+        return Response(content=output_data, media_type="image/png")
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Image processing failed: {str(e)}")

requirements.txt

@@ -45,6 +45,8 @@ httpx                        # Async HTTP client
 # 🖼 IMAGE PROCESSING
 # =========================================
 Pillow                       # Image compression/manipulation
+rembg                        # AI Background removal tools
+onnxruntime                  # Required runtime for rembg (or onnxruntime-gpu)
 
 # =========================================
 # 📑 PDF HANDLING
@@ -62,7 +64,7 @@ aiofiles                     # Async file write/read
 boto3                        # AWS S3 storage
 minio                        # Local S3-compatible storage
 google-cloud-storage         # Google Cloud storage support
-hf                           #huggingface-cli
+hf                           # OPTIONAL huggingface-cli for deployment purposes
 
 # =========================================
 # ⚡ CACHING / QUEUES / BACKGROUND JOBS