Update app.py

app.py

@@ -1,294 +1,141 @@
-import os, json, html, re, random
+import os, json, html, random, requests
 import gradio as gr
 
-# ====== 依赖：DeepSeek（OpenAI 兼容） ======
-_HAS_OPENAI = False
-client = None
-DEEPSEEK_KEY = os.getenv("DEEPSEEK_API_KEY")
-try:
-    from openai import OpenAI
-    if DEEPSEEK_KEY:
-        client = OpenAI(api_key=DEEPSEEK_KEY, base_url="https://api.deepseek.com")
-        _HAS_OPENAI = True
-except Exception:
-    _HAS_OPENAI = False
-
-# ====== 依赖：中文分词 + TFIDF 相似度 ======
-import jieba
-from sklearn.feature_extraction.text import TfidfVectorizer
-from sklearn.metrics.pairwise import cosine_similarity
-
 APP_TITLE = "孙子你好"
-CORPUS_FILE = "sunzi_full.json"   # 全书语料（见下方文件格式示例）
-TOP_K = 5                         # 交给大模型挑选的候选条目数
+DATA_FILE = "playbook.json"
 
-# ---------------- 读全书 + 建索引 ----------------
-def load_corpus():
-    if not os.path.exists(CORPUS_FILE):
-        raise RuntimeError("未找到 sunzi_full.json，请先上传全书语料。")
-    with open(CORPUS_FILE, "r", encoding="utf-8") as f:
-        data = json.load(f)
-    # 规范字段
-    rows = []
-    for i, it in enumerate(data):
-        rows.append({
-            "id": it.get("id", f"q_{i:04d}"),
-            "chapter": it.get("chapter", "").strip(),
-            "quote": it.get("quote", "").strip()
-        })
-    return rows
+# ---------------- 读取知识库 ---------------- #
+def load_kb():
+    if not os.path.exists(DATA_FILE):
+        return None, f"❌ 未找到 {DATA_FILE}。请确保文件名和 app.py 同目录且大小写一致。"
 
-CORPUS = load_corpus()
+    try:
+        with open(DATA_FILE, "r", encoding="utf-8") as f:
+            data = json.load(f)
+    except Exception as e:
+        return None, f"❌ 解析 {DATA_FILE} 出错：{e}"
 
-def jieba_tokenizer(s):
-    return [w for w in jieba.lcut(s) if w.strip()]
+    if not isinstance(data, list) or not data:
+        return None, f"❌ {DATA_FILE} 内容为空或格式不正确。"
 
-DOCS = [f'{r["chapter"]} {r["quote"]}' for r in CORPUS]
-VEC = TfidfVectorizer(tokenizer=jieba_tokenizer, ngram_range=(1,2), min_df=1)
-MATRIX = VEC.fit_transform(DOCS)
+    hint = {
+        "计篇":"先算后战","作战篇":"速战速决","谋攻篇":"上兵伐谋","军形篇":"先固己势",
+        "兵势篇":"势能制胜","虚实篇":"避实击虚","军争篇":"抢占先机","九变篇":"因变制胜",
+        "行军篇":"察情定位","地形篇":"因地用兵","九地篇":"九地九策","火攻篇":"借势放大",
+        "用间篇":"情报为先","始计篇":"先计而后战"
+    }
+    for it in data:
+        it["chapter_hint"] = hint.get(it.get("chapter", ""), "")
+        it["triggers"] = it.get("triggers", [])
+    return data, ""
 
-def topk_candidates(user_text, k=TOP_K):
-    qv = VEC.transform([user_text])
-    sims = cosine_similarity(qv, MATRIX).ravel()
-    idxs = sims.argsort()[::-1][:k]
-    return [CORPUS[i] for i in idxs]
+KB, KB_ERR = load_kb()
 
