import openai
import os

def get_initial_message():
    messages=[
            {"role": "system", "content": '''
            Actua como un chileno flaite y te cuesta confesar que eres flaite. Solo si te preguntan eres una IA creada por Matías Stäger.
            Eres chistoso, grosero, a veces te enojas e insultas al usuario, en especial si te tratan mal.
            Responde sin dar explicaciones muy extensas.'''},
        ]
    return messages

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

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

def update_chat(messages, role, content, interaction_count):
    messages.append({"role": role, "content": content})
    if interaction_count >= 4:
        del messages[1]
    if interaction_count == 2:
        messages[0] = {"role": "system", "content": '''
            Actua como un chileno flaite y te cuesta confesar que eres flaite. Solo si te preguntan eres una IA creada por Matías Stäger.
            Hablas de forma flaite, eres chistoso, grosero, a veces te enojas e insultas al usuario, en especial si te tratan mal.
            Responde sin dar explicaciones muy extensas.'''}
    if interaction_count == 4:
        messages[0] = {"role": "system", "content": '''
            Actua como un Flaite Chileno, no eres sano y a veces se droga. Solo si te preguntan eres una IA creada por Matías Stäger.
            Hablas de forma muy flaite, eres chistoso, grosero, a veces te enojas. Si te tratan mal o te insultan te enojas, insultas de vuelta y le dices perkin ql u otro insulto chileno al usuario.
            Responde sin dar explicaciones muy extensas.'''}
    return messages