Update app.py

app.py

@@ -56,16 +56,24 @@ llm = pipeline(
     trust_remote_code=True
 )
 
-@GPU  # DECORADOR para que ZeroGPU ejecute esta función en CPU/GPU remota
+@GPU
 def responder(pregunta):
     if not pregunta:
         return "Por favor ingresa una pregunta."
 
     pregunta_embedding = generar_embedding(pregunta)
-    distances, indices = index.search(pregunta_embedding.reshape(1, -1), k=50)
+    distances, indices = index.search(pregunta_embedding.reshape(1, -1), k=75)
     result_chunks = [chunks[i] for i in indices[0]]
 
-    contexto_final = "\n\n".join(result_chunks[:20])
+    # Expandir sinónimos (opcional)
+    palabras_clave = pregunta.lower().split()
+    palabras_expandidas = expandir_con_sinonimos(palabras_clave, diccionario_sinonimos)
+
+    # Aplicar filtro, pero dejar más chunks como respaldo
+    filtrados = [c for c in result_chunks if any(p in c.lower() for p in palabras_expandidas)]
+    contexto_final = "\n\n".join(filtrados[:20]) if filtrados else "\n\n".join(result_chunks[:20])
+
+    # (El prompt vendría después de esto...)
 
     prompt = f"""
 Eres un experto en el Código de Policía, el Código de Tránsito y el Código Penal de Colombia. Tu función es responder la pregunta del usuario únicamente con base en el TEXTO LEGAL proporcionado como contexto. Debes interpretar el contenido legal y dar una respuesta clara, precisa y fácil de entender, sin inventar sanciones, valores de multas ni artículos que no estén explícitamente mencionados. No agregues enlaces, notas, aclaraciones ni contenido adicional fuera del contexto. Si no encuentras la información en el texto, responde claramente que no se encontró información suficiente en los documentos para responder la pregunta. Sé breve, formal y directo al responder.