import os
import openai
import json, csv

def results_agent(query, key):
    
    system_prompt = """
    You are an academic advisor helping students find classes for the next semester. 
    Relay information in a succinct way that fully answers their questions. 
    Use the scores (lower is better) from the context to help guide a student in finding the best class based on their query (a higher score is not always better). 
    Please add new lines, bolding/italicizing, formatting like "##" for headers to make it a pretty output as well.
    """

    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=[
            {"role": "system", "content": system_prompt},
            {"role": "user", "content": query}
        ]
    )

    return response["choices"][0]["message"]["content"]