Sam Kumar
d4931cf5dd
in diagnosing integration issues with ComfyUI-SwissArmyKnife 🔧 (docker-compose.yml): add CIVITAI_API_KEY environment variable to docker-compose for CivitAI service integration 📝 (CIVITAI_API_KEY_WIDGET.md): document implementation details for CivitAI API key widget in LoRAInfoExtractor node, including usage examples and security considerations ✨ (MULTIPLE_LORA_SUPPORT.md): enhance LoRAInfoExtractor to support multiple LoRAs from WANVIDLORA list structure, improving processing and output format ♻️ (civitai_service.py): refactor CivitAIService to accept API key as parameter, allowing for better flexibility and debugging ♻️ (nodes.py): update LoRAInfoExtractor to support multiple LoRAs and CivitAI API key input, improving overall functionality and user experience
141 lines
5.1 KiB
Python
141 lines
5.1 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
CivitAI Integration Debug Helper
|
|
|
|
This script helps debug CivitAI integration issues in ComfyUI-SwissArmyKnife.
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
sys.path.append(os.path.dirname(__file__))
|
|
|
|
def check_environment():
|
|
"""Check environment setup for CivitAI integration"""
|
|
print("🔍 CivitAI Integration Debug Helper")
|
|
print("=" * 50)
|
|
|
|
# Check current directory
|
|
cwd = os.getcwd()
|
|
print(f"📁 Current directory: {cwd}")
|
|
|
|
# Check for .env files
|
|
env_files = ['.env', '.env.local', '.env.production']
|
|
print(f"\n📄 Environment Files:")
|
|
|
|
env_file_found = False
|
|
civitai_config_found = False
|
|
|
|
for env_file in env_files:
|
|
if os.path.exists(env_file):
|
|
env_file_found = True
|
|
print(f" ✅ Found: {env_file}")
|
|
try:
|
|
with open(env_file, 'r') as f:
|
|
content = f.read()
|
|
if 'CIVITAI' in content.upper():
|
|
civitai_config_found = True
|
|
print(f" ✅ Contains CivitAI configuration")
|
|
# Show the lines with CIVITAI (masked)
|
|
for line_num, line in enumerate(content.split('\n'), 1):
|
|
if 'CIVITAI' in line.upper():
|
|
if '=' in line and not line.strip().startswith('#'):
|
|
key, value = line.split('=', 1)
|
|
masked_value = value[:8] + '...' + value[-4:] if len(value) > 12 else '***'
|
|
print(f" 📝 Line {line_num}: {key}={masked_value}")
|
|
else:
|
|
print(f" 📝 Line {line_num}: {line}")
|
|
else:
|
|
print(f" ❌ No CivitAI configuration found")
|
|
except Exception as e:
|
|
print(f" ❌ Error reading file: {e}")
|
|
else:
|
|
print(f" ❌ Not found: {env_file}")
|
|
|
|
# Check environment variables
|
|
print(f"\n🌍 Environment Variables:")
|
|
civitai_vars = [k for k in os.environ.keys() if 'civit' in k.lower()]
|
|
if civitai_vars:
|
|
print(f" ✅ CivitAI-related variables found: {len(civitai_vars)}")
|
|
for var in civitai_vars:
|
|
value = os.environ[var]
|
|
masked_value = value[:8] + '...' + value[-4:] if len(value) > 12 else '***'
|
|
print(f" 📝 {var}={masked_value}")
|
|
else:
|
|
print(f" ❌ No CivitAI-related environment variables found")
|
|
|
|
# Test service initialization
|
|
print(f"\n🔧 Service Test:")
|
|
try:
|
|
from utils.civitai_service import CivitAIService
|
|
service = CivitAIService()
|
|
if service.api_key:
|
|
print(f" ✅ CivitAI service initialized with API key")
|
|
else:
|
|
print(f" ❌ CivitAI service has no API key")
|
|
except Exception as e:
|
|
print(f" ❌ Error initializing service: {e}")
|
|
|
|
# Recommendations
|
|
print(f"\n💡 Recommendations:")
|
|
|
|
if not env_file_found:
|
|
print(f" 1. Create a .env file in your project root:")
|
|
print(f" echo 'CIVITAI_API_KEY=your_api_key_here' > .env")
|
|
elif not civitai_config_found:
|
|
print(f" 1. Add CivitAI API key to your .env file:")
|
|
print(f" echo 'CIVITAI_API_KEY=your_api_key_here' >> .env")
|
|
|
|
if not civitai_vars:
|
|
print(f" 2. Get your CivitAI API key from: https://civitai.com/user/account")
|
|
print(f" 3. Set the environment variable or add to .env file")
|
|
|
|
print(f" 4. Restart ComfyUI after setting the API key")
|
|
print(f" 5. Test with a known LoRA file from CivitAI")
|
|
|
|
def test_api_key(api_key):
|
|
"""Test a CivitAI API key"""
|
|
if not api_key:
|
|
print("❌ No API key provided")
|
|
return False
|
|
|
|
print(f"🧪 Testing API key: {api_key[:8]}...{api_key[-4:]}")
|
|
|
|
try:
|
|
import requests
|
|
url = "https://civitai.com/api/v1/models"
|
|
headers = {"Authorization": f"Bearer {api_key}"}
|
|
|
|
print(f"📡 Making test request to CivitAI API...")
|
|
response = requests.get(url, headers=headers, timeout=10, params={"limit": 1})
|
|
|
|
print(f"📊 Response status: {response.status_code}")
|
|
|
|
if response.status_code == 200:
|
|
print(f"✅ API key is valid!")
|
|
return True
|
|
elif response.status_code == 401:
|
|
print(f"❌ API key is invalid or expired")
|
|
return False
|
|
else:
|
|
print(f"⚠️ Unexpected response: {response.status_code}")
|
|
print(f"Response: {response.text[:200]}")
|
|
return False
|
|
|
|
except Exception as e:
|
|
print(f"❌ Error testing API key: {e}")
|
|
return False
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) > 1:
|
|
# Test provided API key
|
|
api_key = sys.argv[1]
|
|
test_api_key(api_key)
|
|
else:
|
|
# Check environment
|
|
check_environment()
|
|
|
|
# Offer to test API key
|
|
api_key = os.environ.get("CIVITAI_API_KEY")
|
|
if api_key:
|
|
print(f"\n" + "="*50)
|
|
test_api_key(api_key) |