# Stage 1: Build stage (installs docker)
FROM python:3.9.13 AS builder

RUN apt-get update && apt-get install -y \
    docker-ce \
    && apt-get clean \
    && rm -rf /var/lib/apt/lists/*

# Clear build cache
RUN docker builder prune -f

FROM python:3.9.13

USER root
RUN apt-get update && \
    apt-get install -y \
    libgl1-mesa-glx \
    ffmpeg \
    && apt-get clean \
    && rm -rf /var/lib/apt/lists/*

RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user \
    PATH=/home/user/.local/bin:$PATH \
    NUMBA_CACHE_DIR=/tmp/numba_cache

WORKDIR $HOME/app

COPY --chown=user . $HOME/app

RUN chmod +x scripts/download_models.sh
RUN bash scripts/download_models.sh

# Install Python dependencies
RUN pip install --no-cache-dir -r requirements.txt

# CMD ["gunicorn", "-b", "0.0.0.0:7860","app:app"]
CMD ["gunicorn", "-b", "0.0.0.0:7860", "--timeout", "300", "app:app"]
# CMD ["gunicorn", "-b", "0.0.0.0:7860", "-w", "4", "--timeout", "300", "app_parallel:app"]
# CMD ["gunicorn", "-b", "0.0.0.0:7860", "--workers", "3", "--timeout", "300", "--worker-class", "gevent", "app:app"]