#!/bin/bash

BASE_DIR="/home/wuyingwen/ai_projects/ComfyUI/models"

declare -A paths=(
  ["Checkpoints"]="checkpoints"
  ["VAE Models"]="vae"
  ["CLIP Models"]="clip"
  ["Lora Models"]="loras"
  ["ControlNet Models"]="controlnet"
  ["Upscale Models"]="upscale_models"
  ["Embeddings"]="embeddings"
  ["Hypernetworks"]="hypernetworks"
  ["Diffusion Models"]="diffusion_models"
  ["Text Encoders"]="text_encoders"
)

echo "=== ComfyUI 已识别模型列表 ==="
for key in "${!paths[@]}"; do
  echo ""
  echo "== $key =="
  dir="$BASE_DIR/${paths[$key]}"
  if [ -d "$dir" ]; then
    files=$(ls "$dir" 2>/dev/null)
    if [ -n "$files" ]; then
      echo "$files"
    else
      echo "(无文件)"
    fi
  else
    echo "(目录不存在)"
  fi
done