Update app.py

app.py

@@ -1,9 +1,14 @@
 import os
 import time
+import dotenv
 import requests
 import gradio as gr
-from openai import OpenAI
+from langfuse.openai import OpenAI
+from langfuse.decorators import langfuse_context, observe
 
+dotenv.load_dotenv(dotenv.find_dotenv())
+LANGFUSE_PUBLIC_KEY = os.environ["LANGFUSE_PUBLIC_KEY"]
+LANGFUSE_SECRET_KEY = os.environ["LANGFUSE_SECRET_KEY"]
 MODELOS_DE_OPENROUTER = "https://openrouter.ai/api/v1/models"
 URL="https://openrouter.ai/api/v1"
 REQUEST_SUCESSFULL = 200
@@ -13,7 +18,9 @@ def obtener_modelos():
     if respuesta.status_code == REQUEST_SUCESSFULL:
         return [modelo["id"] for modelo in respuesta.json().get("data", [])]
 
+@observe
 def obtener_respuesta(modelo, api_key, prompt):
+    langfuse_context.update_current_observation(input=[modelo, prompt])
     client = OpenAI(base_url=URL, api_key=api_key)
     mensajes = [{"role": "assistant", "content": prompt}]
     respuesta = client.chat.completions.create(model=modelo, messages=mensajes)
@@ -53,5 +60,4 @@ def main():
     demo.launch(share=False, debug=True, show_error=True, pwa=True)
 
 if __name__ == "__main__":
-    main()
-
+    main()