#!/usr/bin/env python3
"""
Quick test script to verify OpenAI API key works
"""
import os
from dotenv import load_dotenv

# Load environment variables
load_dotenv()

def test_openai_key():
    try:
        # Get API key
        api_key = os.getenv("OPENAI_API_KEY")
        if not api_key:
            print("❌ No OPENAI_API_KEY found in environment variables")
            return False
            
        print(f"✅ Found API key: {api_key[:20]}...")
        
        # Test with OpenAI v1.0+ syntax
        from openai import OpenAI
        
        client = OpenAI(api_key=api_key)
        
        print("🔄 Testing API connection...")
        
        # Simple test call
        response = client.chat.completions.create(
            model="gpt-3.5-turbo",
            messages=[
                {"role": "user", "content": "Say 'Hello, API is working!'"}
            ],
            max_tokens=20
        )
        
        print("✅ OpenAI API test successful!")
        print(f"Response: {response.choices[0].message.content}")
        return True
        
    except Exception as e:
        print(f"❌ OpenAI API test failed: {e}")
        return False

if __name__ == "__main__":
    print("Testing OpenAI API key...")
    success = test_openai_key()
    if success:
        print("\n🎉 Your OpenAI API key is working correctly!")
    else:
        print("\n💡 Please check your API key and try again.")