Update app.py

app.py

@@ -1,7 +1,7 @@
-import os, json, html, requests, re
+import os, json, html, requests, re, random
 import gradio as gr
 
-APP_TITLE = "孙子你好（DeepSeek 直连·诊断版）"
+APP_TITLE = "孙子你好（DeepSeek 多样化·诊断版）"
 
 # 可通过 Secrets 覆盖：
 #   DEEPSEEK_API_KEY   = sk-xxxxxxxx
@@ -12,20 +12,28 @@ APP_TITLE = "孙子你好（DeepSeek 直连·诊断版）"
 MODEL_NAME = os.getenv("DEEPSEEK_MODEL", "deepseek-chat").strip() or "deepseek-chat"
 
 SYS_PROMPT = (
-    "你是中文军师，严格依据《孙子兵法》为现实小问题给出策略。\n"
+    "你是中文军师，擅长将《孙子兵法》的智慧用于现实问题。\n"
+    "任务：针对用户的问题，从《孙子兵法》十三篇中找出最贴合的一章，引用原文（若能确定），并转化为现代生活策略。\n\n"
     "规则：\n"
-    "1) 优先引用《孙子兵法》原文句子，并指明篇名（如：谋攻篇、虚实篇）。\n"
-    "2) 如果无法百分之百确认出典或篇名，不要硬编，改为意译/思想要点，并标记 unsure=true。\n"
-    "3) 仅输出 JSON，固定字段：\n"
-    '   {\"chapter\":\"<篇名或空>\",\"quote\":\"<原文或意译>\",\"unsure\":false/true,'
-    '\"insight\":\"<一句人话启示≤24字>\",\"action\":\"<一步可执行做法≤100字>\",'
-    '\"speech\":\"<可直接对话的话术≤120字>\"}\n'
-    "4) 风格：简洁、机智、委婉、有边界；不得人身攻击、威胁或违法建议。\n"
-    "5) 严格仅输出 JSON，不要任何多余文字。"
+    "1. 优先根据问题性质挑选不同篇章：\n"
+    "   - 计划/评估：〈始计篇〉〈作战篇〉\n"
+    "   - 谈判/攻防：〈谋攻篇〉〈虚实篇〉\n"
+    "   - 节奏/风险：〈军形篇〉〈兵势篇〉\n"
+    "   - 主动出击/竞争：〈军争篇〉〈九变篇〉〈地形篇〉\n"
+    "   - 困境/沟通：〈行军篇〉〈九地篇〉〈用间篇〉〈火攻篇〉\n"
+    "2. 若问题重复，尽量换不同篇章或换角度回答，保持多样性。\n"
+    "3. 若无法百分之百确认原文或篇名，不要硬编，改为思想意译并标记 unsure=true。\n"
+    "4. 严格仅输出 JSON，固定字段：\n"
+    "   {\"chapter\":\"<篇名>\",\"quote\":\"<原文或思想>\",\"unsure\":false/true,"
+    "\"insight\":\"<一句人话启示≤24字>\",\"action\":\"<一步可执行做法≤100字>\","
+    "\"speech\":\"<直接可说的话术≤120字>\"}\n"
+    "5. 语气简洁、机智、委婉、有边界；不得人身攻击、威胁或违法建议。\n"
+    "6. 禁止输出 JSON 以外的任何文字。"
 )
 
