FROM python:3.10-slim

RUN apt-get update && \
    apt-get install -y build-essential ffmpeg wget && \
    rm -rf /var/lib/apt/lists/*

WORKDIR /app

COPY requirements.txt .

RUN pip install --no-cache-dir -r requirements.txt

COPY ./app ./app

RUN mkdir -p ./app/models && \
    wget -O ./app/models/chu2.pth https://huggingface.co/Wanlau/RVC_BanGDream/resolve/main/Chiyu_v2_48k/Chiyu_v2_48k.pth && \
    wget -O ./app/models/chu2.index https://huggingface.co/Wanlau/RVC_BanGDream/resolve/main/Chiyu_v2_48k/added_IVF256_Flat_nprobe_1_Chiyu_v2_48k_v2.index

EXPOSE 7860

CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]