Update app.py

app.py

@@ -40,24 +40,69 @@ model = AutoModelForCausalLM.from_pretrained(
 
 # === Frases prototipo para verificador semántico ===
 frases_fisica = [
-    "Me golpeó con el puño cerrado", "Me pateó", "Me lanzó contra la pared",
-    "Me estranguló", "Me fracturó una costilla", "Me tiró al piso violentamente"
+    "Me golpeó con el puño cerrado",
+    "Me pateó",
+    "Me lanzó contra la pared",
+    "Me estranguló",
+    "Me fracturó una costilla",
+    "Me tiró al piso violentamente"
 ]
+
 frases_sexual = [
-    "Me obligó a tener relaciones sexuales", "Me tocó sin consentimiento",
-    "Me violó", "Me forzó a tener sexo", "Me agredió sexualmente"
+    "Me obligó a tener relaciones sexuales",
+    "Me tocó sin consentimiento",
+    "Me violó",
+    "Me forzó a tener sexo",
+    "Me agredió sexualmente"
+]
+
+frases_economica = [
+    "No me da dinero",
+    "Me quita mi salario",
+    "Me retiene los recursos económicos",
+    "No me deja trabajar",
+    "Me obliga a depender económicamente"
 ]
+
+frases_psicologica = [
+    "Me insulta todo el tiempo",
+    "Me amenaza constantemente",
+    "Me humilla frente a los niños",
+    "Me hace sentir inútil",
+    "Me controla emocionalmente"
+]
+
+frases_negligencia = [
+    "No me brinda atención médica",
+    "No cuida a los hijos",
+    "Me deja sin alimentos",
+    "No se preocupa por mi salud",
+    "Ignora mis necesidades básicas"
+]
+
+# === Embeddings de referencia ===
 embeds_fisica = modelo_vector.encode(frases_fisica)
 embeds_sexual = modelo_vector.encode(frases_sexual)
+embeds_economica = modelo_vector.encode(frases_economica)
+embeds_psicologica = modelo_vector.encode(frases_psicologica)
+embeds_negligencia = modelo_vector.encode(frases_negligencia)
 
-# === Verificador semántico ===
+# === Verificador semántico completo ===
 def verificar_semantico(descripcion):
     emb_desc = modelo_vector.encode(descripcion)
     tipos_detectados = []
-    if max(util.cos_sim(emb_desc, embeds_fisica)[0]) > 0.7:
+
+    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:
         tipos_detectados.append("sexual")
+    if max(util.cos_sim(emb_desc, embeds_economica)[0]) > 0.85:
+        tipos_detectados.append("económica")
+    if max(util.cos_sim(emb_desc, embeds_psicologica)[0]) > 0.85:
+        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 ===
@@ -115,9 +160,24 @@ def predecir_con_recomendacion(edad, genero, hijos, convivencia_agresor, consumo
                 "- Justificación breve de cada medida, indicando los literales aplicables del Artículo 5 de la Ley 575 de 2000 (identificados por letra).\n\n"
                 "RESUELVE:\n"
                 "PRIMERO: Admitir la solicitud.\n"
-                "SEGUNDO, TERCERO, CUARTO, QUINTO, etc.: Ordenar cada medida en un párrafo separado, numerando de forma secuencial.\n"
+                "II, III, IV, V, etc.: Ordenar cada medida en un párrafo separado, numerando en números romanos.\n"
                 "ÚLTIMO: Cúmplase y notifíquese.\n\n"
-                "No incluyas ejemplos en tu respuesta.\n"
+                "Utiliza únicamente las siguientes expresiones para nombrar las medidas si corresponden al caso:\n"
+                "- Prohibición de acercamiento.\n"
+                "- Prohibición de contacto.\n"
+                "- Protección de la residencia.\n"
+                "- Acompañamiento policial.\n"
+                "- Prohibición de consumo de alcohol.\n"
+                "No uses términos como 'nominar' o 'nombrar medidas'.\n\n"
+                "Ejemplo de redacción de medidas (no lo copies literalmente, solo úsalo como guía de estilo):\n"
+                "II: Ordenar la prohibición al agresor de acercarse a la víctima en un radio de 200 metros.\n"
+                "III: Ordenar la prohibición de todo contacto con la víctima por cualquier medio.\n"
+                "IV: Ordenar la protección de la residencia mediante acompañamiento policial.\n"
+                "V: Ordenar la prohibición de consumo de bebidas alcohólicas.\n\n"
+                "Ejemplo de justificación (solo como guía):\n"
+                "a) Prohibición de acercamiento: conforme al literal b) del Artículo 5 de la Ley 575 de 2000.\n"
+                "b) Prohibición de contacto: conforme al literal c) del Artículo 5.\n\n"
+                "No incluyas estos ejemplos en la respuesta final.\n"
                 "No menciones violencia económica ni sexual salvo que se describa expresamente en los hechos.\n"
                 "Redacta cada justificación en una sola frase única, sin repetir fundamentos.\n"
                 "Limita todo el texto a un máximo de 300 palabras.\n"
@@ -125,6 +185,7 @@ def predecir_con_recomendacion(edad, genero, hijos, convivencia_agresor, consumo
             )
         }
     ]
+]
 
     # Aplicar chat template
     encoded = tokenizer.apply_chat_template(messages, return_tensors="pt").to(model.device)