FROM runpod/stable-diffusion:comfy-ui-4.0.0

RUN apt update

# CUSTOM NODES PREINSTALL
RUN mkdir -p /workspace/ComfyUI/custom_nodes
RUN git clone https://github.com/ltdrdata/ComfyUI-Manager.git /workspace/ComfyUI/custom_nodes/ComfyUI-Manager
RUN git clone https://github.com/pythongosssss/ComfyUI-Custom-Scripts.git /workspace/ComfyUI/custom_nodes/ComfyUI-Custom-Scripts
RUN git clone https://github.com/jags111/efficiency-nodes-comfyui.git /workspace/ComfyUI/custom_nodes/efficiency-nodes-comfyui
# RUN apt-get install -y ffmpeg
# RUN pip install imageio-ffmpeg
# RUN git clone https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite.git /workspace/ComfyUI/custom_nodes/ComfyUI-VideoHelperSuite

# CONTROLNET
### Main Plugin
RUN git clone https://github.com/Kosinkadink/ComfyUI-Advanced-ControlNet.git /workspace/ComfyUI/custom_nodes/ComfyUI-Advanced-ControlNet
### Preprocessor
RUN git clone https://github.com/Fannovel16/comfyui_controlnet_aux.git /workspace/ComfyUI/custom_nodes/comfyui_controlnet_aux
RUN pip install -r /workspace/ComfyUI/custom_nodes/comfyui_controlnet_aux/requirements.txt
### Models
##### SD 1.5
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/main/control_v11p_sd15_openpose.pth
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_seg_fp16.safetensors 
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_openpose_fp16.safetensors
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11e_sd15_ip2p_fp16.safetensors
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11e_sd15_shuffle_fp16.safetensors 
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_canny_fp16.safetensors
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11f1p_sd15_depth_fp16.safetensors 
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_inpaint_fp16.safetensors 
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_lineart_fp16.safetensors 
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_mlsd_fp16.safetensors
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_normalbae_fp16.safetensors
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_scribble_fp16.safetensors
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_softedge_fp16.safetensors 
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15s2_lineart_anime_fp16.safetensors 
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11u_sd15_tile_fp16.safetensors
##### T2I ADAPTER
##### SDXL
RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/thibaud/controlnet-openpose-sdxl-1.0/resolve/main/OpenPoseXL2.safetensors 

# IP ADAPTER
RUN git clone https://github.com/cubiq/ComfyUI_IPAdapter_plus.git /workspace/ComfyUI/custom_nodes/ComfyUI_IPAdapter_plus
### IP ADAPTER MODELS SDXL
RUN wget -P /workspace/ComfyUI/models/ipadapter https://huggingface.co/h94/IP-Adapter/resolve/main/sdxl_models/ip-adapter_sdxl_vit-h.safetensors
RUN wget -P /workspace/ComfyUI/models/ipadapter https://huggingface.co/h94/IP-Adapter/resolve/main/sdxl_models/ip-adapter-plus_sdxl_vit-h.safetensors
RUN wget -P /workspace/ComfyUI/models/ipadapter https://huggingface.co/h94/IP-Adapter/resolve/main/sdxl_models/ip-adapter-plus-face_sdxl_vit-h.safetensors

### CLIP Vision
RUN wget -P /workspace/ComfyUI/models/clip_vision https://huggingface.co/h94/IP-Adapter/resolve/main/models/image_encoder/model.safetensors

# INSIGHTFACE
RUN apt-get install -y build-essential
RUN pip install onnxruntime
RUN pip install insightface

# TOP MODELS
# RUN wget -O /workspace/ComfyUI/models/checkpoints/realistic_vision.safetensors https://civitai.com/api/download/models/130072 download as folder, need fix

# VAE
# RUN wget -P /workspace/ComfyUI/models/vae https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors
# RUN wget -P /workspace/ComfyUI/models/vae https://huggingface.co/stabilityai/sdxl-vae/resolve/main/sdxl_vae.safetensors
# RUN wget -P /workspace/ComfyUI/models/vae https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/VAEs/orangemix.vae.pt 
# RUN wget -P /workspace/ComfyUI/models/vae https://huggingface.co/hakurei/waifu-diffusion-v1-4/resolve/main/vae/kl-f8-anime2.ckpt

### AnimateDiff
# RUN git clone https://github.com/Kosinkadink/ComfyUI-AnimateDiff-Evolved.git /workspace/ComfyUI/custom_nodes/ComfyUI-AnimateDiff-Evolved
# RUN wget -P /workspace/ComfyUI/custom_nodes/ComfyUI-AnimateDiff-Evolved/models https://huggingface.co/guoyww/animatediff/resolve/main/mm_sd_v15.ckpt
# RUN wget -P /workspace/ComfyUI/custom_nodes/ComfyUI-AnimateDiff-Evolved/models https://huggingface.co/guoyww/animatediff/resolve/main/mm_sd_v15_v2.ckpt
# RUN wget -P /workspace/ComfyUI/custom_nodes/ComfyUI-AnimateDiff-Evolved/models https://huggingface.co/CiaraRowles/TemporalDiff/resolve/main/temporaldiff-v1-animatediff.safetensors
# RUN wget -P /workspace/ComfyUI/custom_nodes/ComfyUI-AnimateDiff-Evolved/models https://huggingface.co/manshoety/AD_Stabilized_Motion/resolve/main/mm-Stabilized_high.pth


EXPOSE 3000

CMD ["/start.sh"]