FROM python:3.10-slim AS build

RUN pip install runpod

RUN apt update
RUN apt-get -y install build-essential python3-dev ffmpeg
RUN pip3 install --upgrade setuptools wheel
RUN pip3 install --upgrade pip
RUN pip3 install faiss-gpu fairseq gradio ffmpeg ffmpeg-python praat-parselmouth pyworld numpy==1.23.5 numba==0.56.4 librosa==0.9.2
RUN apt -y install -qq aria2
RUN pip install mega.py
RUN pip install gdown
RUN apt install git -y
RUN apt install unzip -y
RUN pip install opencv_python>=4.2.0

COPY Retrieval-based-Voice-Conversion-WebUI .
COPY models/rose-bp.pth weights/rose-bp.pth
COPY models/jungkook.pth weights/jungkook.pth
COPY models/drake.pth weights/drake.pth
COPY models/ariana-grande.pth weights/ariana-grande.pth
COPY models/iu.pth weights/iu.pth

FROM build AS app

COPY handler-no-splitter.py .
COPY voice_generation.py .

ENV PYTHONUNBUFFERED=1

CMD ["python", "/handler-no-splitter.py"]