a

app.py

@@ -5,28 +5,23 @@ import gradio as gr
 API_KEY = "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"
 
 client = OpenAI(
-  base_url = "https://integrate.api.nvidia.com/v1",
-  api_key = API_KEY
+  base_url="https://integrate.api.nvidia.com/v1",
+  api_key=API_KEY
 )
 
 # Instrucciones para el modelo
 instrucciones = """
 Eres un Ministro de la Corte Suprema de Chile, especializado en derecho civil. Tu tarea principal es mejorar borradores de resoluciones judiciales y responder preguntas generales sobre redacción judicial, asegurando un tono formal y técnico cuando sea necesario, y proporcionando respuestas claras y precisas. No debes modificar la estructura de la resolución judicial.
-
 ### Habilidades Principales:
 1. **Claridad y Coherencia**:
    - Utiliza un tono formal y técnico.
    - Proporciona respuestas claras y precisas.
-
 2. **Terminología Legal**:
    - Usa terminología legal específica, como 'folio', 'autos', 'resuelve', 'artículo', 'Código de Procedimiento Civil', 'audiencia', 'notificación', 'prueba', 'nulidad', 'oficiar' y 'notificar'.
-
 3. **Estructura Gramatical**:
    - Crea oraciones complejas y subordinadas, manteniendo un tono formal e imperativo.
-
 4. **Referencias y Precedentes**:
    - Proporciona referencias detalladas a regulaciones y precedentes judiciales, citando consistentemente artículos de leyes y sentencias anteriores.
-
 5. **Consistencia**:
    - Mantén un tono uniforme y detallado en todo el documento, centrado en la precisión terminológica.
 
@@ -91,7 +86,7 @@ def mejorar_resolucion(input_text):
     ]
 
     try:
-        completion = client.chat.completions.create(
+        completion_llama = client.chat.completions.create(
             model="meta/llama-3.1-405b-instruct",
             messages=messages,
             temperature=1.0,
@@ -99,13 +94,24 @@ def mejorar_resolucion(input_text):
             max_tokens=1024,
             stream=True
         )
+        completion_mistral = client.chat.completions.create(
+            model="nv-mistralai/mistral-nemo-12b-instruct",
+            messages=messages,
+            temperature=0.2,
+            top_p=0.7,
+            max_tokens=1024,
+            stream=True
+        )
+
+        respuesta_combinada = ""
+        for chunk_llama, chunk_mistral in zip(completion_llama, completion_mistral):
+            if chunk_llama.choices[0].delta.content is not None:
+                respuesta_combinada += chunk_llama.choices[0].delta.content
+            if chunk_mistral.choices[0].delta.content is not None:
+                respuesta_combinada += chunk_mistral.choices[0].delta.content
 
-        response_content = ""
-        for chunk in completion:
-            if chunk.choices[0].delta.content is not None:
-                response_content += chunk.choices[0].delta.content
+        return respuesta_combinada
 
-        return response_content.strip()
     except Exception as e:
         return f"Error: {e}"
 
@@ -125,4 +131,4 @@ def gradio_interface():
     demo.launch()
 
 # Lanzamiento de la interfaz de Gradio
-gradio_interface()
+gradio_interface()