Update app.py

app.py

@@ -5,7 +5,7 @@ from transformers import AutoTokenizer, AutoModel, pipeline
 import torch
 import faiss
 import numpy as np
-from spaces import GPU  # Importante para ZeroGPU
+from spaces import GPU  # IMPORTANTE para ZeroGPU
 
 # Token para modelos privados si se requiere
 hf_token = os.getenv("HF_KEY")
@@ -41,7 +41,7 @@ llm = pipeline(
     trust_remote_code=True
 )
 
-@GPU  # DECORADOR para ZeroGPU
+@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."
@@ -55,18 +55,19 @@ def responder(pregunta):
     contexto_final = "\n\n".join(filtrados[:3]) if filtrados else "\n\n".join(result_chunks[:3])
 
     prompt = f"""
-Eres un abogado colombiano especializado en derecho de policía, tránsito y penal.
-Tu única fuente de información es el siguiente texto legal.
-Responde de forma breve, clara y estructurada, sin inventar contenido.
-
-INSTRUCCIONES:
-- No inventes artículos ni leyes.
-- Cita el artículo o frase textual si es posible.
-- Sé directo y sencillo: imagina que respondes a alguien sin formación legal.
-- Si no hay información clara, responde:
-"No encontré información suficiente en los documentos para responder esta pregunta."
-
-TEXTO LEGAL:
+Actúa como un asesor legal colombiano, especializado en el Código de Tránsito, Código de Policía y Código Penal. 
+Analiza el siguiente contexto legal y responde la pregunta de forma clara, completa y profesional.
+
+Instrucciones:
+- Usa un lenguaje formal y preciso.
+- Fundamenta tu respuesta solo en el contexto proporcionado.
+- Cita artículos, normas o sanciones si están presentes en el texto.
+- Finaliza con una recomendación preventiva si aplica.
+- Si no encuentras información suficiente, responde:
+**"No encontré información suficiente en la ley para responder a esta pregunta."**
+- No inventes información ni cites leyes que no estén en el contexto.
+
+CONTEXTO LEGAL:
 {contexto_final}
 
 PREGUNTA: