From 0ff0aff87b71a29d9285721554b492f868730820 Mon Sep 17 00:00:00 2001
From: hz4th_coder <hz4th_coder@tphai.com>
Date: Mon, 17 Aug 2026 21:38:29 +0800
Subject: [PATCH] =?UTF-8?q?v1.2.3=20=E5=BA=95=E9=83=A8=E5=BF=AB=E6=8D=B7?=
 =?UTF-8?q?=E9=97=AE=E9=A2=98AI=E9=A2=84=E6=B5=8B=EF=BC=9A=E5=AF=B9?=
 =?UTF-8?q?=E8=AF=9D=E8=BF=9B=E8=A1=8C=E4=B8=AD=E7=94=B1=E5=A4=A7=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B=E9=A2=84=E6=B5=8B=E7=94=A8=E6=88=B7=E5=8F=AF=E8=83=BD?=
 =?UTF-8?q?=E8=BF=BD=E9=97=AE=E7=9A=84=E9=97=AE=E9=A2=98(=E6=AF=8F?=
 =?UTF-8?q?=E4=B8=AA=E2=89=A430=E5=AD=97)=EF=BC=8C=E4=B8=AA=E6=95=B0?=
 =?UTF-8?q?=E5=90=8E=E5=8F=B0=E5=8F=AF=E9=85=8D(=E9=BB=98=E8=AE=A43?=
 =?UTF-8?q?=EF=BC=8C1-6)=EF=BC=8C=E5=BC=82=E6=AD=A5=E5=88=B7=E6=96=B0?=
 =?UTF-8?q?=E9=98=B2=E4=B8=B2=E6=89=B0=EF=BC=9B=E5=88=9D=E5=A7=8B=E4=B8=8E?=
 =?UTF-8?q?=E5=BC=80=E5=9C=BA=E7=99=BD=E4=B8=80=E8=87=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 admin.py          |  1 +
 api.py            | 16 +++++++++++++++
 chat.py           | 50 +++++++++++++++++++++++++++++++++++++++++++++++
 static/admin.html |  4 ++++
 static/admin.js   |  2 ++
 static/app.js     | 16 +++++++++++++++
 6 files changed, 89 insertions(+)

diff --git a/admin.py b/admin.py
index d3cb1d7..b23806c 100644
--- a/admin.py
+++ b/admin.py
@@ -53,6 +53,7 @@ DEFAULT_CONFIG = {
     "footer_text": "NBA球迷大全 · 数据为模拟演示数据（2025-26 赛季） · LLM: DeepSeek · 向量: Chroma + bge-large-zh",
     "admin_password": "admin123",
     "entity_mark_mode": "first",   # 实体标记：first=只标记首次出现 / all=全部标记
+    "suggestion_count": "3",       # 对话中底部快捷问题预测个数（默认3）
 }
 
 SEARCHABLE = {  # 每个表可搜索的 TEXT 字段
diff --git a/api.py b/api.py
index 03d93f8..761a09d 100644
--- a/api.py
+++ b/api.py
@@ -64,6 +64,22 @@ def boot():
     return jsonify(chat.boot_info())
 
 
+@app.route("/api/suggest", methods=["POST"])
+def api_suggest():
+    """基于对话历史预测底部快捷问题（个数后台可配，默认3）"""
+    body = request.get_json(force=True, silent=True) or {}
+    history = body.get("history") or []
+    try:
+        n = int(admin_mod.get_config().get("suggestion_count", "3") or "3")
+    except Exception:
+        n = 3
+    try:
+        return jsonify({"suggestions": chat.predict_suggestions(history, n)})
+    except Exception as e:
+        log.exception("suggest error")
+        return jsonify({"suggestions": chat.suggest_questions()[:n]}), 200
+
+
 # ------------------------------------------------------------------ 对话
 @app.route("/api/chat", methods=["POST"])
 def api_chat():
diff --git a/chat.py b/chat.py
index e2229a7..09888a0 100644
--- a/chat.py
+++ b/chat.py
@@ -272,3 +272,53 @@ def boot_info():
         "suggestions": suggest_questions(),
         "footer_text": cfg.get("footer_text", "NBA球迷大全 · 数据为模拟演示数据（2025-26 赛季）"),
     }
