FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-runtime

ENV NB_USER jovyan
ENV NB_UID 1000
ENV NB_PREFIX /

RUN export DEBIAN_FRONTEND=noninteractive \
    && apt-get -yq update \
    && apt-get -yq install --no-install-recommends \
        git \
        ffmpeg \
        software-properties-common \
    &&  apt upgrade -y \
    && apt-get clean \
    && rm -rf /var/lib/apt/lists/*

RUN export DEBIAN_FRONTEND=noninteractive \
    && add-apt-repository ppa:deadsnakes/ppa \
    && apt-get -yq update \
    && apt-get -yq install --no-install-recommends \
        python3.10 \
        python3.10-venv \
    &&  apt upgrade -y \
    && apt-get clean \
    && rm -rf /var/lib/apt/lists/*

RUN useradd -M -s /bin/bash -N -u ${NB_UID} ${NB_USER} \
    && mkdir -p ${HOME} \
    && chown -R ${NB_USER}:users ${HOME} \
    && chown -R ${NB_USER}:users /usr/local/bin

USER $NB_UID
WORKDIR /home/${NB_USER}/
ENV PATH=/venv/bin:$PATH

COPY --chown=${NB_USER}:users ./requirements.txt /home/${NB_USER}/requirements.txt
RUN python3.10 -m venv /home/${NB_USER}/venv \
    && /home/${NB_USER}/venv/bin/pip install --upgrade pip wheel \
    && /home/${NB_USER}/venv/bin/pip install -r /home/${NB_USER}/requirements.txt

COPY --chown=${NB_USER}:users ./.env* /home/${NB_USER}/
COPY --chown=${NB_USER}:users ./src /home/${NB_USER}/src

EXPOSE 7860

CMD ["/home/jovyan/venv/bin/python", "-u", "/home/jovyan/src/app.py"]