import openai
import os
from dotenv import load_dotenv

load_dotenv()
openai.api_key = os.getenv("OPENAI_API_KEY")

def ask_ai(question, history=[]):
    """
    Ask a question to the AI teaching assistant.
    
    Parameters:
        question (str): User's question.
        history (list): Previous conversation history as list of dicts (optional).
    
    Returns:
        answer (str): AI's response.
    """
    # Combine chat history with new question
    messages = history + [{"role": "user", "content": question}]
    
    try:
        response = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",  # or "gpt-4" if you have access
            messages=messages,
            temperature=0.7,
            max_tokens=500
        )
        answer = response['choices'][0]['message']['content']
        return answer, messages + [{"role": "assistant", "content": answer}]
    except Exception as e:
        return f"Error: {str(e)}", history