Update app.py

app.py

@@ -1,12 +1,3 @@
-# -*- coding: utf-8 -*-
-"""app.ipynb
-
-Automatically generated by Colab.
-
-Original file is located at
-    https://colab.research.google.com/drive/1WQ2YqAcXFp-14DglIYkV4rawVDSbyhCe
-"""
-
 # -*- coding: utf-8 -*-
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from sentence_transformers import SentenceTransformer, util
@@ -15,6 +6,7 @@ import joblib
 import torch
 import gradio as gr
 import os
+import re
 
 # === Cargar modelos entrenados ===
 modelo_riesgo = joblib.load("modelo_riesgo.pkl")
@@ -38,6 +30,25 @@ model = AutoModelForCausalLM.from_pretrained(
     device_map="auto"
 )
 
+# === CSS personalizado Externado ===
+css_externado = """
+@import url('https://fonts.googleapis.com/css2?family=Roboto&display=swap');
+body {
+    background-color: #A3A845;
+    font-family: 'Roboto', sans-serif;
+}
+.gr-button {
+    background-color: #004225 !important;
+    color: white !important;
+}
+.gr-button:hover {
+    background-color: #006633 !important;
+}
+.gr-input, .gr-output {
+    background-color: white !important;
+}
+"""
+
 # === Frases prototipo para verificador semántico ===
 frases_fisica = [
     "Me golpeó con el puño cerrado",
@@ -47,7 +58,6 @@ frases_fisica = [
     "Me fracturó una costilla",
     "Me tiró al piso violentamente"
 ]
-
 frases_sexual = [
     "Me obligó a tener relaciones sexuales",
     "Me tocó sin consentimiento",
@@ -55,7 +65,6 @@ frases_sexual = [
     "Me forzó a tener sexo",
     "Me agredió sexualmente"
 ]
-
 frases_economica = [
     "No me da dinero",
     "Me quita mi salario",
@@ -63,7 +72,6 @@ frases_economica = [
     "No me deja trabajar",
     "Me obliga a depender económicamente"
 ]
-
 frases_psicologica = [
     "Me insulta todo el tiempo",
     "Me amenaza constantemente",
@@ -71,7 +79,6 @@ frases_psicologica = [
     "Me hace sentir inútil",
     "Me controla emocionalmente"
 ]
-
 frases_negligencia = [
     "No me brinda atención médica",
     "No cuida a los hijos",
@@ -91,7 +98,6 @@ embeds_negligencia = modelo_vector.encode(frases_negligencia)
 def verificar_semantico(descripcion):
     emb_desc = modelo_vector.encode(descripcion)
     tipos_detectados = []
-
     if max(util.cos_sim(emb_desc, embeds_fisica)[0]) > 0.8:
         tipos_detectados.append("física")
     if max(util.cos_sim(emb_desc, embeds_sexual)[0]) > 0.9:
@@ -102,7 +108,6 @@ def verificar_semantico(descripcion):
         tipos_detectados.append("psicológica")
     if max(util.cos_sim(emb_desc, embeds_negligencia)[0]) > 0.85:
         tipos_detectados.append("negligencia")
-
     return tipos_detectados
 
 # === FUNCIÓN PRINCIPAL ===
@@ -140,19 +145,19 @@ def predecir_con_recomendacion(edad, genero, hijos, convivencia_agresor, consumo
     tipos_str = ", ".join(tipos_combinados)
 
     messages = [
-    {
-        "role": "system",
-        "content": "Eres un jurista colombiano experto en violencia intrafamiliar."
-    },
-    {
-        "role": "user",
-        "content": f"""
+        {
+            "role": "system",
+            "content": "Eres un jurista colombiano experto en violencia intrafamiliar."
+        },
+        {
+            "role": "user",
+            "content": f"""
 Analiza este caso y redacta un Auto de Medida Provisional conforme a la Ley 575 de 2000 y la Ley 1257 de 2008.
 
 Hechos del caso:
 Tipo de violencia: {tipos_str}
 Nivel de riesgo: {riesgo}
-Medida cautelar sugerida: {tipo_medida}
+Medida de protección sugerida: {tipo_medida}
 Descripción detallada: {descripcion}
 
 Redacta el Auto con esta estructura formal:
@@ -164,10 +169,10 @@ CONSIDERACIONES:
 
 RESUELVE:
 PRIMERO: Admitir la solicitud.
-II, III, IV, V, etc.: Ordenar cada medida en un párrafo separado, numerando en números romanos.
+II, III, IV, V, etc.: Ordenar cada medida de protección en un párrafo separado, numerando en números romanos.
 ÚLTIMO: Cúmplase y notifíquese.
 
-Utiliza únicamente las siguientes expresiones para nombrar las medidas si corresponden al caso:
+Utiliza únicamente estas expresiones para nombrar las medidas si corresponden:
 - Prohibición de acercamiento.
 - Prohibición de contacto.
 - Protección de la residencia.
@@ -175,15 +180,13 @@ Utiliza únicamente las siguientes expresiones para nombrar las medidas si corre
 - Prohibición de consumo de alcohol.
 No uses términos como 'nominar' o 'nombrar medidas'.
 
-Ejemplo de redacción de medidas (no lo copies literalmente, solo úsalo como guía de estilo):
-II: Ordenar la prohibición al agresor de acercarse a la víctima en un radio de 200 metros.
-III: Ordenar la prohibición de todo contacto con la víctima por cualquier medio.
-IV: Ordenar la protección de la residencia mediante acompañamiento policial.
-V: Ordenar la prohibición de consumo de bebidas alcohólicas.
+A continuación, algunos literales del Artículo 5 de la Ley 575 de 2000 (solo para tu referencia, no los copies literalmente):
 
-Ejemplo de justificación (solo como guía):
-a) Prohibición de acercamiento: conforme al literal b) del Artículo 5 de la Ley 575 de 2000.
-b) Prohibición de contacto: conforme al literal c) del Artículo 5.
+Literal a): Ordenar al agresor el desalojo inmediato de la casa de habitación que comparte con la víctima.
+Literal b): Ordenar al agresor abstenerse de penetrar en cualquier lugar donde se encuentre la víctima.
+Literal c): Ordenar al agresor abstenerse de comunicarse con la víctima por cualquier medio.
+Literal d): Ordenar la protección de la víctima por la autoridad de policía.
+Literal h): Ordenar al agresor abstenerse de consumir bebidas alcohólicas o sustancias psicoactivas.
 
 No incluyas estos ejemplos en la respuesta final.
 No menciones violencia económica ni sexual salvo que se describa expresamente en los hechos.