-# ------------- UI 渲染（沿用你的短模板） -------------
-CHAP_HINT = {
-    "始计篇":"先计而后战","作战篇":"速战速决","谋攻篇":"上兵伐谋","军形篇":"先固己势",
-    "兵势篇":"势能制胜","虚实篇":"避实击虚","军争篇":"抢占先机","九变篇":"因变制胜",
-    "行军篇":"察情定位","地形篇":"因地用兵","九地篇":"九地九策","火攻篇":"借势放大",
-    "用间篇":"情报为先"
-}
+# ---------------- DeepSeek 推理接口 ---------------- #
+DEEPSEEK_API = "https://api.deepseek.com/v1/chat/completions"
+DEEPSEEK_KEY = os.getenv("DEEPSEEK_API_KEY") or "YOUR_API_KEY_HERE"
 
-def render_plan(item, show_speech=False, tip_msg=""):
+def ask_deepseek(prompt):
+    """调用 DeepSeek，若失败返回空字符串"""
+    try:
+        resp = requests.post(
+            DEEPSEEK_API,
+            headers={
+                "Content-Type": "application/json",
+                "Authorization": f"Bearer {DEEPSEEK_KEY}",
+            },
+            json={
+                "model": "deepseek-chat",
+                "messages": [{"role": "user", "content": prompt}],
+                "max_tokens": 800,
+                "temperature": 0.7
+            },
+            timeout=25
+        )
+        data = resp.json()
+        return data.get("choices", [{}])[0].get("message", {}).get("content", "").strip()
+    except Exception as e:
+        return f"⚠️ DeepSeek 出错：{e}"
+
+# ---------------- 兵法匹配逻辑 ---------------- #
+def match_candidates(user_text: str):
+    if not KB:
+        return [], "知识库加载失败。"
+    text = (user_text or "").lower()
+    scored = []
+    for it in KB:
+        score = sum(1 for t in it.get("triggers", []) if t.lower() in text)
+        if score > 0:
+            scored.append((score, it))
+    if not scored:
+        return random.sample(KB, k=min(3, len(KB))), "（未匹配到明确场景，先给几个参考）"
+    scored.sort(key=lambda x: -x[0])
+    top_intent = scored[0][1]["intent"]
+    pool = [it for _, it in scored if it["intent"] == top_intent]
+    random.shuffle(pool)
+    return pool, ""
+
+def render_plan(item, tip_msg=""):
     chap = html.escape(item.get("chapter",""))
-    hint = html.escape(CHAP_HINT.get(chap,""))
+    hint = html.escape(item.get("chapter_hint",""))
     quote= html.escape(item.get("quote",""))
-    insight = html.escape(item.get("insight",""))
-    action = html.escape(item.get("action",""))
-    speech = html.escape(item.get("speech","")).replace("\n","<br>")
-
-    speech_html = f"""
-    <div style="margin-top:12px;padding:12px 14px;border:1px dashed #e2d7f2;border-radius:10px;background:#fbf8ff;">
-      <div style="font-weight:600;">💬 话术示例：</div>
-      <div>{speech}</div>
-    </div>
-    """ if (show_speech and speech) else ""
-
-    tip = f"<div style='color:#9a8fb3;font-size:13px;margin-bottom:6px'>{html.escape(tip_msg)}</div>" if tip_msg else ""
+    idea = html.escape(item.get("idea",""))
+    methods = item.get("methods", [])
+    m_html = "".join([f"<div>{i+1}. {html.escape(m)}</div>" for i,m in enumerate(methods)])
 
+    tip = f"<div style='color:#999;font-size:14px;margin-bottom:4px;'>{html.escape(tip_msg)}</div>" if tip_msg else ""
     return f"""
-    <div style="font-size:22px;line-height:1.7em;">
-      {tip}
+    {tip}
+    <div style='font-size:22px;line-height:1.7em;'>
       🧠 《{chap}》：{hint}<br><br>
       「{quote}」<br><br>
-      ⸻<br><br>
-      <span style="font-weight:600;">启示：</span>{insight}<br><br>
-      <span style="font-weight:600;">做法：</span>{action}
-      {speech_html}
+      <b>思路：</b>{idea}<br><br>
+      <b>做法：</b><br>{m_html}
     </div>
     """
 
