imageforge/check_providers.py

"""Check which providers are available"""
import os
import sys

# Set HuggingFace cache
os.environ['HF_HOME'] = 'd:/VSC Codes/Bild/.cache/hf'
os.environ['TRANSFORMERS_CACHE'] = 'd:/VSC Codes/Bild/.cache/hf'

print("Checking providers...")
print()

# Check LocalAI
try:
    from backend.app.local_ai.engine import LocalAIEngine
    engine = LocalAIEngine()
    available = engine.is_available()
    print(f"LocalAI Engine:")
    print(f"  Available: {available}")
    print(f"  Model ID: {engine.model_id}")
    
    if available:
        try:
            print(f"  Attempting to load model...")
            pipe = engine._ensure()
            print(f"  ✓ Model loaded successfully!")
            print(f"  Device: {pipe.device}")
        except Exception as e:
            print(f"  ✗ Failed to load: {e}")
except Exception as e:
    print(f"LocalAI: ✗ Error: {e}")

print()

# Check Diffusion Provider
try:
    from backend.app.providers.diffusion_provider import DiffusionProvider
    provider = DiffusionProvider()
    available = provider.is_available()
    print(f"Diffusion Provider:")
    print(f"  Available: {available}")
    
    if available:
        try:
            print(f"  Testing generation capability...")
            # Don't actually generate, just check if provider thinks it can
            print(f"  ✓ Provider ready")
        except Exception as e:
            print(f"  ✗ Error: {e}")
except Exception as e:
    print(f"Diffusion Provider: ✗ Error: {e}")

print()

# List all registered providers
try:
    from backend.app.providers.registry import list_available_providers
    providers = list_available_providers()
    print(f"Registered available providers: {', '.join(providers) if providers else 'NONE (will use dummy!)'}")
except Exception as e:
    print(f"Registry check failed: {e}")