FROM python:3.9

# Update apt
RUN apt-get update -y

# Add apt packages
RUN apt-get install libsndfile1 curl wget git-lfs espeak-ng -y

# Deps
# RUN apt-get install libsndfile1 espeak-ng -y

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

# Switch to the "user" user
USER user

# Set home to the user's home directory
ENV HOME=/home/user \
    PATH=/home/user/.local/bin:$PATH

# Set the working directory to the user's home directory
WORKDIR $HOME/app

# Clone the GitHub repository
RUN git clone https://github.com/neural-loop/TTS.git .

RUN pip install --no-cache-dir --upgrade tts
 
# Install dependencies
RUN pip install --no-cache-dir -r requirements.txt

RUN git lfs install
RUN git clone https://huggingface.co/voices/VCTK_Canadian_English model

# Copy the current directory contents into the container at $HOME/app, setting the owner to the user
COPY --chown=user . $HOME/app

RUN sed -i 's/supplemental\//model\/supplemental\//g' model/config.json

# Set the command to run the server
CMD ["python", "TTS/server/server.py", "--model_path", "model/checkpoint_40000.pth", "--config_path", "model/config.json", "--port", "7860"]