Update Dockerfile

Dockerfile

@@ -1,24 +1,80 @@
-FROM python:3.9-slim
-
-WORKDIR /app
-
-# Install minimal system dependencies
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    && rm -rf /var/lib/apt/lists/*
-
-# Copy and install requirements
-COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
-
-# Copy application
-COPY app.py .
-
-# Set environment variables for Gradio
-ENV GRADIO_SERVER_NAME="0.0.0.0"
-ENV GRADIO_SERVER_PORT=7860
-
-# Expose port
-EXPOSE 7860
-
-# Run the app
-CMD ["python", "app.py"]
+FROM python:3.10-slim
+
+WORKDIR /app
+
+# Install system dependencies for AI models and image processing
+RUN apt-get update && apt-get install -y \
+    gcc \
+    g++ \
+    git \
+    curl \
+    wget \
+    libgl1 \
+    libglib2.0-0 \
+    libsm6 \
+    libxext6 \
+    libxrender-dev \
+    libgl1-mesa-glx \
+    ffmpeg \
+    && rm -rf /var/lib/apt/lists/* \
+    && apt-get clean
+
+# Copy requirements first for better Docker layer caching
+COPY requirements.txt .
+
+# Install Python dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Create necessary directories
+RUN mkdir -p \
+    /app/models \
+    /app/temp \
+    /app/uploads \
+    /app/outputs \
+    /app/examples
+
+# Copy application code
+COPY . .
+
+# Pre-download and cache AI models during build
+RUN python -c "
+import rembg
+import torch
+from transformers import pipeline
+
+# Pre-download background removal model
+print('Downloading background removal model...')
+rembg.remove((255, 255, 255, 255))
+
+# Pre-download other essential models
+print('Pre-caching AI models...')
+try:
+    # Face detection model
+    from insightface.app import FaceAnalysis
+    print('Models pre-cached successfully')
+except Exception as e:
+    print(f'Model caching: {e}')
+" || true
+
+# Create non-root user for security
+RUN groupadd -r appuser && useradd -r -g appuser appuser
+RUN chown -R appuser:appuser /app
+USER appuser
+
+# Expose ports
+EXPOSE 7860  # Gradio
+EXPOSE 8000  # FastAPI
+
+# Health check
+HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
+    CMD python -c "import requests; requests.get('http://localhost:7860/', timeout=10)" || exit 1
+
+# Set environment variables
+ENV PYTHONUNBUFFERED=1
+ENV GRADIO_SERVER_NAME="0.0.0.0"
+ENV GRADIO_SERVER_PORT=7860
+ENV MODEL_CACHE_DIR="/app/models"
+ENV TEMP_DIR="/app/temp"
+
+# Run the application
+CMD ["python", "app.py"]