Dockerfile

# Multi-stage build for ML API
FROM python:3.12-slim as builder

# Set working directory
WORKDIR /app

# Install system dependencies
RUN apt-get update && apt-get install -y \
    gcc \
    g++ \
    && rm -rf /var/lib/apt/lists/*

# Copy requirements
COPY requirements.txt ./
COPY api/requirements.txt ./api-requirements.txt

# Install Python dependencies
RUN pip install --no-cache-dir --upgrade pip && \
    pip install --no-cache-dir -r requirements.txt && \
    pip install --no-cache-dir -r api-requirements.txt

# Final stage
FROM python:3.12-slim

WORKDIR /app

# Install runtime dependencies
RUN apt-get update && apt-get install -y \
    libgomp1 \
    && rm -rf /var/lib/apt/lists/*

# Copy Python packages from builder
COPY --from=builder /usr/local/lib/python3.12/site-packages /usr/local/lib/python3.12/site-packages
COPY --from=builder /usr/local/bin /usr/local/bin

# Copy application code
COPY api/ ./api/
COPY Data/ ./Data/
COPY mlruns/best_model_artifacts/ ./mlruns/best_model_artifacts/

# Create non-root user
RUN useradd -m -u 1000 apiuser && \
    chown -R apiuser:apiuser /app

USER apiuser

# Expose port
EXPOSE 8000

# Health check
HEALTHCHECK --interval=30s --timeout=10s --start-period=40s --retries=3 \
    CMD python -c "import requests; requests.get('http://localhost:8000/health')"

# Run the application with preprocessor check
CMD ["/bin/sh", "-c", "python -c 'import os; path=os.getenv(\"PREPROCESSOR_PATH\", \"/app/Data/preprocessor.pkl\"); exit(0 if os.path.exists(path) else 1)' && uvicorn api.main:app --host 0.0.0.0 --port 8000 || (echo \"ERROR: Preprocessor file not found at ${PREPROCESSOR_PATH:-/app/Data/preprocessor.pkl}. Please ensure Data folder is properly mounted or copied.\" && exit 1)"]