s

app.py

@@ -1,117 +1,94 @@
 import os
-import google.generativeai as genai
-import gradio as gr
 from openai import OpenAI
+import gradio as gr
 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
+NVIDIA_API_KEY = os.getenv("NVIDIA_API_KEY")
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 
-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.
+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'.")
 
-# Notas
+# Configuración de clientes
+nvidia_client = OpenAI(
+    base_url="https://integrate.api.nvidia.com/v1",
+    api_key=NVIDIA_API_KEY,
+)
 
-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",
-    },
+gemini_client = OpenAI(
+    api_key=GEMINI_API_KEY,
+    base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
 )
 
-# Mensaje del sistema para OpenAI
-system_message_openai = {
+# Mensaje de sistema
+system_message = {
     "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.
-""",
+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.
+[... más detalles como en el prompt proporcionado ...]
+""".strip(),
 }
 
-# Función para interactuar con OpenAI
-def llama_response(prompt: str, messages: list[ChatMessage]):
+# Función para interactuar con NVIDIA API
+def nvidia_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")
+    metadata_user = ThoughtMetadata(tool_name="NVIDIA")
     messages.append(ChatMessage(role="user", content=prompt, thought_metadata=metadata_user))
 
-    # Generar respuesta del modelo OpenAI
+    # Generar respuesta con NVIDIA API
     try:
-        user_message = {"role": "user", "content": prompt}
-        completion = client.chat.completions.create(
+        completion = nvidia_client.chat.completions.create(
             model="meta/llama-3.1-405b-instruct",
-            messages=[system_message_openai, user_message],
-            temperature=0.8,
+            messages=[system_message, {"role": "user", "content": prompt}],
+            temperature=0.2,
             top_p=0.7,
-            max_tokens=4000,
+            max_tokens=1024,
+            stream=True,
         )
+        response_content = ""
+        for chunk in completion:
+            if chunk.choices[0].delta.content is not None:
+                response_content += chunk.choices[0].delta.content
 
-        response = completion["choices"][0]["message"]["content"]
-        metadata_assistant = ThoughtMetadata(tool_name="OpenAI")
-        messages.append(ChatMessage(role="assistant", content=response, thought_metadata=metadata_assistant))
-
+        metadata_assistant = ThoughtMetadata(tool_name="NVIDIA")
+        messages.append(ChatMessage(role="assistant", content=response_content, thought_metadata=metadata_assistant))
     except Exception as e:
-        error_message = f"Error al procesar la solicitud con OpenAI: {e}"
+        error_message = f"Error al procesar la solicitud con NVIDIA API: {e}"
         messages.append(ChatMessage(role="assistant", content=error_message))
     
     return messages
 
-# Función para interactuar con Google Generative AI
+# Función para interactuar con Gemini API
 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")
+    metadata_user = ThoughtMetadata(tool_name="Gemini")
     messages.append(ChatMessage(role="user", content=prompt, thought_metadata=metadata_user))
 
-    # Generar respuesta del modelo Google
+    # Generar respuesta con Gemini API
     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))
+        response = gemini_client.chat.completions.create(
+            model="gemini-1.5-flash",
+            messages=[system_message, {"role": "user", "content": prompt}],
+            stream=True,
+        )
+        response_content = ""
+        for chunk in response:
+            if chunk.choices[0].delta.content is not None:
+                response_content += chunk.choices[0].delta.content
 
+        metadata_assistant = ThoughtMetadata(tool_name="Gemini")
+        messages.append(ChatMessage(role="assistant", content=response_content, thought_metadata=metadata_assistant))
     except Exception as e:
-        error_message = f"Error al procesar la solicitud con Google Generative AI: {e}"
+        error_message = f"Error al procesar la solicitud con Gemini API: {e}"
         messages.append(ChatMessage(role="assistant", content=error_message))
     
     return messages
@@ -119,51 +96,37 @@ def gemini_response(prompt: str, messages: list[ChatMessage]):
 # 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.
+Este espacio utiliza NVIDIA y Gemini APIs para mejorar borradores de resoluciones judiciales en Derecho de Familia chileno.
 """
 
 # 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í...",
     )
+
+    # Botones y selector
     modelo_selector = gr.Radio(
-        choices=["OpenAI", "Google Generative AI"],
+        choices=["NVIDIA", "Gemini"],
         label="Selecciona el modelo",
-        value="OpenAI"
+        value="NVIDIA"
     )
-    estado = gr.Label(label="Estado")
-
-    # Botones de interacción
-    with gr.Row():
-        submit_button = gr.Button("Submit")
-        clear_button = gr.Button("Limpiar")
+    submit_button = gr.Button("Submit")
+    clear_button = gr.Button("Limpiar")
 
-    # Lógica de los botones
+    # Enlazar acciones a botones
     submit_button.click(
-        lambda prompt, modelo, messages: llama_response(prompt, messages) if modelo == "OpenAI" else gemini_response(prompt, messages),
+        lambda prompt, modelo, messages: nvidia_response(prompt, messages) if modelo == "NVIDIA" else gemini_response(prompt, messages),
         inputs=[text_input, modelo_selector, chatbot],
         outputs=[chatbot],
     )
@@ -173,9 +136,8 @@ with gr.Blocks() as demo:
     gr.Examples(
         examples=examples,
         inputs=[text_input],
-        label="Ejemplos de Borradores",
     )
 
 # Ejecutar la aplicación
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()