import os
from openai import OpenAI
from dotenv import load_dotenv

# Load environment variables
load_dotenv()

# OpenAI configuration
api_key = os.environ.get("OPENAI_API_KEY")
if not api_key:
    raise ValueError("OPENAI_API_KEY environment variable is not set. Please set it and try again.")

client = OpenAI(api_key=api_key)
OPENAI_MODEL = "gpt-3.5-turbo"

def get_ai_response(messages):
    """
    Get a response from the AI model using the OpenAI client.
    :param messages: List of message dictionaries as expected by OpenAI API
    :return: The content of the AI's response
    """
    try:
        response = client.chat.completions.create(
            model=OPENAI_MODEL,
            messages=messages
        )
        return response.choices[0].message.content
    except Exception as e:
        print(f"Error getting AI response: {str(e)}")
        return None