A

app.py

@@ -1,10 +1,11 @@
-from openai import OpenAI
+import re
 import gradio as gr
+from openai import OpenAI
 
 # Configuración del cliente OpenAI con la API de NVIDIA
 client = OpenAI(
-  base_url = "https://integrate.api.nvidia.com/v1",
-  api_key = "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"
+    base_url="https://integrate.api.nvidia.com/v1",
+    api_key="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"
 )
 
 # Instrucciones para el modelo
@@ -72,28 +73,34 @@ Eres un Ministro de la Corte Suprema de Chile, especializado en derecho civil. T
 3. Mantén la consistencia en el tono y terminología.
 """
 
-def mejorar_resolucion(input_text):
-    # Construcción del prompt con instrucciones unificadas
-    prompt = f"Texto o pregunta:\n{input_text}\nRespuesta:"
-
-    # Construcción de los mensajes según especificaciones
+terminologia_legal = ['folio', 'autos', 'resuelve', 'artículo', 'Código de Procedimiento Civil', 'audiencia', 'notificación', 'prueba', 'nulidad', 'oficiar', 'notificar']
+
+def construir_prompt(input_text, contexto=None, tarea=None):
+    """
+    Construye el prompt para el modelo LLaMA 3.1.
+    """
+    prompt = f"Texto o pregunta: {input_text}"
+    if contexto:
+        prompt += f"\nContexto: {contexto}"
+    if tarea:
+        prompt += f"\nTarea: {tarea}"
+    return prompt
+
+def generar_respuesta(prompt):
+    """
+    Genera una respuesta del modelo LLaMA 3.1 dado un prompt específico.
+    """
     messages = [
-        {
-            "role": "system",
-            "content": instrucciones
-        },
-        {
-            "role": "user",
-            "content": prompt
-        }
+        {"role": "system", "content": instrucciones},
+        {"role": "user", "content": prompt}
     ]
 
     try:
         completion = client.chat.completions.create(
             model="meta/llama-3.1-405b-instruct",
             messages=messages,
-            temperature=1.0,
-            top_p=0.7,
+            temperature=0.7,
+            top_p=0.9,
             max_tokens=1024,
             stream=True
         )
@@ -107,20 +114,56 @@ def mejorar_resolucion(input_text):
     except Exception as e:
         return f"Error: {e}"
 
-# Configuración avanzada de Gradio
-def gradio_interface():
-    with gr.Blocks() as demo:
-        gr.Markdown("# Mejora de Resoluciones Judiciales y Respuestas sobre Redacción Judicial")
-        with gr.Row():
-            with gr.Column():
-                input_text = gr.Textbox(label="Introduce tu resolución judicial o pregunta", lines=5, placeholder="Escribe aquí tu resolución judicial o pregunta...")
-                submit_button = gr.Button("Enviar")
-            with gr.Column():
-                output_text = gr.Textbox(label="Respuesta mejorada o respuesta corta", lines=10, placeholder="Aquí aparecerá la respuesta...")
+def validar_respuesta(respuesta, contexto=None, tarea=None):
+    """
+    Valida la coherencia y la calidad de la respuesta generada.
+    """
+    sugerencias = []
+
+    # Verificar la consistencia terminológica
+    if not any(term in respuesta for term in terminologia_legal):
+        sugerencias.append("Incluir terminología legal específica.")
 
-        submit_button.click(fn=mejorar_resolucion, inputs=input_text, outputs=output_text)
+    # Verificar la corrección gramatical básica
+    if not re.match(r'^[A-Z][^.?!]*[.?!]$', respuesta):
+        sugerencias.append("Mejorar la corrección gramatical y estructura de las oraciones.")
 
-    demo.launch()
+    # Asegurar que se mantenga la formalidad y precisión
+    if len(respuesta) < 20:
+        sugerencias.append("Aumentar la longitud para proporcionar una respuesta más útil.")
+
+    return sugerencias
+
+def mejorar_resolucion(input_text):
+    """
+    Función principal que mejora la resolución judicial o responde preguntas.
+    """
+    # Construir el prompt
+    prompt = construir_prompt(input_text, contexto=input_text, tarea="resolucion_judicial")
+
+    # Generar la respuesta
+    respuesta = generar_respuesta(prompt)
+
+    # Validar la respuesta
+    sugerencias = validar_respuesta(respuesta, contexto=input_text, tarea="resolucion_judicial")
+
+    # Si la respuesta no es válida, generar una nueva respuesta
+    if sugerencias:
+        prompt = construir_prompt(input_text, contexto=input_text, tarea="resolucion_judicial")
+        respuesta = generar_respuesta(prompt)
+
+    return respuesta
+
+# Interfaz Gradio
+def gradio_interface(input_text):
+    return mejorar_resolucion(input_text)
+
+iface = gr.Interface(
+    fn=gradio_interface,
+    inputs="text",
+    outputs="text",
+    title="Mejora de Resoluciones Judiciales",
+    description="Introduce una resolución judicial o una pregunta relacionada con derecho civil. El sistema mejorará el borrador o proporcionará una respuesta técnica y formal."
+)
 
-# Lanzamiento de la interfaz de Gradio
-gradio_interface()
+iface.launch(share=True)