Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py
 import os
 from llama_cpp import Llama
 from huggingface_hub import hf_hub_download
@@ -19,12 +18,19 @@ def ensure_model():
 model_path = ensure_model()
 llm = Llama(model_path=model_path, n_ctx=2048, n_threads=4, verbose=False)
 
+# 添加固定的系统提示词，定义身份和语言要求
+SYSTEM_PROMPT = (
+    "你是一名专业的信息安全助手，"
+    "请始终使用中文回答所有问题，"
+    "确保回答专业、准确、简洁。"
+)
+
 def chat_fn(user_message, history):
     history = history or []
-    prompt = ""
+    prompt = SYSTEM_PROMPT + "\n"
     for u, a in history:
-        prompt += f"User: {u}\nAssistant: {a}\n"
-    prompt += f"User: {user_message}\nAssistant:"
+        prompt += f"用户: {u}\n助手: {a}\n"
+    prompt += f"用户: {user_message}\n助手:"
 
     resp = llm(prompt, max_tokens=256, temperature=0.7, top_p=0.9)
     text = resp["choices"][0]["text"].strip()
@@ -32,7 +38,7 @@ def chat_fn(user_message, history):
     return history, ""
 
 with gr.Blocks(css=".gradio-container { max-width: 700px; margin: auto; }") as demo:
-    gr.Markdown("## 🤖 Nemotron‑Qwen‑1.5B 推理对话 Demo")
+    gr.Markdown("## 信息安全助手")
     chatbot = gr.Chatbot(elem_id="chatbot")
     user_input = gr.Textbox(placeholder="输入消息，按 Enter 发送", lines=1)
     user_input.submit(chat_fn, [user_input, chatbot], [chatbot, user_input])