Initial commit: HF Spaces deployment with u2net model

.gitignore

@@ -0,0 +1,30 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+
+# Virtual environment
+venv/
+env/
+ENV/
+
+# IDE
+.vscode/
+.idea/
+*.swp
+*.swo
+
+# OS
+.DS_Store
+Thumbs.db
+
+# Testing
+.pytest_cache/
+.coverage
+htmlcov/
+
+# Local env files
+.env
+.env.local

Dockerfile

@@ -0,0 +1,23 @@
+FROM python:3.11-slim
+
+WORKDIR /app
+
+# Install system dependencies needed for OpenCV and image processing
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    libgl1 \
+    libglib2.0-0 \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy requirements and install Python dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy application code
+COPY app.py .
+
+# Expose port 7860 (Hugging Face Spaces default)
+EXPOSE 7860
+
+# Run the Flask app
+CMD ["python", "app.py"]

README.md

@@ -0,0 +1,65 @@
+---
+title: Background Removal API
+emoji: 🎨
+colorFrom: purple
+colorTo: blue
+sdk: docker
+pinned: false
+license: mit
+---
+
+# 🎨 Background Removal API
+
+**FREE forever** AI-powered background removal API using rembg.
+
+## Features
+
+- ✅ **100% FREE** - No API keys, no limits
+- 🚀 **Fast** - Optimized u2netp model
+- 🎯 **High Quality** - Professional results
+- 🔓 **No Authentication** - Ready to use
+- ♾️ **Unlimited** - Process as many images as you need
+
+## API Endpoints
+
+### POST /remove-background
+Upload an image and get back a PNG with transparent background.
+
+**Example (curl):**
+```bash
+curl -X POST https://YOUR_SPACE.hf.space/remove-background \
+  -F "image=@your-image.jpg" \
+  -o output.png
+```
+
+**Example (JavaScript):**
+```javascript
+const formData = new FormData();
+formData.append('image', fileInput.files[0]);
+
+const response = await fetch('https://YOUR_SPACE.hf.space/remove-background', {
+  method: 'POST',
+  body: formData
+});
+
+const blob = await response.blob();
+```
+
+### GET /health
+Check if API is running.
+
+## Technology
+
+- **rembg** - AI background removal
+- **u2netp** - Lightweight, fast model
+- **Flask** - Python web framework
+- **Hugging Face Spaces** - FREE hosting forever
+
+## Deploy Your Own
+
+1. Create account on https://huggingface.co
+2. Create new Space (SDK: Docker)
+3. Upload these files
+4. 🎉 Done! Your API is live
+
+Built with ❤️ for the community

app.py

@@ -0,0 +1,106 @@
+"""
+Background Removal API for Hugging Face Spaces
+Self-hosted, FREE forever with unlimited usage!
+"""
+from flask import Flask, request, send_file
+from flask_cors import CORS
+from rembg import remove, new_session
+from PIL import Image
+import io
+import os
+
+app = Flask(__name__)
+
+# Allow CORS from your Vercel domain
+CORS(app, origins=["*"])  # In production, replace * with your Vercel domain
+
+# Global session variable
+_session = None
+
+def get_session():
+    """Lazy load rembg session on first use"""
+    global _session
+    if _session is None:
+        print("🔧 Initializing rembg session with u2net model...")
+        try:
+            _session = new_session("u2net")
+            print("✅ Session initialized successfully!")
+        except Exception as e:
+            print(f"❌ Session init error: {e}")
+            raise
+    return _session
+
+@app.route('/')
+def root():
+    """Root endpoint with API info"""
+    return {
+        'name': 'Background Removal API',
+        'version': '2.1.0',
+        'backend': 'Self-hosted rembg (u2net model - High Quality)',
+        'platform': 'Hugging Face Spaces',
+        'features': '100% FREE forever, Unlimited usage, No API keys needed',
+        'endpoints': {
+            'POST /remove-background': 'Remove background from image',
+            'GET /health': 'Health check'
+        }
+    }, 200
+
+@app.route('/health')
+def health_check():
+    """Health check endpoint"""
+    return {
+        'status': 'ok', 
+        'message': 'Background removal API is running',
+        'backend': 'Self-hosted rembg with u2net model (High Quality)',
+        'platform': 'Hugging Face Spaces (FREE forever)'
+    }, 200
+
+@app.route('/remove-background', methods=['POST'])
+def remove_background():
+    """
+    Background removal using rembg with u2netp model
+    NO external APIs - 100% self-hosted!
+    """
+    try:
+        # Get uploaded file
+        if 'image' not in request.files:
+            return {'error': 'No image file provided'}, 400
+        
+        file = request.files['image']
+        input_image = Image.open(file.stream)
+        
+        print(f"🎨 Processing image: {file.filename}")
+        print(f"📐 Size: {input_image.size}")
+        
+        # Get session and remove background
+        session = get_session()
+        output_image = remove(input_image, session=session)
+        
+        print(f"✅ Background removed successfully!")
+        
+        # Convert to PNG bytes
+        img_io = io.BytesIO()
+        output_image.save(img_io, 'PNG', optimize=False)
+        img_io.seek(0)
+        
+        return send_file(
+            img_io,
+            mimetype='image/png',
+            as_attachment=True,
+            download_name=f"{file.filename.rsplit('.', 1)[0]}-nobg.png"
+        )
+    
+    except Exception as e:
+        print(f"❌ Error: {str(e)}")
+        import traceback
+        traceback.print_exc()
+        return {'error': str(e)}, 500
+
+if __name__ == '__main__':
+    port = int(os.environ.get('PORT', 7860))  # HF Spaces uses port 7860
+    print("🚀 Background Removal API starting on Hugging Face Spaces...")
+    print(f"📍 Running on port {port}")
+    print("✅ Using self-hosted rembg (NO external APIs!)")
+    print("🎯 Model: u2netp (optimized for quality)")
+    print("🆓 FREE forever - Unlimited usage!")
+    app.run(host='0.0.0.0', port=port, debug=False)

requirements.txt

@@ -0,0 +1,4 @@
+flask==3.0.0
+flask-cors==4.0.0
+rembg[cpu]==2.0.72
+pillow>=10.4.0