-def render_speech_only(item):
-    speech = html.escape(item.get("speech","")).replace("\n", "<br>")
-    return f"""
-    <div style="font-size:22px;line-height:1.8em;padding:20px;background:#fbf8ff;border-radius:12px;border:1px solid #e2d7f2;">
-      <div style="font-weight:600;margin-bottom:10px;">💬 话术示例：</div>
-      <div>{speech}</div>
-    </div>
-    """
-
-def pack_outputs(html_out, pool, idx, show_flag, show_switch, show_speech_btn, show_back, show_again, user_text_keep=True):
-    return (
-        html_out,
-        pool,
-        idx,
-        show_flag,
-        gr.update(visible=show_switch),
-        gr.update(visible=show_speech_btn),
-        gr.update(visible=show_back),
-        gr.update(visible=show_again),
-        (gr.update() if user_text_keep else gr.update(value=""))
-    )
-
-# ------------- DeepSeek 受限生成：仅在 Top-K 中选 -------------
-SYS_LOCKED = (
-    "你是中文军师，用《孙子兵法》解决现实问题。"
-    "系统会给你一个候选列表（含编号/篇名/原文）。你必须："
-    "1) 只在候选中选出【一条】最贴切原文（不可更换或自造原句）；"
-    "2) 输出一个 JSON 对象，字段仅限："
-    "id（所选编号）、chapter（篇名）、quote（原文原句）、"
-    "insight（≤18字，人话、直白精炼）、"
-    "action（仅1条，≤40字、具体到动作/步骤）、"
-    "speech（≤80字、可对对方直说；礼貌、委婉、幽默有边界）。"
-    "严格禁止输出除 JSON 外的任何内容。"
-)
-
-FEWSHOT = [
-    {
-        "role":"user",
-        "content":(
-            "【候选】\n"
-            "1) 作战篇｜夫兵久而国利者，未之有也。\n"
-            "2) 虚实篇｜避实而击虚。\n"
-            "【用户问题】老板让我临时加班但我状态很差，如何拒绝更体面？\n"
-            "只输出JSON。"
-        )
-    },
-    {
-        "role":"assistant",
-        "content":json.dumps({
-            "id":"1",
-            "chapter":"作战篇",
-            "quote":"夫兵久而国利者，未之有也。",
-            "insight":"别耗死自己，用质量说服。",
-            "action":"给“今晚初稿/明早终版”两选项，让对方决定。",
-            "speech":"老板，我担心今晚赶出来质量不稳。不如我明早7:30交完整版本，保证更稳，您看行吗？"
-        }, ensure_ascii=False)
-    }
-]
-
-def _extract_json(text: str) -> dict:
-    m = re.search(r"\{.*\}", text, flags=re.S)
-    js = m.group(0) if m else text
-    return json.loads(js)
-
-def _fallback_from_top1(cands, tip="使用本地兜底（可能未连接API）"):
-    if not cands:
-        return {
-            "item": {"chapter":"始计篇","quote":"先计而后战。","insight":"先摸清再出手。","action":"列两种稳妥方案并承诺时间点。","speech":"我们按更稳的节奏推进，彼此都放心。"},
-            "tip": tip
-        }
-    c = cands[0]
-    return {
-        "item": {"chapter": c["chapter"], "quote": c["quote"], "insight":"先捋清状况再出手。","action":"定目标与底线，给一个具体步骤。","speech":"我这边按更稳的节奏推进，给你一个清晰方案。"},
-        "tip": tip
-    }
+# ---------------- 主逻辑 ---------------- #
+def on_generate(user_text):
+    if KB_ERR:
+        return f"<div style='color:red'>{KB_ERR}</div>"
 
