Update app.py

app.py

@@ -2,14 +2,48 @@ import gradio as gr
 from transformers import pipeline
 
 # Use a pipeline as a high-level helper
-messages = [
-    {"role": "user", "content": "Who are you?"},
-]
 pipe = pipeline("text-generation", model="X-D-Lab/MindChat-Qwen2-0_5B")
-pipe(messages)
 
+# 初始化对话历史
+history = []
+
+# 设置 prompt 模板
+def format_prompt(history, user_input):
+    prompt = ""
+    for i, (user, assistant) in enumerate(history):
+        prompt += f"[Round {i+1}]\n问：{user}\n答：{assistant}\n"
+    prompt += f"[Round {len(history)+1}]\n问：{user_input}\n答："
+    return prompt
+
+# 响应函数
 def respond(message):
-    result = pipe(message, max_new_tokens=100)
-    return result[0]["generated_text"]
+    global history
+    prompt = format_prompt(history, message)
+    result = pipe(prompt, max_new_tokens=200, do_sample=True, temperature=0.7)[0]["generated_text"]
+
+    # 提取模型回答部分（去掉提示部分）
+    if "答：" in result:
+        answer = result.split("答：")[-1].strip()
+    else:
+        answer = result.strip()
+
+    # 更新历史
+    history.append((message, answer))
+    return answer
+
+# 重置对话
+def reset():
+    global history
+    history = []
+    return "已重置对话历史。"
 
-gr.Interface(fn=respond, inputs="text", outputs="text").launch()
+# 创建界面
+gr.Interface(
+    fn=respond,
+    inputs="text",
+    outputs="text",
+    title="🧠 MindChat 多轮对话演示",
+    description="基于 X-D-Lab/MindChat-Qwen2-0_5B 构建的中文聊天机器人，支持多轮上下文记忆。",
+    allow_flagging="never",
+    live=False
+).launch(share=True)