Update app.py

app.py

@@ -54,7 +54,7 @@ embeds_sexual = modelo_vector.encode(frases_sexual)
 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.7:
         tipos_detectados.append("sexual")
@@ -113,12 +113,12 @@ def predecir_con_recomendacion(edad, genero, hijos, convivencia_agresor, consumo
                 "CONSIDERACIONES:\n"
                 "- Hechos relevantes y valoración del riesgo.\n"
                 "- Fundamentos jurídicos aplicables.\n"
-                "- Justificación breve de cada literal del Artículo 5 aplicable.\n\n"
+                "- Justificación breve de cada literal aplicable del Artículo 5 de la Ley 575 de 2000, identificando por letra (a), (b), etc.\n\n"
                 "RESUELVE:\n"
                 "PRIMERO: Admitir la solicitud.\n"
-                "SEGUNDO: Ordenar las siguientes medidas (una por línea).\n"
-                "TERCERO: Cúmplase y notifíquese.\n\n"
-                "Limita tu respuesta a máximo 600 palabras, en estilo formal."
+                "SEGUNDO, TERCERO, CUARTO, etc.: Ordenar cada medida de manera separada, en un párrafo por medida.\n"
+                "ÚLTIMO: Cúmplase y notifíquese.\n\n"
+                "Redacta en un estilo formal y claro, máximo 400 palabras."
             )
         }
     ]
@@ -136,7 +136,14 @@ def predecir_con_recomendacion(edad, genero, hijos, convivencia_agresor, consumo
 
     # Decodificar
     decoded = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
-    auto_redactado = decoded[0].strip()
+    auto_completo = decoded[0].strip()
+
+    # Recortar desde "CONSIDERACIONES"
+    inicio = auto_completo.find("CONSIDERACIONES")
+    if inicio != -1:
+        auto_redactado = auto_completo[inicio:]
+    else:
+        auto_redactado = auto_completo
 
     return tipos_str, riesgo, tipo_medida, auto_redactado