+
+
+def _parse_json_array(text):
+    """从 LLM 输出中解析 JSON 数组（容错：直接 JSON / 提取中括号段）"""
+    if not text:
+        return []
+    text = text.strip()
+    try:
+        arr = json.loads(text)
+        if isinstance(arr, list):
+            return arr
+    except Exception:
+        pass
+    m = re.search(r"\[.*\]", text, re.S)
+    if m:
+        try:
+            arr = json.loads(m.group(0))
+            if isinstance(arr, list):
+                return arr
+        except Exception:
+            pass
+    return []
+
+
+def predict_suggestions(history=None, n=3):
+    """基于对话历史，让大模型预测用户接下来最可能追问的 n 个问题（底部快捷语句）。
+    每个问题不超过 30 字；LLM 异常时回退到默认快捷问题。"""
+    n = max(1, min(int(n or 3), 6))
+    history = history or []
+    msgs = [{"role": "system", "content": (
+        f"你是「NBA球迷大全」智能助手。根据对话历史，站在用户角度预测他接下来最可能追问的{n}个问题。\n"
+        "要求：\n"
+        "1. 每个问题不超过30个汉字，简洁口语化\n"
+        "2. 必须是用户会直接发送的提问，不要编号、不要引号、不要解释\n"
+        "3. 只输出JSON数组，例如：[\"库里今天拿了几分\",\"湖人下一场什么时候\"]，不要输出任何其他内容")}]
+    for h in history[-6:]:
+        msgs.append({"role": "user", "content": h.get("user", "")})
+        if h.get("assistant"):
+            msgs.append({"role": "assistant", "content": str(h["assistant"])[:600]})
+    if len(msgs) == 1:
+        return DEFAULT_SUGGESTIONS[:n]
+    try:
+        resp = llm.chat(msgs, temperature=0.9, max_tokens=200)
+        arr = _parse_json_array(llm.parse_content(resp))
+        out = [str(x).strip()[:30] for x in arr if str(x).strip()][:n]
+        if out:
+            return out
+    except Exception as e:
+        log.warning("快捷问题预测失败(%s)，回退默认", e)
+    return DEFAULT_SUGGESTIONS[:n]
diff --git a/static/admin.html b/static/admin.html
index 1fc47af..f7267cf 100644
--- a/static/admin.html
+++ b/static/admin.html
@@ -157,6 +157,10 @@ td.num { text-align:center; }
           <textarea id="cfg-suggestions" style="min-height:160px"></textarea>
           <div class="tip">💡 一行一个问题。保存后刷新前台页面即可看到新的快捷问题。</div>
         </div>
+        <div class="row2">
+          <div><label>对话中底部快捷问题预测个数（默认3，范围1-6）</label><input type="number" id="cfg-suggestion_count" min="1" max="6">
+          <div class="tip">对话进行中，大模型根据上下文预测用户可能追问的问题数量（每个≤30字）</div></div>
+        </div>
         <div class="row2">
           <div><label>管理员密码（留空则不修改）</label><input type="password" id="cfg-admin_password" placeholder="••••••"></div>
           <div><label>实体标记模式</label><select id="cfg-entity_mark_mode">
diff --git a/static/admin.js b/static/admin.js
index 106d1f3..aa225f0 100644
--- a/static/admin.js
+++ b/static/admin.js
@@ -208,6 +208,7 @@ async function loadConfig() {
   $("#cfg-footer_text").value = d.footer_text || "";
   $("#cfg-admin_password").value = "";
   $("#cfg-entity_mark_mode").value = d.entity_mark_mode === "all" ? "all" : "first";
+  $("#cfg-suggestion_count").value = d.suggestion_count || "3";
   try { $("#cfg-suggestions").value = JSON.parse(d.suggestions || "[]").join("\n"); }
   catch (e) { $("#cfg-suggestions").value = ""; }
   $("#cfg-msg").textContent = "";
@@ -224,6 +225,7 @@ $("#cfg-save").addEventListener("click", async () => {
   const pw = $("#cfg-admin_password").value;
   if (pw) payload.admin_password = pw;
   payload.entity_mark_mode = $("#cfg-entity_mark_mode").value;
+  payload.suggestion_count = String(Math.min(6, Math.max(1, parseInt($("#cfg-suggestion_count").value || "3"))));
   try {
     await api("/api/admin/config", { method: "PUT", body: JSON.stringify(payload) });
     $("#cfg-msg").textContent = "✅ 保存成功，前台刷新后生效";
diff --git a/static/app.js b/static/app.js
index 4e9940b..9765a1b 100644
--- a/static/app.js
+++ b/static/app.js
@@ -122,6 +122,9 @@ async function sendChat(text) {
     addMsg("bot", html);
     chatHistory.push({ user: text, assistant: d.reply });
     if (chatHistory.length > 20) chatHistory.splice(0, chatHistory.length - 20);
+    // 大模型预测下一轮快捷问题（异步刷新底部 chips）
+    const mark = chatHistory.length;
+    refreshChips(mark);
   } catch (e) {
     typing.remove();
     addMsg("bot", "⚠️ 网络异常，请稍后再试。");
@@ -139,6 +142,19 @@ function askQuick(q) {
   sendChat(q);
 }
 
+/* 底部快捷问题：对话进行中由大模型预测用户可能追问的问题（异步刷新） */
+async function refreshChips(mark) {
+  try {
+    const r = await fetch("/api/suggest", { method: "POST", headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ history: chatHistory.slice(-6) }) });
+    const d = await r.json();
+    if (!d.suggestions || !d.suggestions.length) return;
+    if (mark !== chatHistory.length) return;   // 期间用户又发了新消息 → 丢弃过期预测
+    $("#chips").innerHTML = d.suggestions.map((q) => `<button>${esc(q)}</button>`).join("");
+    $$("#chips button").forEach((b) => b.addEventListener("click", () => askQuick(b.textContent)));
+  } catch (e) {}
+}
+
 /* 聊天区事件委托：快捷语句 / 实体标记 / 新闻链接 / 卡片 */
 $("#chat-list").addEventListener("click", (e) => {
   const q = e.target.closest(".quick-q");