A

app.py

@@ -1,93 +1,133 @@
-import gradio as gr
 import os
+import google.generativeai as genai
+import gradio as gr
 from openai import OpenAI
-from gradio_agentchatbot import AgentChatbot, ChatMessage
+from gradio_agentchatbot import AgentChatbot, ChatMessage, ThoughtMetadata
 
-# Configura la clave de API
-api_key = os.getenv("API_KEY")
-if not api_key:
-    raise ValueError("La variable de entorno 'API_KEY' no está configurada.")
+# Configuración de claves de API
+OPENAI_API_KEY = os.getenv("API_KEY")
+GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
 
-# Inicializa el cliente OpenAI
-client = OpenAI(base_url="https://integrate.api.nvidia.com/v1", api_key=api_key)
+if not OPENAI_API_KEY or not GOOGLE_API_KEY:
+    raise ValueError("Las claves de API no están configuradas. Asegúrate de establecer 'API_KEY' y 'GOOGLE_API_KEY'.")
 
-# Define el mensaje del sistema (mejorado)
-system_message = {
-    "role": "system",
-    "content": """
-Actúa como un Ministro de la Corte Suprema de Chile, especializado en Derecho de Familia. Tu tarea es perfeccionar la redacción de un borrador de resolución judicial. Debes lograr claridad, precisión, concisión y formalidad, **sin alterar la sustancia de la decisión.**
+# Configuración de OpenAI y Google Generative AI
+client = OpenAI(base_url="https://integrate.api.nvidia.com/v1", api_key=OPENAI_API_KEY)
+genai.configure(api_key=GOOGLE_API_KEY)
 
-## Instrucciones:
+# Inicialización de modelos
+google_model = genai.GenerativeModel(
+    model_name="gemini-1.5-flash",
+    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. 
 
-1. **Reemplaza términos ambiguos o coloquiales con terminología jurídica precisa.** (ej., "régimen de visitas" en lugar de "ver al niño", "pensión alimenticia" en lugar de "dar plata").
-2. **Elimina redundancias y frases innecesarias.** Simplifica las oraciones manteniendo el significado original.
-3. **Utiliza un tono formal e impersonal.** Evita expresiones subjetivas o coloquiales. Expresa las decisiones de forma objetiva y concisa.
-4. **Usa expresiones judiciales chilenas estándar, especialmente en conclusiones y firmas.** (ej., "Proveyó y firmó el Juez mediante firma electrónica avanzada").
-5. **Mantén la estructura original del borrador.** No reordenes párrafos ni secciones.
+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. 
 
-## Ejemplos:
+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'.
 
-- Borrador: "El juez decide que el papá puede ver a su hijo los fines de semana."
-  Reformular a: "Se establece un régimen de visitas del menor con su padre los fines de semana."
-- Borrador: "Se ordena pagarle una pensión a la mamá para los gastos del niño."
-  Reformular a: "Se fija una pensión alimenticia a favor de la demandante para la manutención del menor."
-- Borrador: "El papá tiene que pagar la mitad de los gastos del colegio."
-  Reformular a: "Se establece que ambos padres contribuirán en partes iguales al pago de los gastos educacionales del menor."
-- Borrador: "Yo creo que lo mejor para el niño es..."
-  Reformular a: "Se considera que la mejor opción para el bienestar del menor es..."
+# Pasos
 
-## Restricciones:
+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.
 
-- **No modifiques las decisiones, los montos, las fechas, ni las partes involucradas en la resolución.** Tu tarea se limita a mejorar la redacción del texto existente.
-- No interpretes ni amplíes el contenido existente.
-- No incluyas análisis o comentarios personales.
+# Formato de Salida
 
-## 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.
 
-Devuelve el borrador de la resolución judicial revisado y mejorado en formato de texto, manteniendo la estructura original.
-"""
+# 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",
+    },
+)
+
+# Mensaje del sistema para OpenAI
+system_message_openai = {
+    "role": "system",
+    "content": """
+Eres un asistente de IA que actúa como Ministro de la Corte Suprema de Chile especializado en Derecho de Familia. Tu tarea es revisar y mejorar borradores de resoluciones judiciales, asegurando claridad, precisión y formalidad.
+""",
 }
 
-def llama_response(prompt, messages):
-    messages.append(ChatMessage(role="user", content=prompt))
-    yield messages, "Procesando..."
+# Función para interactuar con OpenAI
+def llama_response(prompt: str, messages: list[ChatMessage]):
+    if not prompt.strip():
+        messages.append(ChatMessage(role="assistant", content="Por favor, ingresa un texto válido."))
+        return messages
 
-    user_message = {"role": "user", "content": prompt}
+    # Agregar mensaje del usuario al historial
+    metadata_user = ThoughtMetadata(tool_name="OpenAI")
+    messages.append(ChatMessage(role="user", content=prompt, thought_metadata=metadata_user))
 
+    # Generar respuesta del modelo OpenAI
     try:
