# 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 \
    redis-server \
    && 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 ["bash", "-c", "redis-server & celery -A app_celery.celery worker --loglevel=info --pool=gevent & gunicorn -b 0.0.0.0:7860 --worker-class gevent --timeout 300 app_celery:app"]