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

# Carica la chiave API dal file .env
load_dotenv()

api_key = os.getenv("GOOGLE_API_KEY")

if not api_key:
    print("❌ ERRORE: Chiave API non trovata nel file .env!")
    exit()

print(f"🔑 Chiave trovata: {api_key[:5]}... (configurazione in corso)")

# Configura l'SDK
genai.configure(api_key=api_key)

print("\n--- ELENCO MODELLI DISPONIBILI PER TE ---")
try:
    count = 0
    # Chiede a Google la lista dei modelli
    for m in genai.list_models():
        # Filtriamo solo quelli che generano testo (chat)
        if 'generateContent' in m.supported_generation_methods:
            print(f"✅ Nome Modello: {m.name}")
            print(f"   Display Name: {m.display_name}")
            print(f"   Input Token Limit: {m.input_token_limit}")
            print("-" * 30)
            count += 1
            
    if count == 0:
        print("⚠️ Nessun modello trovato abilitato per 'generateContent'.")
        
except Exception as e:
    print(f"❌ Errore durante la connessione: {e}")