check_gemini_models.py

"""
Check available Gemini models for your API key
"""
import os
from dotenv import load_dotenv

load_dotenv()

try:
    import google.generativeai as genai
    
    api_key = os.getenv('GEMINI_API_KEY')
    if not api_key:
        print("❌ GEMINI_API_KEY not found in .env file")
        exit(1)
    
    genai.configure(api_key=api_key)
    
    print("✅ Successfully connected to Gemini API")
    print("\n📋 Available models that support image generation/vision:")
    print("-" * 60)
    
    for model in genai.list_models():
        # Check if model supports vision/content generation
        if 'generateContent' in model.supported_generation_methods:
            print(f"\n✓ {model.name}")
            print(f"  Display name: {model.display_name}")
            print(f"  Supported methods: {', '.join(model.supported_generation_methods)}")
            
    print("\n" + "-" * 60)
    print("\n💡 Recommendation: Use one of the models listed above")
    print("   Update gemini_classification_service.py if needed")
    
except ImportError:
    print("❌ google-generativeai not installed")
    print("Run: pip install google-generativeai")
except Exception as e:
    print(f"❌ Error: {e}")