#!/bin/bash
set -euo pipefail

TMP_CACHE=$(mktemp -d)
export HF_HUB_CACHE="$TMP_CACHE"
TMP=$(mktemp -d)
MODELS_DIR="ComfyUI/models"
LORAS_DIR="$MODELS_DIR/loras/ltx23"

hf_cli_install() {
    if ! command -v hf &> /dev/null; then
        echo "hf not found, installing..."
        curl -LsSf https://hf.co/cli/install.sh | bash
    else
        echo "hf is already installed."
    fi
}

hf_download_flat() {
    local repo="$1" file="$2" dest="$3"
    hf download "$repo" "$file" --local-dir "$TMP"
    mkdir -p "$dest"
    mv "$TMP/$file" "$dest/$(basename "$file")"
}

hf_cli_install 
#hf_download_flat "CornLogic/10EROS_1.4_Int8_ConvRot" "10Eros_v1.4_DMD_int8_convrot.safetensors" "$MODELS_DIR/checkpoints"
# hf_download_flat "TenStrip/LTX2.3-10Eros" "10Eros_v1.4_bf16.safetensors" "$MODELS_DIR/checkpoints"
hf_download_flat "TenStrip/LTX2.3-10Eros" "text_encoders/gemma-3-12b-it-ablit-norms-biproj-fp8mixed.safetensors" "$MODELS_DIR/text_encoders"
hf_download_flat "Lightricks/LTX-2.3" "ltx-2.3-spatial-upscaler-x2-1.1.safetensors" "$MODELS_DIR/latent_upscale_models"
hf_download_flat "Lightricks/LTX-2.3" "ltx-2.3-temporal-upscaler-x2-1.0.safetensors" "$MODELS_DIR/latent_upscale_models"
#hf_download_flat "TenStrip/LTX2.3_Distilled_Lora_1.1_Experiments" "ltx-2.3-22b-distilled-lora-1.1_fro90_ceil72_condsafe.safetensors" "$LORAS_DIR"

hf_download_flat "Kijai/LTX2.3_comfy" "diffusion_models/ltx-2.3-22b-dev_transformer_only_int8_convrot.safetensors" "$MODELS_DIR/diffusion_models"

hf_download_flat "Kijai/LTX2.3_comfy" "vae/LTX23_audio_vae_bf16.safetensors" "$MODELS_DIR/vae"
hf_download_flat "Kijai/LTX2.3_comfy" "vae/LTX23_video_vae_bf16.safetensors" "$MODELS_DIR/vae"
hf_download_flat "Kijai/LTX2.3_comfy" "vae/taeltx2_3.safetensors" "$MODELS_DIR/vae"
hf_download_flat "Kijai/LTX2.3_comfy" "text_encoders/ltx-2.3_text_projection_bf16.safetensors" "$MODELS_DIR/text_encoders"

hf_download_flat "TenStrip/LTX2.3_DMD_Lora" "LTX2.3_DMD_reshaped_r256.safetensors" "$LORAS_DIR"
hf_download_flat "Kijai/LTX2.3_comfy" "loras/LTX-2.3-OmniNFT-RL-Lora_bf16.safetensors" "$LORAS_DIR"
hf_download_flat "TenStrip/LTX2.3_JoyAI_Lora_Extracted" "JoyAI-Echo-content_r256.safetensors" "$LORAS_DIR"

hf_download_flat "maximsobolev275/LTX-10Eros-LoRA-r768" "LTX_10Eros-v14_LoRA_fro99-avgrank104.safetensors" "$LORAS_DIR"
hf_download_flat "maximsobolev275/LTX2.3_DMD-10eros-resized" "LTX2.3_DMD_fro99-avgrank47.safetensors" "$LORAS_DIR"
hf_download_flat "maximsobolev275/LTX-PinkCherry-LoRA-r768" "LTX_PinkCherry-v15_LoRA_fro99-avgrank56.safetensors" "$LORAS_DIR"
hf_download_flat "maximsobolev275/LTX-SulphurExperimental-LoRA-Optimized" "LTX_SulphurEXP_LoRA_fro99-avgrank105.safetensors" "$LORAS_DIR"

hf_download_flat "cdkkkk/setup" "ltx2/DR34ML4Y_LT3X_V3.safetensors" "$LORAS_DIR"
hf_download_flat "cdkkkk/setup" "ltx2/pornstar_kissing_v1_rank16_000002000.safetensors" "$LORAS_DIR"
hf_download_flat "cdkkkk/setup" "ltx2/ltxdeepthroat_v01.safetensors" "$LORAS_DIR"
hf_download_flat "cdkkkk/setup" "ltx2/pussyjob_sulphur_v1.1.safetensors" "$LORAS_DIR"
hf_download_flat "cdkkkk/setup" "ltx2/Kissing_Lora_2000.safetensors" "$LORAS_DIR"
hf_download_flat "cdkkkk/setup" "ltx2/LTX2.3_reasoning_Sulphur-2_I2V_V4.safetensors" "$LORAS_DIR"
hf_download_flat "cdkkkk/setup" "ltx2/LTX2.3-Twerking.safetensors" "$LORAS_DIR"
hf_download_flat "cdkkkk/setup" "ltx2/ltx23-cunilingus-32epoc-sulphured-audio-k3nk.safetensors" "$LORAS_DIR"


rm -rf "$TMP_CACHE"
rm -rf "$TMP"