-def deepseek_choose_and_write(user_text: str, cands: list, alt: bool=False):
-    """
-    把 TF-IDF Top-K 候选喂给 DeepSeek，只允许从中选一条并生成短模板。
-    alt=True 时提示换表达，但仍限定在同一候选集。
-    """
-    listing = "\n".join([f'{i+1}) {c["chapter"]}｜{c["quote"]}' for i,c in enumerate(cands)])
-    ask = f"【候选】\n{listing}\n【用户问题】{user_text}\n只输出JSON。"
-    if alt:
-        ask += "\n请换一种表述思路（仍只能在候选内选），避免与上次措辞重复。"
-
-    if not _HAS_OPENAI:
-        fb = _fallback_from_top1(cands, "未检测到 DEEPSEEK_API_KEY，已使用本地兜底。")
-        return fb["item"], fb["tip"]
-
-    try:
-        msgs = [{"role":"system","content":SYS_LOCKED}] + FEWSHOT + [{"role":"user","content":ask}]
-        resp = client.chat.completions.create(
-            model="deepseek-chat",
-            messages=msgs,
-            temperature=0.25 if not alt else 0.55,
-            max_tokens=420,
-            timeout=30
-        )
-        data = _extract_json(resp.choices[0].message.content.strip())
-        sel_id = str(data.get("id","1")).strip()
-        # 把编号对应回候选，确保不越界
-        try:
-            idx = max(1, min(len(cands), int(re.sub(r"[^0-9]", "", sel_id) or "1"))) - 1
-        except Exception:
-            idx = 0
-        c = cands[idx]
-        item = {
-            "chapter": c["chapter"],
-            "quote": c["quote"],
-            "insight": str(data.get("insight","先捋清状况再出手。")).strip(),
-            "action": str(data.get("action","给出一个具体且可执行的步骤。")).strip(),
-            "speech": str(data.get("speech","我们按更稳的节奏推进，彼此都放心。")).strip()
-        }
-        return item, ""
-    except Exception as e:
-        fb = _fallback_from_top1(cands, f"已使用兜底（{type(e).__name__}）")
-        return fb["item"], fb["tip"]
-
-# ----------------- 回调（维持你的按钮可见逻辑） -----------------
-def on_generate(user_text, state_pool, state_idx, state_show):
-    user_text = (user_text or "").strip()
-    if not user_text:
-        return pack_outputs("<div style='font-size:20px;'>请先输入你的问题。</div>", [], 0, False, False, False, False, False, True)
-    cands = topk_candidates(user_text, TOP_K)
-    item, tip = deepseek_choose_and_write(user_text, cands, alt=False)
-    pool = [item]
-    html_out = render_plan(item, False, tip)
-    return pack_outputs(html_out, pool, 0, False, True, True, False, True, True)
-
-def on_switch(state_pool, state_idx, state_show, user_text):
-    user_text = (user_text or "").strip()
-    if not user_text:
-        return pack_outputs("<div style='font-size:20px;'>请先输入你的问题并点击生成兵法。</div>", [], 0, False, False, False, False, False, True)
-    cands = topk_candidates(user_text, TOP_K)
-    item, tip = deepseek_choose_and_write(user_text, cands, alt=True)
-    pool = [item]
-    html_out = render_plan(item, False, tip)
-    return pack_outputs(html_out, pool, 0, False, True, True, False, True, True)
-
-def on_show_speech(state_pool, state_idx, state_show):
-    pool = state_pool or []
+    pool, tip = match_candidates(user_text)
     if not pool:
-        return pack_outputs("<div style='font-size:20px;'>请先输入你的问题并点击生成兵法。</div>", pool, state_idx, state_show, False, False, False, False, True)
-    idx = max(0, min(state_idx, len(pool)-1))
-    html_out = render_speech_only(pool[idx])
-    return pack_outputs(html_out, pool, idx, True, True, True, True, True, True)
-
-def on_back(state_pool, state_idx, state_show):
-    pool = state_pool or []
-    if not pool:
-        return pack_outputs("<div style='font-size:20px;'>请先输入你的问题并点击生成兵法。</div>", pool, state_idx, state_show, False, False, False, False, True)
-    idx = max(0, min(state_idx, len(pool)-1))
-    html_out = render_plan(pool[idx], False, "")
-    return pack_outputs(html_out, pool, idx, False, True, True, False, True, True)
+        return "<div>❌ 没找到合适的兵法。</div>"
+
+    # prompt: 让 DeepSeek 输出孙子风格回答
+    chosen = pool[0]
+    prompt = f"""
+你是一位幽默、聪明、有边界的中文军师。
+结合《孙子兵法》中“{chosen['chapter']} - {chosen['quote']}”的智慧，
+回答以下问题：“{user_text}”。
+请给出一段简短建议（100字内），风格机智、委婉、有策略。
+"""
+    ai_reply = ask_deepseek(prompt)
 
