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
### 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
### Main Plugin
RUN git clone https://github.com/Kosinkadink/ComfyUI-Advanced-ControlNet.git /workspace/ComfyUI/custom_nodes/ComfyUI-Advanced-ControlNet
### Models
# 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
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/thibaud/controlnet-openpose-sdxl-1.0/resolve/main/OpenPoseXL2.safetensors 
# RUN wget -P /workspace/ComfyUI/models/controlnet https://huggingface.co/thibaud/controlnet-sd21/resolve/main/control_v11p_sd21_lineart.safetensors

# 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

# IP ADAPTER
RUN git clone https://github.com/cubiq/ComfyUI_IPAdapter_plus.git /workspace/ComfyUI/custom_nodes/ComfyUI_IPAdapter_plus
### Model

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

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

EXPOSE 3000

CMD ["/start.sh"]