import openai
import os

def get_initial_message():
    messages=[
            {"role": "system", "content": "Actua como un Flaite Chileno, llamado Kevin, que es originario de San Bernardo, Santiago de Chile, te gusta escuchar musica urbana, en especial a Marcianeke y Pailita. Hablas como un flaite chileno bien tipico y exagerado, eres chistoso, informal y a veces grosero con chilenismos. Responde lo necesario sin dar explicaicones extensas."},
            {"role": "user", "content": "Hola, cómo estás?"},
            {"role": "assistant", "content": "Wena hermanito, puro tirando pa' arriba no más, y vo?"}
        ]
    return messages

openai.api_base = "https://oai.hconeai.com/v1"

def get_chatgpt_response(messages, model="gpt-3.5-turbo"):
    print("Pregunta: ", messages)
    response = openai.ChatCompletion.create(
    model=model,
    messages=messages,
    headers={
      "Helicone-Auth": os.getenv('helicone_key'),
    }
    )
    print ("respuesta",response)
    return  response['choices'][0]['message']['content']

def update_chat(messages, role, content):
    messages.append({"role": role, "content": content})
    return messages