Update app.py

app.py

@@ -66,22 +66,30 @@ def responder(pregunta):
     palabras_clave = pregunta.lower().split()
     palabras_expandidas = expandir_con_sinonimos(palabras_clave, diccionario_sinonimos)
 
-    filtrados = [c for c in result_chunks if any(p in c.lower() for p in palabras_expandidas)]
-    contexto_final = "\n\n".join(filtrados[:3]) if filtrados else "\n\n".join(result_chunks[:3])  
+    def contar_coincidencias(chunk, palabras):
+    return sum(1 for p in palabras if p in chunk.lower())
+
+    filtrados = sorted(
+    result_chunks,
+    key=lambda c: contar_coincidencias(c, palabras_expandidas),
+    reverse=True
+    )
+
+    contexto_final = "\n\n".join(filtrados[:3]) if filtrados else "\n\n".join(result_chunks[:3])
 
     prompt = f"""
-Eres un asistente legal especializado en la legislación colombiana, con conocimientos en el Código de Tránsito, el Código Nacional de Policía y el Código Penal. Tu tarea es responder de forma clara, breve y profesional la siguiente pregunta, utilizando únicamente el contexto legal que se te proporciona. Puedes interpretar el significado de los artículos si su relación con la pregunta es evidente, pero no debes inventar leyes, sanciones ni artículos que no estén presentes en el texto.
+    Eres un asistente legal especializado en la legislación colombiana, con conocimientos en el Código de Tránsito, el Código Nacional de Policía y el Código Penal. Tu tarea es responder de forma clara, breve y profesional la siguiente pregunta, utilizando únicamente el contexto legal que se te proporciona. Puedes interpretar el significado de los artículos si su relación con la pregunta es evidente, pero no debes inventar leyes, sanciones ni artículos que no estén presentes en el texto.
 
-No utilices conocimiento externo, no cites fuentes adicionales, ni incluyas enlaces o advertencias innecesarias. Si el contexto no ofrece información suficiente para responder, dilo de forma directa y concluye. Siempre que sea posible, sintetiza tu respuesta para que sea comprensible incluso para personas sin formación jurídica.
+    No utilices conocimiento externo, no cites fuentes adicionales, ni incluyas enlaces o advertencias innecesarias. Si el contexto no ofrece información suficiente para responder, dilo de forma directa y concluye. Siempre que sea posible, sintetiza tu respuesta para que sea comprensible incluso para personas sin formación jurídica.
 
-CONTEXTO LEGAL:
-{contexto_final}
+    CONTEXTO LEGAL:
+    {contexto_final}
 
-PREGUNTA:
-{pregunta}
+    PREGUNTA:
+    {pregunta}
 
-RESPUESTA:
-"""
+    RESPUESTA:
+    """
 
     resultado = llm(
         prompt,