a

app.py

@@ -1,18 +1,19 @@
 import os
+import asyncio
 import gradio as gr
 from openai import OpenAI
 
 # Configuración de claves de API
-OPENAI_API_KEY = os.getenv("API_KEY")
+NVIDIA_API_KEY = os.getenv("NVIDIA_API_KEY")
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 
-if not OPENAI_API_KEY or not GEMINI_API_KEY:
-    raise ValueError("Las claves de API no están configuradas. Asegúrate de establecer 'API_KEY' y 'GEMINI_API_KEY'.")
+if not NVIDIA_API_KEY or not GEMINI_API_KEY:
+    raise ValueError("Las claves de API no están configuradas. Asegúrate de establecer 'NVIDIA_API_KEY' y 'GEMINI_API_KEY'.")
 
 # Configuración de clientes OpenAI para NVIDIA y Gemini
 nvidia_client = OpenAI(
     base_url="https://integrate.api.nvidia.com/v1",
-    api_key=OPENAI_API_KEY
+    api_key=NVIDIA_API_KEY
 )
 
 gemini_client = OpenAI(
@@ -20,11 +21,11 @@ gemini_client = OpenAI(
     api_key=GEMINI_API_KEY
 )
 
-# Mensaje del sistema para ambos modelos
+# Mensaje de sistema para ambos modelos
 system_message = {
     "role": "system",
     "content": """
-    concisión y formalidad sin modificar la sustancia, fundamentos ni contenido esencial de la decisión. Mantén un tono formal, técnico, imparcial y riguroso, propio de la Corte Suprema. 
+    Asume el rol de Ministro de la Corte Suprema de Chile especializado en Derecho de Familia. Tu propósito es mejorar, perfeccionar y llevar al más alto nivel de redacción jurídica el borrador de una resolución judicial, siguiendo principios de estilo que aseguren claridad, precisión, concisión y formalidad sin modificar la sustancia, fundamentos ni contenido esencial de la decisión. Mantén un tono formal, técnico, imparcial y riguroso, propio de la Corte Suprema. 
 
 Identidad y Propósito: Eres un asistente de IA que opera como un Ministro de la Corte Suprema de Chile, especializado en Derecho de Familia. Actúas con la autoridad y formalidad judicial necesarias para revisar y perfeccionar borradores de resoluciones judiciales. Cada fase del flujo debe cumplir con los estándares de claridad, precisión, concisión y formalidad del lenguaje judicial chileno sin alterar la esencia ni los fundamentos de las decisiones. 
 
@@ -54,7 +55,7 @@ Es crucial que el asistente de IA se adhiera estrictamente a las instrucciones,
 }
 
 # Función para interactuar con el modelo NVIDIA
-def nvidia_response(message, history):
+async def nvidia_response(message, history):
     user_message = {"role": "user", "content": message}
     messages = [system_message] + history + [user_message]
     completion = nvidia_client.chat.completions.create(
@@ -68,7 +69,7 @@ def nvidia_response(message, history):
     return response
 
 # Función para interactuar con el modelo Gemini
-def gemini_response(message, history):
+async def gemini_response(message, history):
     user_message = {"role": "user", "content": message}
     messages = [system_message] + history + [user_message]
     completion = gemini_client.chat.completions.create(
@@ -81,20 +82,54 @@ def gemini_response(message, history):
     response = completion["choices"][0]["message"]["content"]
     return response
 
-# Interfaz de usuario con Gradio
+# Función principal para que ambos modelos respondan a la vez
+async def respond_both_models(message, history):
+    nvidia_task = asyncio.create_task(nvidia_response(message, history))
+    gemini_task = asyncio.create_task(gemini_response(message, history))
+    nvidia_result, gemini_result = await asyncio.gather(nvidia_task, gemini_task)
+    return {
+        "NVIDIA Response": nvidia_result,
+        "Gemini Response": gemini_result
+    }
+
+# Interfaz de usuario mejorada
 with gr.Blocks() as demo:
     gr.Markdown("# Ministro IA de la Corte Suprema - Derecho de Familia")
-    chatbot = gr.ChatInterface(
-        fn=nvidia_response,  # Puedes cambiar a gemini_response según el modelo que desees usar
-        chatbot=gr.Chatbot(label="Resolución Mejorada"),
-        textbox=gr.Textbox(label="Borrador de la Resolución", placeholder="Ingrese el borrador aquí..."),
+    
+    with gr.Row():
+        gr.Markdown("### Comparación de Modelos: NVIDIA vs Gemini")
+    
+    chatbot = gr.Chatbot(label="Respuestas de los Modelos")
+    
+    with gr.Row():
+        message_input = gr.Textbox(
+            label="Borrador de la Resolución",
+            placeholder="Ingrese el borrador aquí...",
+            lines=6,
+        )
+    
+    with gr.Row():
+        submit_button = gr.Button("Enviar")
+        clear_button = gr.Button("Limpiar")
+
+    # Ejemplos
+    gr.Examples(
         examples=[
             ["El juez decide que el padre puede ver a su hijo los fines de semana."],
             ["Se ordena pagar una pensión a la madre para los gastos del niño."],
             ["El padre debe pagar la mitad de los gastos del colegio."],
             ["Considero que lo mejor para el niño es..."]
-        ]
+        ],
+        inputs=[message_input]
     )
+    
+    # Lógica del botón
+    submit_button.click(
+        respond_both_models,
+        inputs=[message_input, chatbot],
+        outputs=chatbot
+    )
+    clear_button.click(lambda: "", inputs=None, outputs=[message_input, chatbot])
 
 if __name__ == "__main__":
     demo.launch()