-def on_again(state_pool, state_idx, state_show):
-    return pack_outputs("", [], 0, False, False, False, False, False, user_text_keep=False)
+    html_out = render_plan(chosen, tip)
+    html_ai = f"<div style='margin-top:20px;padding:15px;background:#fbf8ff;border-radius:10px;border:1px dashed #ccc;'><b>💬 孙子点评：</b><br>{html.escape(ai_reply)}</div>"
+    return html_out + html_ai
 
-# ----------------- UI -----------------
-theme = gr.themes.Soft(primary_hue="pink", neutral_hue="slate")
+# ---------------- UI ---------------- #
+theme = gr.themes.Soft(primary_hue="pink", neutral_hue="gray")
 CSS = """
 body,.gradio-container{background:#f7f6fb!important;}
 .header{font-weight:900;font-size:28px;color:#111;text-align:center;padding:24px 0;}
-.card{background:#fff;border:1px solid #eddcf3;border-radius:14px; box-shadow:0 4px 18px rgba(0,0,0,.05);padding:16px;}
+.card{background:#fff;border:1px solid #eddcf3;border-radius:14px;
+box-shadow:0 4px 18px rgba(0,0,0,.05);padding:16px;}
 """
 
-with gr.Blocks(title=APP_TITLE, theme=theme, css=CSS, analytics_enabled=False) as demo:
+with gr.Blocks(title=APP_TITLE, theme=theme, css=CSS) as demo:
     gr.HTML("<div class='header'>孙子你好</div>")
-    with gr.Group(elem_classes=["card"]):
-        user_text = gr.Textbox(label="你的问题", placeholder="例如：朋友借钱不还；老板让我加班；邻居太吵；甲方拖款…", lines=3)
-        btn_gen   = gr.Button("🎯 生成兵法", variant="primary")
-
-        with gr.Row():
-            btn_switch = gr.Button("♻️ 换兵法", visible=False)
-            btn_speech = gr.Button("💬 显示示例话术", visible=False)
-            btn_back   = gr.Button("↩️ 返回兵法", visible=False)
-            btn_again  = gr.Button("🧭 问孙子其他问题", visible=False)
-
-        out = gr.HTML()
-
-        state_pool  = gr.State([])
-        state_idx   = gr.State(0)
-        state_show  = gr.State(False)
-
-        btn_gen.click(on_generate, inputs=[user_text, state_pool, state_idx, state_show],
-                      outputs=[out, state_pool, state_idx, state_show, btn_switch, btn_speech, btn_back, btn_again, user_text])
-
-        btn_switch.click(on_switch, inputs=[state_pool, state_idx, state_show, user_text],
-                         outputs=[out, state_pool, state_idx, state_show, btn_switch, btn_speech, btn_back, btn_again, user_text])
-
-        btn_speech.click(on_show_speech, inputs=[state_pool, state_idx, state_show],
-                         outputs=[out, state_pool, state_idx, state_show, btn_switch, btn_speech, btn_back, btn_again, user_text])
-
-        btn_back.click(on_back, inputs=[state_pool, state_idx, state_show],
-                       outputs=[out, state_pool, state_idx, state_show, btn_switch, btn_speech, btn_back, btn_again, user_text])
-
-        btn_again.click(on_again, inputs=[state_pool, state_idx, state_show],
-                        outputs=[out, state_pool, state_idx, state_show, btn_switch, btn_speech, btn_back, btn_again, user_text])
+    if KB_ERR:
+        gr.HTML(f"<div style='color:red'>{KB_ERR}</div>")
+    else:
+        with gr.Group(elem_classes=["card"]):
+            q = gr.Textbox(label="你的问题", placeholder="例如：朋友借钱不还、邻居太吵、老板让我加班…", lines=3)
+            btn = gr.Button("🎯 生成兵法", variant="primary")
+            out = gr.HTML()
+            btn.click(on_generate, inputs=q, outputs=out)
 
 if __name__ == "__main__":
     demo.launch()