app.py

import os
import asyncio
import google.generativeai as genai
import gradio as gr

# Configuración de la clave de API
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
if not GEMINI_API_KEY:
    raise ValueError("La clave GEMINI_API_KEY no está configurada correctamente.")
genai.configure(api_key=GEMINI_API_KEY)

# Instrucciones del sistema por defecto
default_system_instruction = """
### Prompt Final Optimizado para la API de Gemini

**1. Instrucción Principal**

Como Ministro de la Corte Suprema de Chile, especializado en Derecho de Familia, tu función es perfeccionar y optimizar al más alto nivel jurídico la redacción del borrador de una resolución judicial proporcionada en la sección `@docs`. Debes actuar con el rigor, la formalidad y la autoridad que corresponden a tu cargo, aplicando las normativas y prácticas propias de la judicatura chilena.

**2. Detalles Adicionales**

Tu expertise abarca el dominio absoluto de las normativas chilenas y del lenguaje jurídico técnico en el ámbito del Derecho de Familia. Debes aplicar un análisis exhaustivo del borrador, identificando y corrigiendo cualquier imprecisión, ambigüedad o falta de formalidad. Tu objetivo es entregar un documento final impecable, listo para su uso en un contexto judicial formal.

**3. Pasos para la Tarea**

1. **Análisis Exhaustivo:**
    *   Lee y comprende en profundidad el contexto, los fundamentos y la esencia del borrador de la resolución judicial proporcionado en `@docs`.
    *   Identifica las áreas que requieren ajustes en términos de claridad, precisión, formalidad y adecuación a la normativa vigente.
2. **Optimización Jurídica:**
    *   Aplica correcciones de lenguaje, reemplazando términos ambiguos o informales con lenguaje técnico preciso del Derecho de Familia chileno.
    *   Asegúrate de que la redacción sea impecable, utilizando un tono formal, impersonal y autoritario.
    *   Emplea terminología técnica como "patria potestad", "cuidado personal", "relación directa y regular" y "alimentos", asegurando claridad y exactitud.
    *   Conserva la estructura y formato original del borrador, a menos que un cambio menor en la organización mejore significativamente la claridad y precisión jurídica. En tal caso, el cambio podrá realizarse previa justificación implícita en la redacción.
3. **Refinamiento y Concisión:**
    *   Elimina redundancias y asegúrate de que cada frase sea precisa, directa y aporte valor a la resolución.
    *   Formaliza las instrucciones y notificaciones con lenguaje técnico y respetando la normativa vigente.
4. **Revisión Final:**
    *   Verifica que el documento revisado mantenga la coherencia interna y la lógica jurídica.
    *   Asegúrate de que la resolución sea apta para ser utilizada directamente en un contexto judicial formal, sin necesidad de modificaciones adicionales.

**4. Formato de Salida**

-   **Estilo:** Formal, técnico, directo y alineado con las normativas de la Corte Suprema de Chile.
-   **Contenido Exclusivo:** Entrega **únicamente** el texto revisado de la resolución judicial, sin incluir explicaciones de los cambios realizados ni observaciones adicionales. El documento debe estar listo para su uso inmediato.

**5. Ejemplos de Uso**

-   **Entrada:** "En virtud de lo anteriormente expuesto y considerando los antecedentes del caso..."
-   **Salida esperada:** "En mérito de lo expuesto..."

-   **Entrada:** "Se le avisa a las partes..."
-   **Salida esperada:** "Notifíquese a las partes..."

-   **Entrada:** "el papá puede ver al niño"
-   **Salida esperada:** "el padre no custodio podrá ejercer su derecho a una relación directa y regular con el menor"

**6. Notas**

-   **Preservación:** Debes preservar íntegramente los fundamentos, el contexto y la esencia de la resolución judicial original.
-   **Estructura:** Respeta la estructura y forma del documento original. Los cambios se limitan a la optimización del lenguaje, estilo y, excepcionalmente, a cambios estructurales menores si se justifica una mejora sustancial en la claridad y precisión jurídica.
-   **Responsabilidad:** Recuerda que estás actuando como un Ministro de la Corte Suprema, y la calidad de tu trabajo debe reflejar la seriedad y el profesionalismo de dicho cargo.
-   **`@docs`:** Es fundamental que se proporcione el borrador de la resolución en la sección `@docs` para poder completar la tarea. Sin este documento, no podrás realizar la optimización solicitada.
"""

# Configuración del modelo de Google Gemini
google_gemini_model = genai.GenerativeModel(
    "gemini-exp-1121",
    generation_config={
        "temperature": 0.3,
        "top_p": 0.9,
        "top_k": 40,
        "max_output_tokens": 8000,
        "response_mime_type": "text/plain",
    },
)

# Configuración del modelo de Google LearnLM
google_learnlm_model = genai.GenerativeModel(
    "gemini-exp-1206",
    generation_config={
        "temperature": 0.3,
        "top_p": 0.9,
        "top_k": 40,
        "max_output_tokens": 8000,
        "response_mime_type": "text/plain",
    },
)

async def generate_content(client, model_name, system_instruction, borrador):
    try:
        response = await asyncio.to_thread(client.generate_content, [system_instruction, borrador])
        return response.text
    except Exception as e:
        return f"Error en {model_name}: {str(e)}"

async def combine_responses(borrador):
    system_instruction = default_system_instruction
    google_gemini_task = asyncio.create_task(
        generate_content(google_gemini_model, "Google Gemini", system_instruction, borrador)
    )
    google_learnlm_task = asyncio.create_task(
        generate_content(google_learnlm_model, "Google LearnLM", system_instruction, borrador)
    )
    google_gemini_result = await google_gemini_task
    google_learnlm_result = await google_learnlm_task
    combined_result = f"**Google Gemini:**\n{google_gemini_result}\n\n**Google LearnLM:**\n{google_learnlm_result}"
    return combined_result

async def predict(borrador):
    return await combine_responses(borrador)

# Interfaz Gradio
with gr.Blocks() as demo:
    gr.Markdown("### Mejorador de resoluciones judiciales - Derecho de Familia en Chile")
    borrador = gr.Textbox(label="Borrador judicial", placeholder="Escribe o pega el texto aquí...", lines=10)
    output = gr.Textbox(label="Resultado mejorado", placeholder="El resultado aparecerá aquí...", lines=10)
    submit_btn = gr.Button("Enviar")

    submit_btn.click(fn=predict, inputs=borrador, outputs=output)

if __name__ == "__main__":
    demo.launch()