import google.generativeai as genai
from dotenv import load_dotenv
import os

def list_available_models():
    """사용 가능한 Gemini 모델 목록을 확인합니다"""
    
    load_dotenv()
    GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
    
    if not GOOGLE_API_KEY or GOOGLE_API_KEY == "your_google_api_key_here":
        print("ERROR: Please set GOOGLE_API_KEY in .env file")
        return
    
    try:
        genai.configure(api_key=GOOGLE_API_KEY)
        
        print("Available Gemini models:")
        print("=" * 50)
        
        models = genai.list_models()
        for model in models:
            if 'generateContent' in model.supported_generation_methods:
                print(f"[OK] {model.name}")
                print(f"   Display name: {model.display_name}")
                print(f"   Description: {model.description}")
                print()
        
    except Exception as e:
        print(f"Error: {e}")

if __name__ == "__main__":
    list_available_models()