#!/usr/bin/env bash
set -e

# Optional: show each command as it runs (uncomment next line if you want)
# set -x

echo "=== Installing prerequisites (unzip, wget) ==="
apt-get update
apt-get install -y unzip wget

echo "=== Downloading ControlNet + Checkpoint models ==="
mkdir -p /ComfyUI/models/controlnet
mkdir -p /ComfyUI/models/checkpoints

cd /ComfyUI/models/controlnet
wget -O diffusion_pytorch_model_promax.safetensors \
  "https://huggingface.co/saliacoel/MyCustomNodes/resolve/main/diffusion_pytorch_model_promax.safetensors"

cd /ComfyUI/models/checkpoints
wget -O SaliaHighlady_Speedy.safetensors \
  "https://huggingface.co/saliacoel/MyCustomNodes/resolve/main/SaliaHighlady_Speedy.safetensors"

echo "=== Downloading + installing custom nodes zip packages ==="
mkdir -p /ComfyUI/custom_nodes
cd /ComfyUI/custom_nodes

wget -O comfyui-TRT_VAE.zip \
  "https://huggingface.co/saliacoel/wan/resolve/main/comfyui-TRT_VAE.zip"
unzip -o comfyui-TRT_VAE.zip
rm -f comfyui-TRT_VAE.zip

wget -O comfyui-salia_online.zip \
  "https://huggingface.co/saliacoel/wan/resolve/main/comfyui-salia_online.zip"
unzip -o comfyui-salia_online.zip
rm -f comfyui-salia_online.zip

wget -O comfyui-saliaLite.zip \
  "https://huggingface.co/saliacoel/wan/resolve/main/comfyui-saliaLite.zip"
unzip -o comfyui-saliaLite.zip
rm -f comfyui-saliaLite.zip

wget -O Salia_RMBG_TRT.zip "https://huggingface.co/saliacoel/wan/resolve/main/Salia_RMBG_TRT.zip" && unzip -o Salia_RMBG_TRT.zip && rm Salia_RMBG_TRT.zip

wget -O comfyui-salia_facedetailer.zip \
  "https://huggingface.co/saliacoel/wan/resolve/main/comfyui-salia_facedetailer.zip"
unzip -o comfyui-salia_facedetailer.zip
rm -f comfyui-salia_facedetailer.zip

echo "=== Done. ==="