import os import google.generativeai as genai import gradio as gr from openai import OpenAI from gradio_agentchatbot import AgentChatbot, ChatMessage, ThoughtMetadata # Configuración de claves de API OPENAI_API_KEY = os.getenv("API_KEY") GOOGLE_API_KEY = os.getenv("GOOGLE_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'.") # 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) # 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. 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", }, ) # 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. """, } # 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 # 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_openai, user_message], temperature=0.8, top_p=0.7, max_tokens=4000, ) response = completion["choices"][0]["message"]["content"] metadata_assistant = ThoughtMetadata(tool_name="OpenAI") messages.append(ChatMessage(role="assistant", content=response, thought_metadata=metadata_assistant)) 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: error_message = f"Error al procesar la solicitud con Google Generative AI: {e}" messages.append(ChatMessage(role="assistant", content=error_message)) return messages # Descripción de la aplicación description = """ # 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. ### 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."], ["El papá tiene que pagar la mitad de los gastos del colegio."], ["Yo creo que lo mejor para el niño es..."] ] # Interfaz de usuario con Gradio with gr.Blocks() as demo: 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í...", ) 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()