FROM jupyterlab-pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10-devel as base
ARG IMAGE=kolors-faceid
ARG IMAGE_UPPER=Kolors-FaceID
ARG BRANCH=dcu
RUN cd /root && git clone -b $BRANCH http://developer.hpccube.com/codes/chenpangpang/$IMAGE.git
WORKDIR /root/$IMAGE/$IMAGE_UPPER
RUN apt-get update && apt-get install -y gcc g++
RUN pip install -r requirements.txt
RUN wget --no-check-certificate  --content-disposition \
    'https://cancon.hpccube.com:65024/directlink/4/onnxruntime/DAS1.2/onnxruntime-1.15.0+das.opt1.dtk24042-cp310-cp310-manylinux_2_28_x86_64.whl' && \
    pip install onnxruntime-1.15.0+das.opt1.dtk24042-cp310-cp310-manylinux_2_28_x86_64.whl
RUN pip install numpy==1.24.4

#########
# Prod  #
#########
FROM jupyterlab-pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10-devel
ARG IMAGE=kolors-faceid
ARG IMAGE_UPPER=Kolors-FaceID
COPY chenyh/$IMAGE/frpc_linux_amd64_v0.3 /opt/conda/lib/python3.10/site-packages/gradio/
RUN chmod +x /opt/conda/lib/python3.10/site-packages/gradio/frpc_linux_amd64_v0.3
COPY chenyh/$IMAGE/Kwai-Kolors/Kolors /root/$IMAGE_UPPER/Kwai-Kolors/Kolors
COPY chenyh/$IMAGE/Kwai-Kolors/Kolors-IP-Adapter-FaceID-Plus /root/$IMAGE_UPPER/Kwai-Kolors/Kolors-IP-Adapter-FaceID-Plus
COPY chenyh/$IMAGE/antelopev2 /root/$IMAGE_UPPER/.insightface/models/antelopev2
RUN apt-get update && apt install -y libglib2.0-dev
COPY --from=base /opt/conda/lib/python3.10/site-packages /opt/conda/lib/python3.10/site-packages
COPY --from=base /root/$IMAGE/$IMAGE_UPPER /root/$IMAGE_UPPER
COPY --from=base /root/$IMAGE/启动器.ipynb /root/$IMAGE/start.sh  /root/
COPY --from=base /root/$IMAGE/assets/  /root/