import os import asyncio import gradio as gr from openai import OpenAI # Configuración de claves de API (¡IMPORTANTE! - Reemplaza con tus claves) NVIDIA_API_KEY = os.getenv("NVIDIA_API_KEY") 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 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 (sin cambios) system_message = { """ # ... 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. """ } # Función para interactuar con la API (generalizada) async def call_api(client, model_name, message, history): try: user_message = {"role": "user", "content": message} messages = [system_message] + history + [user_message] completion = client.chat.completions.create( model=model_name, messages=messages, temperature=0.5, top_p=0.7, max_tokens=4000 ) return completion.choices[0].message.content # Accediendo a la respuesta con el nuevo formato OpenAI 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) async def gemini_response(message, history): return await call_api(gemini_client, "gemini-exp-1114", message, history) # Función principal (con manejo de errores y spinner) async def respond_both_models(message, history, loading_state): loading_state = True yield [history + [("Procesando...", None)], loading_state] nvidia_result, gemini_result = await asyncio.gather( nvidia_response(message, history), gemini_response(message, history) ) loading_state = False responses = [] if "Error API:" in nvidia_result: responses.append((message, f"**Error NVIDIA:**\n{nvidia_result}")) else: responses.append((message, f"**NVIDIA Response:**\n{nvidia_result}")) if "Error API:" in gemini_result: responses.append((message, f"**Error Gemini:**\n{gemini_result}")) else: responses.append((message, f"**Gemini Response:**\n{gemini_result}")) yield [history + responses, loading_state] # Interfaz de usuario (con ejemplos y tema) with gr.Blocks(theme=gr.themes.Soft()) as demo: gr.Markdown("# Ministro IA de la Corte Suprema - Derecho de Familia") with gr.Row(): gr.Markdown("### Comparación de Modelos: NVIDIA vs Gemini") chatbot = gr.Chatbot(label="Respuestas de los Modelos") message_input = gr.Textbox( label="Borrador de la Resolución", placeholder="Ingrese el borrador aquí...", lines=6, ) submit_button = gr.Button("Enviar") clear_button = gr.Button("Limpiar") loading_state = gr.State(False) submit_button.click( respond_both_models, inputs=[message_input, chatbot, loading_state], outputs=[chatbot, loading_state] ) clear_button.click(lambda: ["", False], inputs=None, outputs=[message_input, chatbot, loading_state]) gr.Examples( examples=[ ["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 ) if __name__ == "__main__": demo.launch()