from openai import OpenAI
import os
my_key = os.environ.get('MY_OPENAI_KEY')
client = OpenAI(
    api_key= my_key
)

def expense_classifier(expense):
    chat_completion = client.chat.completions.create(
    messages=[
        {   "role": "system",
            "content": [
        {
          "type": "text",
          "text": """Eres un asistente contable muy eficiente.
          El usuario ingresará un gasto y tú lo clasificaráas en una categoria general de las siguientes:
          [insumo, gasto fijo, sueldo].
          Por favor sólo la categoria como respuesta.
          
            """
        }
      ]
        },
        {
            "role": "user",
            "content": expense,
        }
    ],
    model="gpt-4o-mini",
    temperature=0
)
    return chat_completion.choices[0].message.content