Update Dockerfile

Dockerfile

@@ -1,3 +1,4 @@
+# Use Python 3.10 for stability
 FROM python:3.10-slim
 
 # Install system dependencies
@@ -6,57 +7,74 @@ RUN apt-get update && apt-get install -y \
     curl \
     libgl1-mesa-glx \
     libglib2.0-0 \
-    ffmpeg \
     libsm6 \
     libxext6 \
     libxrender-dev \
+    ffmpeg \
+    libgomp1 \
     && rm -rf /var/lib/apt/lists/*
 
 # Set environment variables
 ENV U2NET_HOME=/tmp/u2net_models
 ENV OMP_NUM_THREADS=2
 ENV PYTHONUNBUFFERED=1
+ENV PYTHONDONTWRITEBYTECODE=1
 
+# Create app directory
 WORKDIR /app
 
-# Copy requirements
+# Copy requirements first for better caching
 COPY requirements.txt .
 
-# Install Python packages
+# Install Python dependencies
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt
 
-# Create cache directory
-RUN mkdir -p /tmp/u2net_models
+# Create cache directory for models
+RUN mkdir -p /tmp/u2net_models && chmod 777 /tmp/u2net_models
 
-# Pre-download essential models
+# Pre-download essential models (non-blocking)
 RUN python -c "
 import sys
-sys.stderr = sys.stdout
-try:
-    from rembg import new_session
-    print('Pre-downloading essential models...')
-    essential_models = ['u2netp', 'silueta', 'u2net']
-    for model in essential_models:
-        try:
-            print(f'Downloading {model}...')
-            session = new_session(model)
-            print(f'✅ {model} downloaded')
-        except Exception as e:
-            print(f'⚠️ {model}: {str(e)[:100]}')
-except Exception as e:
-    print(f'Model pre-download error: {e}')
-"
+import subprocess
+import threading
+
+def download_model(model):
+    try:
+        print(f'Pre-downloading {model}...')
+        import rembg
+        session = rembg.new_session(model)
+        print(f'✅ {model} downloaded successfully')
+    except Exception as e:
+        print(f'⚠️ {model}: {str(e)[:100]}')
+
+models = ['u2netp', 'silueta', 'u2net']
+threads = []
+
+for model in models:
+    thread = threading.Thread(target=download_model, args=(model,))
+    thread.start()
+    threads.append(thread)
+
+for thread in threads:
+    thread.join(timeout=120)
+
+print('Pre-download complete')
+" || echo "Model pre-download may continue at runtime"
 
 # Copy application
 COPY app.py .
 
-# Create non-root user
+# Create non-root user for security
 RUN useradd -m -u 1000 appuser && chown -R appuser:appuser /app
 USER appuser
 
-# Expose port
+# Expose port (Hugging Face uses 7860)
 EXPOSE 7860
 
+# Health check
+HEALTHCHECK --interval=30s --timeout=3s --start-period=5s --retries=3 \
+  CMD curl -f http://localhost:7860/api/health || exit 1
+
 # Run the application
 CMD ["python", "app.py"]