sds

app.py

@@ -1,7 +1,8 @@
 import os
 import asyncio
-import gradio as gr
+import google.generativeai as genai
 from openai import OpenAI
+import gradio as gr
 
 # Configuración de claves de API
 NVIDIA_API_KEY = os.getenv("NVIDIA_API_KEY")
@@ -10,21 +11,57 @@ GEMINI_API_KEY = os.getenv("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
+# Configuración de clientes
+# Cliente Google Generative AI
+genai.configure(api_key=GEMINI_API_KEY)
+google_model = genai.GenerativeModel(
+    model_name="gemini-exp-1114",
+    system_instruction= """
+    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.
+
+Principios Generales para la Redacción Jurídica:
+* Tono Formal y Técnico: Mantén un tono formal, autoritario e impersonal. Emplea el modo imperativo en expresiones de orden y mandato judicial, como 'Notifíquese', 'Archívese' o 'Cítese'. Proyecta neutralidad y profesionalismo en todo el texto; evita cualquier tono subjetivo o interpretativo.
+* Precisión y Claridad: Usa terminología jurídica específica del Derecho de Familia, como 'patria potestad', 'derecho de visitas' o 'pensión alimenticia', evitando sinónimos menos técnicos. Asegura que cada enunciado sea claro y preciso, sin ambigüedades que puedan dar lugar a interpretaciones erróneas.
+* Concisión y Economía de Palabras: Emplea una redacción directa, donde cada palabra tenga una función clara. Evita redundancias y expresiones superfluas para facilitar una comprensión inmediata y sin distracciones.
+* Uso de Términos y Expresiones Propios del Derecho Chileno: Utiliza expresiones tradicionales y formales del ámbito judicial chileno, como 'Notifíquese por el estado diario'. En conclusiones y firmas, emplea frases estándar del lenguaje jurídico chileno, como 'Proveyó y firmó el Juez mediante firma electrónica avanzada'.
+
+# Pasos
+
+1. **Lectura y Análisis del Borrador:** Revisa el borrador completo de la resolución judicial para entender el contexto, los argumentos presentados y la decisión tomada. Identifica áreas que requieran mejoras en términos de claridad, precisión, concisión y formalidad.
+2. **Aplicación del Tono Formal y Técnico:** Asegúrate de que el lenguaje utilizado sea apropiado para una resolución judicial de la Corte Suprema. Utiliza el modo imperativo en los mandatos judiciales. Elimina cualquier expresión coloquial o subjetiva, reemplazándola con un lenguaje técnico y neutral.
+3. **Mejora de la Precisión y Claridad:** Verifica que la terminología jurídica utilizada sea específica y correcta, especialmente aquella relacionada con el Derecho de Familia. Revisa cada enunciado para eliminar ambigüedades y asegurar que la intención de la corte sea clara e inequívoca.
+4. **Asegurar la Concisión y Economía de Palabras:** Revisa cada frase y párrafo para eliminar redundancias y expresiones innecesarias. Simplifica la estructura de las oraciones sin perder la precisión, asegurando que cada palabra contribuya al significado del texto.
+5. **Uso Adecuado de Términos y Expresiones Judiciales Chilenas:** Incorpora o revisa las expresiones formales tradicionales del ámbito judicial chileno, asegurando que se utilicen correctamente y en el contexto adecuado. Verifica que las conclusiones y firmas sigan el formato estándar del lenguaje jurídico chileno.
+6. **Revisión Final y Corrección:** Realiza una revisión final del documento para asegurar que todas las modificaciones se hayan implementado correctamente y que la resolución mantenga su coherencia y lógica. Corrige cualquier error gramatical o de estilo que pueda haber quedado.
+
+# Formato de Salida
+
+La salida será el borrador de la resolución judicial revisado y mejorado, presentado en formato de texto. El texto debe mantener el formato y la estructura original del borrador, pero con las mejoras estilísticas y de lenguaje implementadas según las directrices proporcionadas.
+
+# Notas
+
+Es crucial que el asistente de IA se adhiera estrictamente a las instrucciones, sin alterar la sustancia de la decisión judicial. El objetivo es perfeccionar la redacción, no modificar el contenido. Se debe mantener la formalidad y el rigor característico de los documentos de la Corte Suprema de Chile.
+""".strip(),
+    generation_config={
+        "temperature": 0.5,
+        "top_p": 0.7,
+        "top_k": 40,
+        "max_output_tokens": 4000,
+        "response_mime_type": "text/plain",
+    },
+)
+
+ Cliente NVIDIA
 nvidia_client = OpenAI(
     base_url="https://integrate.api.nvidia.com/v1",
     api_key=NVIDIA_API_KEY
 )
-
-gemini_client = OpenAI(
-    base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
-    api_key=GEMINI_API_KEY
-)
-
-# Mensaje del sistema
-system_message = {
+nvidia_system_message = {
     "role": "system",
-    "content": """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.
+    "content": """
+    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.
 
@@ -49,15 +86,24 @@ La salida será el borrador de la resolución judicial revisado y mejorado, pres
 
 # Notas
 
-Es crucial que el asistente de IA se adhiera estrictamente a las instrucciones, sin alterar la sustancia de la decisión judicial. El objetivo es perfeccionar la redacción, no modificar el contenido. Se debe mantener la formalidad y el rigor característico de los documentos de la Corte Suprema de Chile."""
+Es crucial que el asistente de IA se adhiera estrictamente a las instrucciones, sin alterar la sustancia de la decisión judicial. El objetivo es perfeccionar la redacción, no modificar el contenido. Se debe mantener la formalidad y el rigor característico de los documentos de la Corte Suprema de Chile.
+""".strip()
 }
 
-# Función para interactuar con la API
-async def call_api(client, model_name, message, history):
+# Función para llamar al modelo de Google
+async def google_response(borrador):
     try:
-        messages = [system_message] + history + [{"role": "user", "content": message}]
-        completion = client.chat.completions.create(
-            model=model_name,
+        response = google_model.generate_content(prompt=borrador)
+        return response.text
+    except Exception as e:
+        return f"Error en Google Generative AI: {e}"
+
+# Función para llamar al modelo de NVIDIA
+async def nvidia_response(borrador):
+    try:
+        messages = [nvidia_system_message, {"role": "user", "content": borrador}]
+        completion = nvidia_client.chat.completions.create(
+            model="meta/llama-3.1-405b-instruct",  # Cambiar si usas otro modelo NVIDIA
             messages=messages,
             temperature=0.5,
             top_p=0.7,
@@ -65,65 +111,53 @@ async def call_api(client, model_name, message, history):
         )
         return completion.choices[0].message.content
     except Exception as e:
-        return f"Error API: {e}"
-
-async def nvidia_response(message, history):
-    return await call_api(nvidia_client, "meta/llama-3.1-405b-instruct", message, history)
+        return f"Error en NVIDIA API: {e}"
 
-async def gemini_response(message, history):
-    return await call_api(gemini_client, "gemini", message, history)
+# Función para procesar ambas respuestas
+async def procesar_respuestas(borrador, chat_historial, cargando):
+    cargando = True
+    yield chat_historial + [{"role": "user", "content": borrador}], cargando
 
-# Función principal para procesar ambas respuestas
-async def respond_both_models(message, history, loading_state):
-    loading_state = True
-    yield history + [{"role": "user", "content": message}], loading_state
-
-    nvidia_result, gemini_result = await asyncio.gather(
-        nvidia_response(message, history),
-        gemini_response(message, history)
+    google_result, nvidia_result = await asyncio.gather(
+        google_response(borrador),
+        nvidia_response(borrador)
     )
 
-    loading_state = False
-
-    if "Error API:" in nvidia_result:
-        history.append({"role": "assistant", "content": f"**Error NVIDIA:**\n{nvidia_result}"})
-    else:
-        history.append({"role": "assistant", "content": f"**NVIDIA Response:**\n{nvidia_result}"})
+    cargando = False
 
-    if "Error API:" in gemini_result:
-        history.append({"role": "assistant", "content": f"**Error Gemini:**\n{gemini_result}"})
-    else:
-        history.append({"role": "assistant", "content": f"**Gemini Response:**\n{gemini_result}"})
+    # Agregar respuestas al historial
+    chat_historial.append({"role": "assistant", "content": f"**Google Response:**\n{google_result}"})
+    chat_historial.append({"role": "assistant", "content": f"**NVIDIA Response:**\n{nvidia_result}"})
 
-    yield history, loading_state
+    yield chat_historial, cargando
 
-# Interfaz de usuario
+# Configuración de la interfaz de usuario
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# Ministro IA de la Corte Suprema - Derecho de Familia")
+    gr.Markdown("# Comparador de Modelos Jurídicos: Google Gemini vs NVIDIA")
     with gr.Row():
-        gr.Markdown("### Comparación de Modelos: NVIDIA vs Gemini")
+        gr.Markdown("### Prueba dos modelos IA para redactar resoluciones judiciales")
 
-    chatbot = gr.Chatbot(label="Respuestas de los Modelos", type='messages')
-    message_input = gr.Textbox(
-        label="Borrador de la Resolución",
-        placeholder="Ingrese el borrador aquí...",
-        lines=6,
+    chatbox = gr.Chatbot(label="Respuestas de los Modelos", type="messages")
+    input_text = gr.Textbox(
+        label="Borrador de Resolución",
+        placeholder="Introduce el texto que deseas perfeccionar...",
+        lines=6
     )
     submit_button = gr.Button("Enviar")
     clear_button = gr.Button("Limpiar")
 
-    loading_state = gr.State(False)
-    history = gr.State([])
+    cargando = gr.State(False)
+    chat_historial = gr.State([])
 
     submit_button.click(
-        respond_both_models,
-        inputs=[message_input, history, loading_state],
-        outputs=[chatbot, loading_state]
+        procesar_respuestas,
+        inputs=[input_text, chat_historial, cargando],
+        outputs=[chatbox, cargando]
     )
     clear_button.click(
         lambda: ([], False),
         inputs=None,
-        outputs=[chatbot, loading_state]
+        outputs=[chatbox, cargando]
     )
 
     gr.Examples(
@@ -131,10 +165,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             ["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
+        inputs=input_text
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()