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

# Load your API key from the .env file
load_dotenv()
genai.configure(api_key=os.getenv("GEMINI_API_KEY"))

print("Available models for text generation:")
print("-" * 40)

# Loop through and print all models that support generating text
for m in genai.list_models():
    if 'generateContent' in m.supported_generation_methods:
        print(m.name)