Update app.py

app.py

@@ -140,52 +140,59 @@ 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"Analiza este caso y redacta un Auto de Medida Provisional conforme a la Ley 575 de 2000 y la Ley 1257 de 2008.\n\n"
-                f"Hechos del caso:\n"
-                f"Tipo de violencia: {tipos_str}\n"
-                f"Nivel de riesgo: {riesgo}\n"
-                f"Medida cautelar sugerida: {tipo_medida}\n"
-                f"Descripción detallada: {descripcion}\n\n"
-                "Redacta el Auto con esta estructura formal:\n\n"
-                "CONSIDERACIONES:\n"
-                "- Relación breve de los hechos y valoración del riesgo.\n"
-                "- Fundamentos jurídicos aplicables. Cita únicamente el Artículo 5 de la Ley 575 de 2000 y los Artículos 16 y 17 de la Ley 1257 de 2008 si corresponde.\n"
-                "- 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"
-                "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"
-                "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"
-                "Usa un estilo claro, formal y coherente."
-                )
-            }
-        ]
-    ]
+    {
+        "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}
+Descripción detallada: {descripcion}
+
+Redacta el Auto con esta estructura formal:
+
+CONSIDERACIONES:
+- Relación breve de los hechos y valoración del riesgo.
+- Fundamentos jurídicos aplicables. Cita únicamente el Artículo 5 de la Ley 575 de 2000 y los Artículos 16 y 17 de la Ley 1257 de 2008 si corresponde.
+- Justificación breve de cada medida, indicando los literales aplicables del Artículo 5 de la Ley 575 de 2000 (identificados por letra).
+
+RESUELVE:
+PRIMERO: Admitir la solicitud.
+II, III, IV, V, etc.: Ordenar cada medida 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:
+- Prohibición de acercamiento.
+- Prohibición de contacto.
+- Protección de la residencia.
+- Acompañamiento policial.
+- 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.
+
+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.
+
+No incluyas estos ejemplos en la respuesta final.
+No menciones violencia económica ni sexual salvo que se describa expresamente en los hechos.
+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)