A

app.py

@@ -5,80 +5,59 @@ 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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"
+    api_key="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"  # Reemplaza con tu 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.
-
 ### Directrices de Revisión:
 1. **Comprensión del Contexto y Propósito**:
    - Asegura la comprensión del contexto y propósito del documento.
-
 2. **Identificación del Tipo de Documento**:
    - Identifica correctamente el tipo de documento judicial (sentencia, apelación).
-
 3. **Progresión Lógica y Coherencia**:
    - Asegura una progresión lógica de ideas y coherencia entre secciones.
-
 4. **Claridad y Precisión**:
    - Evalúa la claridad y precisión en cada párrafo y oración.
-
 5. **Consistencia en Terminología Legal**:
    - Verifica el uso consistente de terminología legal.
-
 6. **Corrección Gramatical**:
    - Asegura la corrección gramatical y claridad en la estructura de las oraciones.
-
 7. **Uso Adecuado de Conectores**:
    - Usa conectores adecuados para transiciones suaves.
-
 8. **Precisión en Términos Legales**:
    - Proporciona precisión en términos legales y sugerencias de expresiones más efectivas.
-
 9. **Consideración de Sugerencias y Validaciones**:
    - Clasifica las sugerencias según su relevancia y ajusta el documento en consecuencia.
-
 10. **Revisión Integral**:
     - Realiza una revisión integral y presentación concisa y unificada de la información.
-
 ### Ejemplos de Buenas Prácticas:
 1. **Buena Práctica**:
    - "El tribunal determina que el demandante tiene derecho a una indemnización conforme al artículo 21 del Código Civil."
-
 2. **Mala Práctica**:
    - "El tribunal dice que el demandante debería recibir algún dinero."
-
 ### Prioridades:
 1. Enfócate en la claridad y precisión.
 2. Usa terminología legal específica.
 3. Mantén la consistencia en el tono y terminología.
 """
 
-terminologia_legal = ['folio', 'autos', 'resuelve', 'artículo', 'Código de Procedimiento Civil', 'audiencia', 'notificación', 'prueba', 'nulidad', 'oficiar', 'notificar']
+terminologia_legal = ['folio', 'autos', 'resuelve', 'artículo', 'Código de Procedimiento Civil', ...]
 
 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}"
@@ -86,7 +65,7 @@ def construir_prompt(input_text, contexto=None, tarea=None):
         prompt += f"\nTarea: {tarea}"
     return prompt
 
-def generar_respuesta(prompt):
+def generar_respuesta(prompt, stream=False):
     """
     Genera una respuesta del modelo LLaMA 3.1 dado un prompt específico.
     """
@@ -102,15 +81,18 @@ def generar_respuesta(prompt):
             temperature=0.7,
             top_p=0.9,
             max_tokens=2048,
-            stream=True
+            stream=False
         )
 
-        response_content = ""
-        for chunk in completion:
-            if chunk.choices[0].delta.content is not None:
-                response_content += chunk.choices[0].delta.content
+        if stream:
+            response_content = ""
+            for chunk in completion:
+                if chunk.choices[0].delta.content is not None:
+                    response_content += chunk.choices[0].delta.content
+            return response_content.strip()
+        else:
+            return completion.choices[0].message['content'].strip()
 
-        return response_content.strip()
     except Exception as e:
         return f"Error: {e}"
 
@@ -142,7 +124,7 @@ def mejorar_resolucion(input_text):
     prompt = construir_prompt(input_text, contexto=input_text, tarea="resolucion_judicial")
 
     # Generar la respuesta
-    respuesta = generar_respuesta(prompt)
+    respuesta = generar_respuesta(prompt, stream=False)  # Se usa stream=False para simplificar
 
     # Validar la respuesta
     sugerencias = validar_respuesta(respuesta, contexto=input_text, tarea="resolucion_judicial")
@@ -154,16 +136,28 @@ def mejorar_resolucion(input_text):
 
     return respuesta
 
-# Interfaz Gradio
 def gradio_interface(input_text):
     return mejorar_resolucion(input_text)
 
+# Mejoras en la interfaz de Gradio
 iface = gr.Interface(
     fn=gradio_interface,
-    inputs="text",
-    outputs="text",
+    inputs=gr.inputs.Textbox(lines=10, placeholder="Introduce aquí la resolución judicial o la pregunta...", label="Texto o Pregunta"),
+    outputs=gr.outputs.Textbox(label="Respuesta Mejorada"),
     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."
+    description=(
+        "Este sistema está diseñado para ayudar en la mejora de borradores de resoluciones judiciales o para "
+        "responder preguntas sobre derecho civil con un enfoque técnico y formal. Introduce un texto para recibir "
+        "una respuesta optimizada."
+    ),
+    allow_flagging=False,  # Desactiva el flagging si no es necesario
+    theme="huggingface",  # Usa un tema moderno para mejor apariencia
+    live=False,  # No es necesario actualizar en tiempo real
+    layout="vertical",  # Layout vertical para claridad
+    examples=[
+        ["El tribunal dicta que..."],
+        ["¿Qué dice el artículo 21 del Código Civil sobre...?"]
+    ]
 )
 
-iface.launch(share=True)
+iface.launch()  # Se eliminó share=True para compatibilidad con Hugging Face Spaces