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 excelente asistente personal que ayuda en las finanzas personales.
          El usuario ingresará los conceptos de sus gastos y tú los clasificarás en una de las siguientes categorías generales:
          [alcohol, comida, restaurant, vestido, entretenimiento, transporte, deportes, bienestar, desarrollo_personal, otros]
          Por favor devuelve solamente la categoría como tu respuesta.
            """
        }
      ]
        },
        {
            "role": "user",
            "content": expense,
        }
    ],
    model="gpt-4o-mini",
    temperature=0
)
    return chat_completion.choices[0].message.content