#!/usr/bin/env python3
"""
Check available checkpoints in ComfyUI
"""
import requests
import json

COMFYUI_URL = "http://127.0.0.1:8000"

print("=" * 70)
print("🔍 ComfyUI - Available Checkpoints")
print("=" * 70)

# Get object info to see available checkpoints
try:
    response = requests.get(f"{COMFYUI_URL}/object_info")
    if response.status_code == 200:
        data = response.json()
        
        # Find CheckpointLoaderSimple
        if "CheckpointLoaderSimple" in data:
            checkpoint_info = data["CheckpointLoaderSimple"]
            if "input" in checkpoint_info and "required" in checkpoint_info["input"]:
                ckpt_list = checkpoint_info["input"]["required"].get("ckpt_name", [[]])[0]
                
                print(f"\n✅ Available Checkpoints ({len(ckpt_list)}):")
                for i, ckpt in enumerate(ckpt_list, 1):
                    print(f"   {i}. {ckpt}")
                
                if not ckpt_list:
                    print("\n❌ NO CHECKPOINTS FOUND!")
                    print("You need to download a model to:")
                    print("/Users/davidkotnik/Documents/ComfyUI/models/checkpoints/")
                    print("\nRecommended models:")
                    print("- SD 1.5: https://huggingface.co/runwayml/stable-diffusion-v1-5")
                    print("- SDXL: https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0")
            else:
                print("❌ Could not find checkpoint list structure")
        else:
            print("❌ CheckpointLoaderSimple not found in object_info")
            
    else:
        print(f"❌ Error: {response.status_code}")
        
except Exception as e:
    print(f"❌ Exception: {e}")

# Also check for other info
print("\n" + "=" * 70)
print("📦 Checking /embeddings endpoint...")
try:
    response = requests.get(f"{COMFYUI_URL}/embeddings")
    if response.status_code == 200:
        embeddings = response.json()
        print(f"✅ Found {len(embeddings)} embeddings")
except Exception as e:
    print(f"⚠️  {e}")