FROM python:3.10

RUN apt-get update && apt-get install -y \
    ffmpeg \
    pkg-config \
    build-essential \
    libavcodec-dev \
    libavformat-dev \
    libavutil-dev \
    libswresample-dev \
    libswscale-dev \
    libavdevice-dev \
    libavfilter-dev \
    git \
    && rm -rf /var/lib/apt/lists/*

RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user \
    PATH=/home/user/.local/bin:$PATH
WORKDIR $HOME/app

RUN pip install --no-cache-dir --upgrade pip setuptools wheel
RUN pip install --no-cache-dir av==12.3.0
# QUOTES ADDED HERE:
RUN pip install --no-cache-dir torch torchaudio "numpy<2.0.0" gradio huggingface_hub \
    soundfile hydra-core julius encodec demucs flashy omegaconf antlr4-python3-runtime \
    xformers tqdm typing-extensions num2words spacy sentencepiece \
    transformers einops librosa torchdiffeq dora-search encodec \
    rotary_embedding_torch hydra_colorlog \
    pesq pystoi torchmetrics==1.2.0 peft scipy
RUN pip install --no-cache-dir --no-deps git+https://github.com/facebookresearch/audiocraft.git

COPY --chown=user . $HOME/app

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