#!/usr/bin/env bash
set -euo pipefail

# Change to ComfyUI directory
cd /comfyui

# Ensure all target directories exist
mkdir -p \
    models/style_models \
    models/clip \
    models/clip_vision \
    models/diffusion_models \
    models/loras \
    models/sams \
    models/rembg \
    models/grounding-dino \
    models/ultralytics/bbox \
    models/ultralytics/segm

# Download filebrowser get.sh and run it
wget https://raw.githubusercontent.com/filebrowser/get/master/get.sh && chmod +x get.sh && ./get.sh

# Download required models for ReActor using comfy-cli
comfy model download --url https://huggingface.co/black-forest-labs/FLUX.1-Redux-dev/resolve/main/flux1-redux-dev.safetensors --set-hf-api-token "$HF_TOKEN" --relative-path models/style_models --filename flux1-redux-dev.safetensors

# comfy model download --url https://huggingface.co/comfyanonymous/flux_text_encoders/blob/main/clip_l.safetensors --relative-path models/clip --filename clip_l.safetensors
comfy model download --url https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/t5xxl_fp16.safetensors --relative-path models/clip --filename t5xxl_fp16.safetensors
comfy model download --url https://huggingface.co/funnewsr/sigclip_vision_patch14_384/resolve/main/sigclip_vision_patch14_384.safetensors --relative-path models/clip_vision --filename sigclip_vision_patch14_384.safetensors
comfy model download --url https://huggingface.co/black-forest-labs/FLUX.1-Fill-dev/resolve/main/flux1-fill-dev.safetensors --relative-path models/diffusion_models --filename flux1-fill-dev.safetensors
comfy model download --url https://huggingface.co/ali-vilab/ACE_Plus/resolve/main/subject/comfyui_subject_lora16.safetensors --relative-path models/loras --filename comfyui_subject_lora16.safetensors
comfy model download --url https://huggingface.co/GleghornLab/sam_vit_b_01ec64.pth/resolve/main/sam_vit_b_01ec64.pth --relative-path models/sams --filename sam_vit_b_01ec64.pth
comfy model download --url https://huggingface.co/bewiz/GroundingDINO_SwinT_OGC/resolve/main/groundingdino_swint_ogc.pth --relative-path models/grounding-dino --filename GroundingDINO_SwinT_COG.pth
comfy model download --url https://huggingface.co/xingren23/comfyflow-models/resolve/main/ultralytics/bbox/face_yolov8m.pt --relative-path models/ultralytics/bbox --filename face_yolov8m.pt
comfy model download --url https://huggingface.co/xingren23/comfyflow-models/resolve/main/ultralytics/bbox/hand_yolov8s.pt --relative-path models/ultralytics/bbox --filename hand_yolov8s.pt
comfy model download --url https://huggingface.co/GritTin/LoraStableDiffusion/resolve/c7766cc3c9b8b4f914932ce27f1cd48f25434636/Eyeful_v2-Paired.pt --relative-path models/ultralytics/bbox --filename Eyeful_v2-Paired.pt
comfy model download --url https://huggingface.co/jags/yolov8_model_segmentation-set/resolve/main/hair_yolov8n-seg_60.pt --relative-path models/ultralytics/segm --filename hair_yolov8n-seg_60.pt
comfy model download --url https://huggingface.co/thisisAce/flux/resolve/main/aidmaRealisticSkin-FLUX-v0.1.safetensors --relative-path models/loras --filename aidmaRealisticSkin-FLUX-v0.1.safetensors
comfy model download --url https://huggingface.co/thisisAce/flux/resolve/main/closeupface-v1.safetensors --relative-path models/loras --filename closeupface-v1.safetensors
comfy model download --url https://huggingface.co/thisisAce/flux/resolve/main/Esc_Makeup.safetensors --relative-path models/loras --filename Esc_Makeup.safetensors
comfy model download --url https://huggingface.co/thisisAce/flux/resolve/main/fluxRealSkin-V2.safetensors --relative-path models/loras --filename fluxRealSkin-V2.safetensors
comfy model download --url https://huggingface.co/thisisAce/flux/resolve/main/luscious_lips_flux_v1.safetensors --relative-path models/loras --filename luscious_lips_flux_v1.safetensors
comfy model download --url "https://huggingface.co/thisisAce/flux/resolve/main/skin%20tone%20style%20v2-step00001500.safetensors" --relative-path models/loras --filename "skin tone style v2-step00001500.safetensors"
comfy model download --url https://huggingface.co/sergio75/Lya/resolve/main/Lya01_ep08_rp01_step800_bf16.safetensors --relative-path models/loras --filename Lya01_ep08_rp01_step800_bf16.safetensors
comfy model download --url https://huggingface.co/f5aiteam/rembg/resolve/0097d75761f9310c2041e7c64bb30846071ccb06/RMBG-1.4.pth --relative-path models/rembg --filename RMBG-1.4.pth
  
# Symlink for /comfyui/models/impact directory
mkdir -p /comfyui/models/impact
ln -sf /comfyui/models/sams/sam_vit_b_01ec64.pth /comfyui/models/impact/sam_vit_b_01ec64.pth
ln -sf /comfyui/models/grounding-dino/GroundingDINO_SwinT_COG.pth /comfyui/models/impact/GroundingDINO_SwinT_COG.pth
ln -sf /comfyui/models/ultralytics/bbox/hand_yolov8s.pt /comfyui/models/impact/hand_yolov8s.pt
ln -sf /comfyui/models/ultralytics/bbox/Eyeful_v2-Paired.pt /comfyui/models/impact/Eyeful_v2-Paired.pt
ln -sf /comfyui/models/ultralytics/bbox/face_yolov8m.pt /comfyui/models/impact/face_yolov8m.pt
ln -sf /comfyui/models/ultralytics/segm/hair_yolov8n-seg_60.pt /comfyui/models/impact/hair_yolov8n-seg_60.pt

echo "All models downloaded and symlinks created."