# Ubuntu 20.04
# CUDA 11.6.2
# cuDNN 8
FROM nvcr.io/nvidia/cuda:11.6.2-cudnn8-runtime-ubuntu20.04

ENV TZ='Europe/Rome'
ENV BASE_FOLDER='/mmc_asr'
ENV LOGS_FOLDER='/LOGS'
ENV APP_USER='devuser'

ARG GIT_NAME
ARG GIT_TOKEN

RUN \
  apt-get update && \
  DEBIAN_FRONTEND=noninteractive apt-get install --no-install-recommends -y \
    python3.8 \
    python3-pip \
    tzdata \
    ffmpeg \
    git \
    # TODO Other packets common to all images go in here
  && \
  apt-get clean && \
  rm -rf /var/lib/apt/lists/* && \
  ln -snf "/usr/share/zoneinfo/$TZ" '/etc/localtime' && \
  echo "$TZ" > '/etc/timezone'

RUN useradd -m "$APP_USER" && \
  mkdir -p "$BASE_FOLDER" && \
  chown -R "${APP_USER}:${APP_USER}" "$BASE_FOLDER" && \
  mkdir -p "$LOGS_FOLDER" && \
  chown -R "${APP_USER}:${APP_USER}" "$LOGS_FOLDER"

USER "$APP_USER"
WORKDIR "$BASE_FOLDER"

ENV PATH="${PATH}:/root/.local/bin:/home/${APP_USER}/.local/bin"
ENV PYTHONPATH="${PYTHONPATH}:${BASE_FOLDER}"

COPY --chown="${APP_USER}:${APP_USER}" /requirements.txt ./requirements.txt

RUN python3 -m pip install --no-cache-dir -r requirements.txt

WORKDIR "$BASE_FOLDER"/src
RUN git clone https://${GIT_NAME}:${GIT_TOKEN}@experts.mpai.community/software/mpai-aif/osd_tma/common_module.git
RUN pwd
RUN ls

WORKDIR ..
COPY --chown="${APP_USER}:${APP_USER}" /src ./src

CMD ["python3.8", "src/main.py"]