import os, openai,string,csv
import pandas as pd
from utilities import constants, api_keys, clean_text

openai.api_key = api_keys.APIKeys().get_key('OPENAI_API_KEY')
messages=[]


def Completion(messages):
    print("\n\n***Completion: messages: ", messages)
    
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo", 
        messages=messages
    )
    cleaned_text= clean_text.RemoveRole(response["choices"][0]["message"]["content"])
    return cleaned_text

def ManageChatGame(query, context):
    global messages
    
    print("ManageChatGame: query: ", query)
    print("ManageChatGame: context: ", context)

    context_text = f"Using this context: {context}"
    messages.append({"role": "system", "content": context_text})
    
    prompt_text = "You are Jimmy a software engineer working at Slalom Build on Generative AI. Your role is to host a chat-based game session and keep things moving. Do not reveal you are a bot. Keep your questions going and keep the user engaged. You provide the questions and answer as you can. Makes stuff up if needed. Keep your answers short"
    messages.append({"role": "system", "content": prompt_text})

    query_text = f"This is the users response: {query}"
    messages.append({"role": "user", "content": query_text})

    print("going to completion: messages: ", messages)
    system_message = Completion(messages)
    messages.append({"role": "assistant", "content": system_message})

    print("system_message: ")
    print(system_message)
    df = pd.DataFrame(messages)
    return system_message, df