# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/11.2.1/ubuntu2004/devel/cudnn8/Dockerfile
FROM nvidia/cuda:11.2.1-cudnn8-devel-ubuntu20.04
ENV DEBIAN_FRONTEND noninteractive

WORKDIR /content
RUN apt-get update -y && apt-get upgrade -y && apt-get install -y sudo && apt-get install -y python3-pip && pip3 install --upgrade pip
RUN apt-get install -y gnupg wget htop sudo git git-lfs software-properties-common build-essential cmake curl
RUN apt-get install -y ffmpeg libavcodec-dev libavformat-dev libavdevice-dev libgl1 libgtk2.0-0 jq libdc1394-22-dev libraw1394-dev libopenblas-base

ENV PATH="/home/admin/.local/bin:${PATH}"

RUN pip3 install pandas scipy matplotlib torch torchvision torchaudio gradio altair imageio-ffmpeg pocketsphinx jq "numpy<1.24" yt_dlp ffmpeg-python

RUN git lfs install

RUN git clone https://huggingface.co/camenduru/Wav2Lip && cd Wav2Lip
COPY app.py /content/app.py

RUN adduser --disabled-password --gecos '' user

RUN chown -R user:user /content
RUN chmod -R 777 /content
USER user

WORKDIR /content/Wav2Lip

EXPOSE 7860

CMD ["python3", "/content/app.py"]