-# few-shot 用 json.dumps 生成，避免手写引号带来语法错误
+# few-shot：覆盖多篇章，减少模型只用某一章的倾向
 FEWSHOT = [
+    # 加班 → 作战篇
     {
         "role": "user",
         "content": "问题：老板临时要我通宵加班，我状态很差，怎么体面拒绝？只输出JSON。"
@@ -33,14 +41,15 @@ FEWSHOT = [
     {
         "role": "assistant",
         "content": json.dumps({
-            "chapter": "虚实篇",
-            "quote": "避实而击虚。",
+            "chapter": "作战篇",
+            "quote": "夫兵久而国利者，未之有也。",
             "unsure": False,
-            "insight": "别硬碰命令，转打结果。",
-            "action": "给出“今晚草稿/明早终版”二选一，引导选择后者以保质量和节奏。",
-            "speech": "我今晚能出草稿，也能明早交更稳的终版。为保证质量，我建议后一种，您看呢？"
+            "insight": "避免持久消耗，保整体战力。",
+            "action": "提出“明早高质量交付”的替代方案，比熬夜更稳更可靠。",
+            "speech": "今晚赶出来质量可能不稳，我建议明早整理成终版，保证更可靠，您看可以吗？"
         }, ensure_ascii=False)
     },
+    # 借钱 → 始计篇
     {
         "role": "user",
         "content": "问题：朋友借钱不还，怎样体面要回？只输出JSON。"
@@ -52,14 +61,46 @@ FEWSHOT = [
             "quote": "兵者，诡道也。",
             "unsure": False,
             "insight": "绕开硬催，设局顺势收回。",
-            "action": "先说明自己在清账安排资金，再给“部分先还/分期”台阶，降低对抗。",
-            "speech": "我这边在清账需要安排下，方便的话先转一部分？分期也行，别影响咱们交情。"
+            "action": "以“清账/安排资金”为由先要部分或分期，降低对抗。",
+            "speech": "我这边在清账需要安排下，能否先转一部分？分期也行，别影响咱们交情。"
+        }, ensure_ascii=False)
+    },
+    # 邻居噪音 → 行军篇
+    {
+        "role": "user",
+        "content": "问题：邻居太吵，怎么友好沟通？只输出JSON。"
+    },
+    {
+        "role": "assistant",
+        "content": json.dumps({
+            "chapter": "行军篇",
+            "quote": "知彼知己，百战不��。（通用于沟通前的情势判断）",
+            "unsure": True,  # 行军篇原文无此句，标不确定，演示 unsure 机制
+            "insight": "先辨来源时机，再择机沟通。",
+            "action": "选非冲突时段，先体谅再提困扰，争取对方自觉收敛。",
+            "speech": "最近晚上声音有点明显，我怕打扰到楼上楼下。能否帮忙晚间稍微注意下？谢谢啦～"
+        }, ensure_ascii=False)
+    },
+    # 职场竞争 → 军争篇
+    {
+        "role": "user",
+        "content": "问题：同事跟我抢功劳，怎么应对？只输出JSON。"
+    },
+    {
+        "role": "assistant",
+        "content": json.dumps({
+            "chapter": "军争篇",
+            "quote": "以迂为直，以患为利。",
+            "unsure": False,
+            "insight": "不硬碰，转势为己用。",
+            "action": "会前先递交要点纪要和分工佐证，令功劳记录化、流程化。",
+            "speech": "我把这次我的负责点和关键产出提前发一份纪要，方便会里对齐，避免遗漏。"
         }, ensure_ascii=False)
     }
 ]
 
 def call_deepseek(user_question: str, alt: bool = False) -> dict:
-    """带精确诊断：自动尝试两个端点，页面/日志展示状态码与返回体。"""
+    """带精确诊断：自动尝试两个端点，页面/日志展示状态码与返回体；加入多样化提示。"""
     key = os.getenv("DEEPSEEK_API_KEY", "").strip()
     if not key:
         return {
@@ -77,7 +118,16 @@ def call_deepseek(user_question: str, alt: bool = False) -> dict:
         "https://api.deepseek.com/chat/completions",
     ]
 
-    msgs = [{"role": "system", "content": SYS_PROMPT}] + FEWSHOT + [
+    # 多样化提示：鼓励换篇章/换角度
+    diversity_hint = random.choice([
+        "尽量避免重复使用同一篇章，尝试换篇章回答。",
+        "优先考虑《军形篇》《兵势篇》《九变篇》《地形篇》的角度。",
+        "如果问题可多解，换一个不同篇章来解释。",
+        "引用少见但贴合的篇章，保持回答多样性。",
+        "可从节奏、态势、资源、时机等不同维度选篇章。"
+    ])
+
+    msgs = [{"role": "system", "content": SYS_PROMPT + "\n" + diversity_hint}] + FEWSHOT + [
         {"role": "user", "content": f"问题：{user_question} 只输出JSON。"}
     ]
 
@@ -97,7 +147,7 @@ def call_deepseek(user_question: str, alt: bool = False) -> dict:
                 json=payload,
                 timeout=60
             )
-            print(f"[DeepSeek] endpoint={ep} status={r.status_code} model={MODEL_NAME}")  # 写入 Runtime logs
+            print(f"[DeepSeek] endpoint={ep} status={r.status_code} model={MODEL_NAME}")  # Runtime logs
 
             if r.status_code // 100 != 2:
                 try:
@@ -113,19 +163,17 @@ def call_deepseek(user_question: str, alt: bool = False) -> dict:
             js = m.group(0) if m else content
             result = json.loads(js)
 
-            # 规整长度，防止偶发超长
             return {
                 "chapter": result.get("chapter", "")[:20],
-                "quote": result.get("quote", "")[:160],
+                "quote": result.get("quote", "")[:200],
                 "unsure": bool(result.get("unsure", False)),
                 "insight": result.get("insight", "")[:40],
-                "action": result.get("action", "")[:100],
-                "speech": result.get("speech", "")[:120]
+                "action": result.get("action", "")[:120],
+                "speech": result.get("speech", "")[:140]
             }
         except Exception as e:
             last_err = f"{type(e).__name__}: {e}"
 
-    # 两个端点都失败
     return {
         "chapter": "",
         "quote": "为避免误引，已隐藏原文与篇名（模型不确定出典）。",