FROM nvidia/cuda:11.6.2-cudnn8-devel-ubuntu20.04

ARG DEBIAN_FRONTEND=noninteractive

# install python via pyenv
RUN apt-get update && apt-get install -y --no-install-recommends \
    make \
    build-essential \
    libssl-dev \
    zlib1g-dev \
    libbz2-dev \
    libreadline-dev \
    libsqlite3-dev \
    wget \
    curl \
    llvm \
    libncurses5-dev \
    libncursesw5-dev \
    xz-utils \
    tk-dev \
    libffi-dev \
    liblzma-dev \
    git \
    ca-certificates \
    libgl1 \
    && rm -rf /var/lib/apt/lists/*



# install deps
RUN apt-get update && apt-get install -y --no-install-recommends \
    ffmpeg libsndfile1 \
    && rm -rf /var/lib/apt/lists/*


# Set up a new user named "user" with user ID 1000
RUN useradd -m -u 1000 user

# Switch to the "user" user
USER user


ENV PATH="/home/user/.pyenv/shims:/home/user/.pyenv/bin:$PATH"
ARG PYTHON_VERSION=3.8

RUN curl -s -S -L https://raw.githubusercontent.com/pyenv/pyenv-installer/master/bin/pyenv-installer | bash && \
    pyenv install $PYTHON_VERSION && \
    pyenv global $PYTHON_VERSION

ENV HOME=/home/user \
    CUDA_HOME=/usr/local/cuda \
    PATH=/home/user/.local/bin:$PATH \
    LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH} \
    LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:${LIBRARY_PATH}


WORKDIR $HOME/app
RUN git clone -b master https://github.com/justinjohn0306/Wav2Lip $HOME/app
COPY . $HOME/app/


COPY ./checkpoints/s3fd-619a316812.pth  $HOME/app/face_detection/detection/sfd/s3fd.pth

RUN pip install --upgrade pip
RUN pip install git+https://github.com/elliottzheng/batch-face.git@master
RUN pip install --no-cache-dir -r requirements.txt
ENV PYTHONUNBUFFERED=1

CMD ["python", "app.py"]