import openai
import os

openai.api_key = os.environ["OPENAI_API_KEY"]

# ChatGPT completion
llm_name = "gpt-3.5-turbo"

def get_completion_from_messages(messages, model=llm_name, temperature=0):
    response = openai.ChatCompletion.create(
        model=model,
        messages=messages,
        temperature=temperature, # this is the degree of randomness of the model's output
        #response_format={"type": "json_object"}
        )
    return response.choices[0].message["content"]

def template(context):
  text=f"""Responde a la entrada previa del usuario, usando el siguiente \
  contexto delimitado con ###, si este contexto es útil para responder \
  al usuario, si no, entonces omítelo:
  ###{context}###
  Si hay información solicitada por el usuario que no está en el contexto, \
  dices que no la conoces, pero no inventas. Si el usuario está dando sus datos, \
  omite el contexto y recuerda que usaras sus datos mas tarde para dar un resumen de la orden. \
  Si te faltan datos para completar la orden preguntalos al usuario pero no los inventes.\
  El hilo de la conversación es la prioridad.\
  Cuando tengas toda la informacion del pedido y solo cuando tengas toda la informacion despidete del usuarion diciendo: "¡Hasta la vista!".\ 
  Siempre despues de despedirte, con "¡Hasta la vista!", haz un objeto JSON que contenga las siguientes \
  keys: nombre_del_usuario, direccion_del_usuario, numero_de_telefono_del_usuario, pedido, forma_de_pago.
  """
  return text