@@ -191,8 +194,8 @@ Redacta cada justificación en una sola frase única, sin repetir fundamentos.
 Limita todo el texto a un máximo de 300 palabras.
 Usa un estilo claro, formal y coherente.
 """
-    }
-]
+        }
+    ]
 
     # Aplicar chat template
     encoded = tokenizer.apply_chat_template(messages, return_tensors="pt").to(model.device)
@@ -216,7 +219,7 @@ Usa un estilo claro, formal y coherente.
     else:
         auto_redactado = auto_completo
 
-    # Post-procesar: eliminar líneas duplicadas
+    # Eliminar líneas duplicadas
     lineas = auto_redactado.splitlines()
     lineas_sin_duplicados = []
     vistos = set()
@@ -227,7 +230,7 @@ Usa un estilo claro, formal y coherente.
     auto_redactado = "\n".join(lineas_sin_duplicados)
 
     # Renumerar medidas en números romanos
-    romanos = ["II", "III", "IV", "V", "VI", "VII", "VIII", "IX", "X"]
+    romanos = ["I","II", "III", "IV", "V", "VI", "VII", "VIII", "IX", "X"]
     lineas = auto_redactado.splitlines()
     contador = 0
     for i, linea in enumerate(lineas):
@@ -238,13 +241,17 @@ Usa un estilo claro, formal y coherente.
             contador += 1
     auto_redactado = "\n".join(lineas)
 
-
     return tipos_str, riesgo, tipo_medida, auto_redactado
 
 # === Interfaz Gradio ===
-with gr.Blocks(theme=gr.themes.Soft()) as interfaz:
-    gr.Markdown("<h1 style='text-align:center; color:#004aad;'>LEGALFAMI – Asistente Legal con Razonamiento Jurídico</h1>")
-    gr.Markdown("Predice tipo de violencia, riesgo, medida cautelar y genera un Auto conforme a la Ley 575 Art.5.")
+with gr.Blocks(css=css_externado) as interfaz:
+    gr.Markdown("""
+<div style='display: flex; align-items: center; gap: 12px;'>
+    <img src='https://tuservidor.com/logoexternado.jpg' style='height: 60px;'/>
+    <h1 style='color:#004225; font-family: Roboto, sans-serif;'>LEGALFAMI – Asistente Legal con Razonamiento Jurídico</h1>
+</div>
+""")
+    gr.Markdown("<p style='text-align:center;'>Predice tipo de violencia, nivel de riesgo, medida de protección y redacta un Auto conforme a la Ley 575 Art.5.</p>")
 
     with gr.Row():
         with gr.Column():
@@ -260,7 +267,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as interfaz:
         with gr.Column():
             tipo_violencia_out = gr.Textbox(label="🛑 Tipo de Violencia Detectada", interactive=False)
             riesgo_out = gr.Textbox(label="⚠️ Nivel de Riesgo Estimado", interactive=False)
-            medida_out = gr.Textbox(label="🧾 Tipo de Medida Cautelar Sugerida", interactive=False)
+            medida_out = gr.Textbox(label="🧾 Tipo de Medida de Protección Sugerida", interactive=False)
             recomendacion_out = gr.Textbox(label="📋 Auto de Medida Provisional", lines=12, interactive=False)
 
     boton.click(