Update app.py

app.py

@@ -30,8 +30,13 @@ def consulta_penalista(mensaje, tipo_consulta):
     """Procesa una consulta jurídica y devuelve la respuesta del modelo."""
     global chat_history
     
-    # Añadir el mensaje del usuario al historial
-    chat_history.append({"role": "user", "content": f"[{tipo_consulta}] {mensaje}"})
+    # Crear un prompt que indique directamente qué responder
+    prompt = f"""A continuación hay una consulta jurídica sobre el sistema penal español.
+
+Consulta: {mensaje}
+
+Respuesta sobre el delito consultado según la legislación española:
+- Artículos del Código Penal español: """
     
     try:
         # Crear el prompt completo (sistema + historial)
@@ -45,11 +50,11 @@ def consulta_penalista(mensaje, tipo_consulta):
             num_return_sequences=1,  # Número de respuestas a generar
 
             # (NUEVO)
-            temperature=0.7,  # Controlar la creatividad (valores más bajos = más determinista)
+            temperature=0.5,  # Controlar la creatividad (valores más bajos = más determinista)
             top_p=0.9,  # Usar muestreo de núcleo (nucleus sampling)
-
             truncation=True,
-            pad_token_id=tokenizer.eos_token_id
+            pad_token_id=tokenizer.eos_token_id,
+            do_sample=True
         )
         
         respuesta_texto = respuesta[0]["generated_text"]