## docker build -t cosyvoice:1.0.0 -f cosyvoice.df .

# 使用官方 Ubuntu 22.04 基础镜像
FROM ubuntu:22.04

# 设置非交互式安装以避免阻塞
ENV DEBIAN_FRONTEND=noninteractive

RUN apt update -y && apt upgrade -y \
&& apt install software-properties-common -y \
&& add-apt-repository ppa:deadsnakes/ppa -y \
&& apt update -y \
&& apt install zip unzip wget curl lrzsz vim git git-lfs sox libsox-dev ffmpeg -y \
&& apt install python3.10 -y \
&& apt install python3-pip -y \
&& mkdir /data

RUN pip config set global.index-url https://mirrors.aliyun.com/pypi/simple \
&& pip config set install.trusted-host mirrors.aliyun.com \
&& pip install --upgrade pip setuptools wheel \
&& pip install pynini==2.1.6.post1 


RUN cd /data \
## 使用gitee仓库,记得修改 .gitmodules 文件
&& git clone --recursive https://github.com/FunAudioLLM/CosyVoice.git \
&& cd /data/CosyVoice && git submodule update --init --recursive \
&& mkdir -p /data/CosyVoice/pretrained_models \
&& git clone https://www.modelscope.cn/iic/CosyVoice2-0.5B.git pretrained_models/CosyVoice2-0.5B \
&& git clone https://www.modelscope.cn/iic/CosyVoice-300M-SFT.git pretrained_models/CosyVoice-300M-SFT \
&& git clone https://www.modelscope.cn/iic/CosyVoice-300M-Instruct.git pretrained_models/CosyVoice-300M-Instruct \
&& git clone https://www.modelscope.cn/iic/CosyVoice-ttsfrd.git pretrained_models/CosyVoice-ttsfrd \
&& cd /data/CosyVoice/pretrained_models/CosyVoice-ttsfrd/  \
&& unzip resource.zip -d . \
&& pip install ttsfrd_dependency-0.1-py3-none-any.whl \
&& pip install ttsfrd-0.4.2-cp310-cp310-linux_x86_64.whl \
&& cd /data/CosyVoice/

RUN cd /data/CosyVoice && pip install -r requirements.txt

WORKDIR /data/CosyVoice/

EXPOSE 50000

RUN python3 --version 


### 指定使用的GPU
# CUDA_VISIBLE_DEVICES=6 python3 webui.py --port 50000 --model_dir pretrained_models/CosyVoice2-0.5B 
# python3 webui.py --port 50000 --model_dir pretrained_models/CosyVoice2-0.5B