# 1. 首先clone Propainter-Webui
git clone https://github.com/halfzm/ProPainter-Webui.git

## 这里使用的是propainter-webui代码库，这个项目里面内嵌了propainter，
## 如果内嵌的propainter代码不是最新的，可以之间使用挂载的形式重新把最新的代码挂载进去
## github: https://github.com/sczhou/ProPainter

# 2. 下载依赖：
https://cdn-media.huggingface.co/frpc-gradio-0.2/frpc_linux_amd64 # 下载完成后的使用参考Dockerfile

# 3. 下载镜像文件，参照 https://github.com/halfzm/ProPainter-Webui#prepare-pretrained-models 里面的说明准备好两个目录：
/ProPainter-Webui/ckpt
/ProPainter-Webui/weights # 这两个目录通过挂载卷的形式挂载到容器内部

# 下载 https://huggingface.co/bert-base-uncased 放到ckpt里面：/ProPainter-Webui/ckpt/bert-base-uncased

# 4. 解决CPU Mode only问题
git clone https://github.com/IDEA-Research/GroundingDINO.git
cd GroundingDINO && python setup.py build
# 编译完成之后，找到文件 _C.cpython-310-x86_64-linux-gnu.so 放到目录/ProPainter-Webui/groundingdino中

# 5. 修改/ProPainter-Webui/requirements.txt，注释掉下面两行
torch==2.0.1+cu117
torchvision==0.15.2+cu117

# 6.打包镜像
docker build -t propainter-ui:v1.2 .

# 7.启动docker时设置环境变量 CUDA_HOME=/usr/local/cuda，可以参考docker-compose.yml