FROM quay.io/ascend/vllm-ascend:v0.10.0rc1

WORKDIR /workspace

RUN cp /etc/apt/sources.list /etc/apt/sources.list.bak && \
    echo "deb [trusted=yes] https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ jammy main restricted universe multiverse" > /etc/apt/sources.list && \
    echo "deb [trusted=yes] https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ jammy-updates main restricted universe multiverse" >> /etc/apt/sources.list && \
    echo "deb [trusted=yes] https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ jammy-backports main restricted universe multiverse" >> /etc/apt/sources.list && \
    echo "deb [trusted=yes] https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ jammy-security main restricted universe multiverse" >> /etc/apt/sources.list && \
    apt-get update && \
    apt-get install -y ffmpeg libsox-dev redis && \
    rm -rf /var/lib/apt/lists/*

COPY wav /workspace/wav
COPY GPT-SoVITS /workspace/GPT-SoVITS
COPY constraints_gsv.txt /workspace/

RUN pip install -r GPT-SoVITS/extra-req.txt --no-deps \
    && pip install -r GPT-SoVITS/requirements.txt -c constraints_gsv.txt 

COPY launch_gsv.sh /workspace/
ENTRYPOINT ["/bin/bash", "launch_gsv.sh"]