Update app.py

app.py

@@ -1,8 +1,14 @@
+import os
 import time
-import requests
+import requests 
 import gradio as gr
-from openai import OpenAI
+from langfuse.decorators import observe, langfuse_context
+from langfuse.openai import OpenAI
 
+# set Langfuse api keys and host address
+LANGFUSE_PUBLIC_KEY = os.getenv('LANGFUSE_PUBLIC_KEY')
+LANGFUSE_SECRET_KEY = os.getenv('LANGFUSE_SECRET_KEY')
+LANGFUSE_HOST = os.getenv('LANGFUSE_HOST')
 MODELOS_DE_OPENROUTER = "https://openrouter.ai/api/v1/models"
 URL="https://openrouter.ai/api/v1"
 REQUEST_SUCESSFULL = 200
@@ -12,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
     client = OpenAI(base_url=URL, api_key=api_key)
     mensajes = [{"role": "assistant", "content": prompt}]
     respuesta = client.chat.completions.create(model=modelo, messages=mensajes)