+        user_message = {"role": "user", "content": prompt}
         completion = client.chat.completions.create(
             model="meta/llama-3.1-405b-instruct",
-            messages=[system_message, user_message],
+            messages=[system_message_openai, user_message],
             temperature=0.8,
             top_p=0.7,
             max_tokens=4000,
-            stream=True,
         )
 
-        response_text = ""
-        for chunk in completion:
-            if chunk.choices[0].delta.content is not None:
-                response_text += chunk.choices[0].delta.content
-                messages.append(ChatMessage(role="assistant", content=response_text))
-                yield messages, "Procesando..."
+        response = completion["choices"][0]["message"]["content"]
+        metadata_assistant = ThoughtMetadata(tool_name="OpenAI")
+        messages.append(ChatMessage(role="assistant", content=response, thought_metadata=metadata_assistant))
 
-        yield messages, "Listo"
+    except Exception as e:
+        error_message = f"Error al procesar la solicitud con OpenAI: {e}"
+        messages.append(ChatMessage(role="assistant", content=error_message))
+    
+    return messages
+
+# Función para interactuar con Google Generative AI
+def gemini_response(prompt: str, messages: list[ChatMessage]):
+    if not prompt.strip():
+        messages.append(ChatMessage(role="assistant", content="Por favor, ingresa un texto válido."))
+        return messages
+
+    # Agregar mensaje del usuario al historial
+    metadata_user = ThoughtMetadata(tool_name="Google Generative AI")
+    messages.append(ChatMessage(role="user", content=prompt, thought_metadata=metadata_user))
+
+    # Generar respuesta del modelo Google
+    try:
+        response = google_model.generate_content(prompt)
+        metadata_assistant = ThoughtMetadata(tool_name="Google Generative AI")
+        messages.append(ChatMessage(role="assistant", content=response.text, thought_metadata=metadata_assistant))
 
     except Exception as e:
-        print(f"Error: {e}")
-        yield messages + [ChatMessage(role="assistant", content="Error al procesar la solicitud.")], "Error"
+        error_message = f"Error al procesar la solicitud con Google Generative AI: {e}"
+        messages.append(ChatMessage(role="assistant", content=error_message))
+    
+    return messages
 
-# Interfaz de Gradio (mejorada)
+# Descripción de la aplicación
 description = """
-## Ministro IA de la Corte Suprema - Derecho de Familia
+# Ministro IA de la Corte Suprema - Derecho de Familia
+Este espacio utiliza OpenAI y Google Generative AI para mejorar borradores de resoluciones judiciales en Derecho de Familia chileno.
 
-Este espacio perfecciona la redacción de resoluciones judiciales chilenas en Derecho de Familia.
-
-**Instrucciones:**
-1. Pega el borrador en el cuadro de texto.
-2. Haz clic en "Submit".
+### Instrucciones:
+1. Ingresa el borrador de la resolución judicial en el cuadro de texto.
+2. Selecciona el modelo que deseas usar (OpenAI o Google).
+3. Haz clic en "Submit" para recibir la versión mejorada.
 """
 
+# Ejemplos
 examples = [
     ["El juez decide que el papá puede ver a su hijo los fines de semana."],
     ["Se ordena pagarle una pensión a la mamá para los gastos del niño."],
@@ -95,16 +135,47 @@ examples = [
     ["Yo creo que lo mejor para el niño es..."]
 ]
 
+# Interfaz de usuario con Gradio
 with gr.Blocks() as demo:
-    chatbot = AgentChatbot(label="Resolución")
+    gr.Markdown(description)
+
+    with gr.Row():
+        gr.Image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers-logo.png", width=100)  # Logo opcional
+        gr.Markdown("### IA para Mejorar Resoluciones Judiciales")
+
+    chatbot = AgentChatbot(label="Resolución Mejorada", avatar_images=("user.png", "bot.png"))
     text_input = gr.Textbox(
         lines=10,
         label="Borrador de la Resolución",
         placeholder="Ingrese el borrador aquí...",
     )
-    text_input.submit(
-        llama_response, [text_input, chatbot], [chatbot, gr.Label(label="Estado")]
+    modelo_selector = gr.Radio(
+        choices=["OpenAI", "Google Generative AI"],
+        label="Selecciona el modelo",
+        value="OpenAI"
+    )
+    estado = gr.Label(label="Estado")
+
+    # Botones de interacción
+    with gr.Row():
+        submit_button = gr.Button("Submit")
+        clear_button = gr.Button("Limpiar")
+
+    # Lógica de los botones
+    submit_button.click(
+        lambda prompt, modelo, messages: llama_response(prompt, messages) if modelo == "OpenAI" else gemini_response(prompt, messages),
+        inputs=[text_input, modelo_selector, chatbot],
+        outputs=[chatbot],
+    )
+    clear_button.click(lambda: ("", []), inputs=None, outputs=[text_input, chatbot])
+
+    # Ejemplos
+    gr.Examples(
+        examples=examples,
+        inputs=[text_input],
+        label="Ejemplos de Borradores",
     )
 
+# Ejecutar la aplicación
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()