Update app.py

app.py

@@ -49,15 +49,13 @@ def generar_embedding(texto):
 
 # LLM para generar respuesta final
 llm = pipeline(
-    "text-generation"
-,
+    "text-generation",
     model="meta-llama/Llama-3.1-8B-Instruct",
     token=hf_token,
     trust_remote_code=True
 )
 
 @GPU  # DECORADOR para que ZeroGPU ejecute esta función en CPU/GPU remota
-
 def responder(pregunta):
     if not pregunta:
         return "Por favor ingresa una pregunta."
@@ -68,20 +66,23 @@ 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[:15]) if filtrados else "\n\n".join(result_chunks[:15])
 
     prompt = f"""
-Eres un abogado colombiano especializado en el Código de Tránsito, Código de Policía y Código Penal. 
-Tu tarea es analizar el siguiente contexto legal y responder a la pregunta de forma clara y precisa.
-
-Instrucciones:
-- Usa únicamente el contexto legal proporcionado para fundamentar tu respuesta.
-- Si encuentras artículos relevantes, cítalos de forma clara.
-- Si el texto no contiene una respuesta exacta, intenta dar una explicación general basada en el contexto.
-- Sé directo, claro y utiliza un lenguaje accesible.
-- Finaliza con una breve recomendación si es pertinente.
+Eres un abogado colombiano especializado en derecho de tránsito, policía y penal.
+Tu única fuente de información es el siguiente CONTEXTO LEGAL.  
+No tienes permitido inventar artículos, leyes o consecuencias que no estén citadas o explícitas en el contexto.
+
+INSTRUCCIONES:
+- Fundamenta tu respuesta exclusivamente en el CONTEXTO LEGAL.
+- Si mencionas consecuencias, citas o artículos, deben estar explícitamente en el texto.
+- No incluyas enlaces web, imágenes, disclaimers institucionales ni logos.
+- No agregues conclusiones si no están fundamentadas.
+- Usa un lenguaje claro, profesional, breve y entendible para una persona sin conocimientos jurídicos.
+- Si no encuentras nada útil en el contexto, responde:
+"No encontré información suficiente en los documentos para responder esta pregunta."
 
 CONTEXTO LEGAL:
 {contexto_final}
@@ -92,13 +93,12 @@ PREGUNTA:
 RESPUESTA:
 """
 
-
     resultado = llm(
         prompt,
         max_new_tokens=500,
         temperature=0.4,
         top_p=0.9,
-        repetition_penalty=1.2
+        repetition_penalty=1.3
     )[0]["generated_text"]
 
     if "RESPUESTA:" in resultado: