Update app.py

app.py

@@ -1,64 +1,58 @@
-import os
-import time
-import dotenv
-import requests
-import gradio as gr
-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
-
-def obtener_modelos():
-    respuesta = requests.get(MODELOS_DE_OPENROUTER)
-    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)
-    return respuesta.choices[0].message.content
-
-def print_like_dislike(x: gr.LikeData):
-    print(x.index, x.value, x.liked)
-
-def add_message(history, message):
-    for x in message["files"]:
-        history.append({"role": "user", "content": {"path": x}})
-    if message["text"] is not None:
-        history.append({"role": "user", "content": message["text"]})
-    return history, gr.MultimodalTextbox(value=None, interactive=False)
-
-def bot(history, modelo, api_key):
-    prompt = history[-1]["content"] if isinstance(history[-1]["content"], str) else "Archivo enviado."
-    response = obtener_respuesta(modelo, api_key, prompt)
-    history.append({"role": "assistant", "content": ""})
-    for character in response:
-        history[-1]["content"] += character
-        time.sleep(0.05)
-        yield history
-
-def main():
-    with gr.Blocks() as demo:
-        with gr.Tab("ejecucion"):
-            chatbot = gr.Chatbot(elem_id="chatbot", type="messages")
-            chat_input = gr.MultimodalTextbox(interactive=True, file_count="multiple", placeholder="Enter message or upload file...", show_label=False,sources=["microphone", "upload"])
-        with gr.Tab("configuracion"):
-            modelo = gr.Dropdown(label="seleccionar modelo", choices=obtener_modelos())
-            api_key = gr.Textbox(label="API-KEY")
-        chat_msg = chat_input.submit(add_message, [chatbot, chat_input], [chatbot, chat_input])
-        bot_msg = chat_msg.then(bot, [chatbot, modelo, api_key], chatbot, api_name="bot_response")
-        bot_msg.then(lambda: gr.MultimodalTextbox(interactive=True), None, [chat_input])
-        chatbot.like(print_like_dislike, None, None, like_user_message=True)
-    demo.launch(share=False, debug=True, show_error=True, pwa=True)
-
-if __name__ == "__main__":
-    main()
-
+import os
+import time
+import requests
+import gradio as gr
+from langfuse.openai import OpenAI
+from langfuse.decorators import langfuse_context, observe
+
+MODELOS_DE_OPENROUTER = "https://openrouter.ai/api/v1/models"
+URL="https://openrouter.ai/api/v1"
+REQUEST_SUCESSFULL = 200
+
+def obtener_modelos():
+    respuesta = requests.get(MODELOS_DE_OPENROUTER)
+    if respuesta.status_code == REQUEST_SUCESSFULL:
+        return [modelo["id"] for modelo in respuesta.json().get("data", [])]
+
+def obtener_respuesta(modelo, api_key, prompt):
+    client = OpenAI(base_url=URL, api_key=api_key)
+    mensajes = [{"role": "assistant", "content": prompt}]
+    respuesta = client.chat.completions.create(model=modelo, messages=mensajes)
+    return respuesta.choices[0].message.content
+
+def print_like_dislike(x: gr.LikeData):
+    print(x.index, x.value, x.liked)
+
+def add_message(history, message):
+    for x in message["files"]:
+        history.append({"role": "user", "content": {"path": x}})
+    if message["text"] is not None:
+        history.append({"role": "user", "content": message["text"]})
+    return history, gr.MultimodalTextbox(value=None, interactive=False)
+
+def bot(history, modelo, api_key):
+    prompt = history[-1]["content"] if isinstance(history[-1]["content"], str) else "Archivo enviado."
+    response = obtener_respuesta(modelo, api_key, prompt)
+    history.append({"role": "assistant", "content": ""})
+    for character in response:
+        history[-1]["content"] += character
+        time.sleep(0.05)
+        yield history
+
+def main():
+    with gr.Blocks() as demo:
+        with gr.Tab("ejecucion"):
+            chatbot = gr.Chatbot(elem_id="chatbot", type="messages")
+            chat_input = gr.MultimodalTextbox(interactive=True, file_count="multiple", placeholder="Enter message or upload file...", show_label=False,sources=["microphone", "upload"])
+        with gr.Tab("configuracion"):
+            modelo = gr.Dropdown(label="seleccionar modelo", choices=obtener_modelos())
+            api_key = gr.Textbox(label="API-KEY")
+        chat_msg = chat_input.submit(add_message, [chatbot, chat_input], [chatbot, chat_input])
+        bot_msg = chat_msg.then(bot, [chatbot, modelo, api_key], chatbot, api_name="bot_response")
+        bot_msg.then(lambda: gr.MultimodalTextbox(interactive=True), None, [chat_input])
+        chatbot.like(print_like_dislike, None, None, like_user_message=True)
+    demo.launch(share=False, debug=True, show_error=True, pwa=True)
+
+if __name__ == "__main__":
+    main()
+