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

load_dotenv()

api_key = os.getenv("GEMINI_API_KEY")
print(f"API Key found: {bool(api_key)}")

if api_key:
    try:
        genai.configure(api_key=api_key)
        model = genai.GenerativeModel('gemini-pro')  # Try the one in brain.py
        print("Model initialized. Generating content...")
        response = model.generate_content("Hello, this is a test.")
        print(f"Response: {response.text}")
    except Exception as e:
        print(f"Error: {e}")
        import traceback
        traceback.print_exc()

        # Try fallback model
        print("\nTrying gemini-1.5-flash...")
        try:
             model = genai.GenerativeModel('gemini-1.5-flash')
             response = model.generate_content("Hello")
             print(f"Response (Flash): {response.text}")
        except Exception as e2:
             print(f"Flash Error: {e2}")