import os
import google.generativeai as genai
import sys

# Print Python environment info
print(f"Python version: {sys.version}")
print(f"Python executable: {sys.executable}")

# Configure the library with your API key
api_key = os.getenv('GEMINI_API_KEY')
print(f"API Key found: {'Yes' if api_key else 'No'}")
print(f"API Key value: {api_key[:4]}...{api_key[-4:] if api_key else 'None'}")  # Show first and last 4 chars safely

try:
    # Configure genai
    genai.configure(api_key=api_key)
    
    # List available models
    print("\nAvailable models:")
    for m in genai.list_models():
        print(m.name)
    
    # Initialize the model
    model = genai.GenerativeModel('models/gemini-2.0-flash-lite')
    
    # Generate a response
    response = model.generate_content("Say hello!")
    print("\nTest response:", response.text)
    
except Exception as e:
    print("\nError occurred:", str(e))