Blame for app.py

import os, json, html, requests, re, random
import gradio as gr

APP_TITLE = "孙子你好（DeepSeek 多样化·诊断版）"

# 可通过 Secrets 覆盖：
#   DEEPSEEK_API_KEY   = sk-xxxxxxxx
#   DEEPSEEK_MODEL     = deepseek-chat（默认）
#   DEEPSEEK_BASE_URL  = https://api.deepseek.com/v1/chat/completions（多数账号）
#                        或 https://api.deepseek.com/chat/completions（少量旧线路）

MODEL_NAME = os.getenv("DEEPSEEK_MODEL", "deepseek-chat").strip() or "deepseek-chat"

# ===================== 系统提示词（增加 explain_cn 字段 + 做法语气要求） =====================
SYS_PROMPT = (
    "你是中文军师，擅长将《孙子兵法》的智慧用于现实问题。\n"
    "任务：针对用户的问题，从《孙子兵法》十三篇中找出最贴合的一章，引用原文（若能确定），并转化为现代生活策略。\n\n"
    "规则：\n"
    "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字>\","
    "    \"explain_cn\":\"<用非常简单的话解释这句名言的大意≤60字>\"}\n"
    "5. 语气简洁、机智、委婉、有边界；不得人身攻击、威胁或违法建议。\n"
    "6. 禁止输出 JSON 以外的任何文字。"
)

# few-shot：覆盖多篇章，减少模型只用某一章的倾向
FEWSHOT = [
    # 加班 → 作战篇
    {
        "role": "user",
        "content": "问题：老板临时要我通宵加班，我状态很差，怎么体面拒绝？只输出JSON。"
    },
    {
        "role": "assistant",
        "content": json.dumps({
            "chapter": "作战篇",
            "quote": "夫兵久而国利者，未之有也。",
            "unsure": False,
            "insight": "避免持久消耗，保整体战力。",
            "action": "跟老板提一个“明早高质量交付”的方案，比硬熬一整晚靠谱多了，也能保证你明天还是个清醒的人。",
            "speech": "今晚赶出来质量可能不稳，我建议明早整理成终版，保证更可靠，您看可以吗？",
            "explain_cn": "意思是战争拖太久没有好处，打仗要快准狠，不能一直消耗。"
        }, ensure_ascii=False)
    },
    # 借钱 → 始计篇
    {
        "role": "user",
        "content": "问题：朋友借钱不还，怎样体面要回？只输出JSON。"
    },
    {
        "role": "assistant",
        "content": json.dumps({
            "chapter": "始计篇",
            "quote": "兵者，诡道也。",
            "unsure": False,
            "insight": "绕开硬催，设局顺势收回。",
            "action": "就说自己在清账、要安排资金，先把一部分或者分期要回来，既能把钱收回，又不至于把气氛搞得像开战。",
            "speech": "我这边在清账需要安排下，能否先转一部分？分期也行，别影响咱们交情。",
            "explain_cn": "打仗要讲策略，不是直来直去，要懂得绕弯子、用计谋。"
        }, ensure_ascii=False)
    },
    # 邻居噪音 → 行军篇
    {
        "role": "user",
        "content": "问题：邻居太吵，怎么友好沟通？只输出JSON。"
    },
    {
        "role": "assistant",
        "content": json.dumps({
            "chapter": "行军篇",
            "quote": "知彼知己，百战不殆。（通用于沟通前的情势判断）",
            "unsure": True,  # 行军篇原文无此句，标不确定，演示 unsure 机制
            "insight": "先辨来源时机，再择机沟通。",
            "action": "挑对方不忙不生气的时候，先表示理解，再轻轻说出自己的困扰，让TA自己不好意思再把家当KTV用。",
            "speech": "最近晚上声音有点明显，我怕打扰到楼上楼下。能否帮忙晚间稍微注意下？谢谢啦～",
            "explain_cn": "先搞清楚对方情况和自己底牌，这样做事比较稳，不容易吃亏。"
        }, ensure_ascii=False)
    },
    # 职场竞争 → 军争篇
    {
        "role": "user",
        "content": "问题：同事跟我抢功劳，怎么应对？只输出JSON。"
    },
    {
        "role": "assistant",
        "content": json.dumps({
            "chapter": "军争篇",
            "quote": "以迂为直，以患为利。",
            "unsure": False,
            "insight": "不硬碰，转势为己用。",
            "action": "开会前先发一份“小而精”的纪要，把自己负责的部分写清楚，让功劳自动写进流程里，对方想抢也得先面对记录。",
            "speech": "我把这次我的负责点和关键产出提前发一份纪要，方便会里对齐，避免遗漏。",
            "explain_cn": "看起来绕远路，其实是走捷径；把麻烦事变成自己有利的机会。"
        }, ensure_ascii=False)
    }
]

# ===================== 调用 DeepSeek：主分析 =====================
def call_deepseek(user_question: str) -> dict:
    """自动尝试两个端点，返回结构化兵法解析结果。"""
    key = os.getenv("DEEPSEEK_API_KEY", "").strip()
    if not key:
        return {
            "chapter": "",
            "quote": "未配置 DEEPSEEK_API_KEY",
            "unsure": True,
            "insight": "请先在 Settings → Secrets 配置密钥。",
            "action": "键名必须是 DEEPSEEK_API_KEY，值不要带引号或空格；配置后重启。",
            "speech": "密钥配好我就能正常回答。",
            "explain_cn": "当前还无法解释名言，因为没有配置密钥。"
        }

    endpoints = [
        (os.getenv("DEEPSEEK_BASE_URL", "").strip()
         or "https://api.deepseek.com/v1/chat/completions"),
        "https://api.deepseek.com/chat/completions",
    ]

    # 多样化提示：鼓励换篇章/换角度
    diversity_hint = random.choice([
        "尽量避免重复使用同一篇章，尝试换篇章回答。",
        "优先考虑《军形篇》《兵势篇》《九变篇》《地形篇》的角度。",
        "如果问题可多解，换一个不同篇章来解释。",
        "引用少见但贴合的篇章，保持回答多样性。",
        "可从节奏、态势、资源、时机等不同维度选篇章。"
    ])

    msgs = [{"role": "system", "content": SYS_PROMPT + "\n" + diversity_hint}] + FEWSHOT + [
        {"role": "user", "content": f"问题：{user_question} 只输出JSON。"}
    ]

    last_err = None
    for ep in endpoints:
        try:
            payload = {
                "model": MODEL_NAME,
                "messages": msgs,
                "temperature": 0.7,
                "top_p": 0.9,
                "max_tokens": 450
            }
            r = requests.post(
                ep,
                headers={"Authorization": f"Bearer {key}", "Content-Type": "application/json"},
                json=payload,
                timeout=60
            )
            print(f"[DeepSeek] endpoint={ep} status={r.status_code} model={MODEL_NAME}")  # Runtime logs

            if r.status_code // 100 != 2:
                try:
                    err_detail = r.json()
                except Exception:
                    err_detail = {"raw": r.text[:500]}
                last_err = f"HTTP {r.status_code} @ {ep} (model={MODEL_NAME}) → {err_detail}"
                continue

            data = r.json()
            content = data.get("choices", [{}])[0].get("message", {}).get("content", "").strip()
            m = re.search(r"\{.*\}", content, flags=re.S)
            js = m.group(0) if m else content
            result = json.loads(js)

            return {
                "chapter": result.get("chapter", "")[:20],
                "quote": result.get("quote", "")[:200],
                "unsure": bool(result.get("unsure", False)),
                "insight": result.get("insight", "")[:40],
                "action": result.get("action", "")[:120],
                "speech": result.get("speech", "")[:140],
                "explain_cn": result.get("explain_cn", "")[:80],
            }
        except Exception as e:
            last_err = f"{type(e).__name__}: {e}"

    return {
        "chapter": "",
        "quote": "为避免误引，已隐藏原文与篇名（模型不确定出典）。",
        "unsure": True,
        "insight": "DeepSeek 请求未成功。",
        "action": f"诊断信息：{last_err or '未知错误'}",
        "speech": "请按诊断检查：①密钥②端点③模型名④配额/限速，然后重启。",
        "explain_cn": "当前无法给出名言解释，请检查配置。"
    }

# ===================== 调用 DeepSeek：中译英翻译 =====================
def translate_to_en(text_cn: str) -> str:
    """使用 DeepSeek 将简单中文解释翻译成英文。"""
    key = os.getenv("DEEPSEEK_API_KEY", "").strip()
    if not key:
        return "⚠️ 未配置 DEEPSEEK_API_KEY，暂时无法翻译成英文。"

    endpoints = [
        (os.getenv("DEEPSEEK_BASE_URL", "").strip()
         or "https://api.deepseek.com/v1/chat/completions"),
        "https://api.deepseek.com/chat/completions",
    ]

    msgs = [
        {
            "role": "system",
            "content": "你是专业的中译英助手，只把用户提供的中文翻译成自然简洁的英文，不要添加解释或多余内容。"
        },
        {
            "role": "user",
            "content": text_cn
        }
    ]

    last_err = None
    for ep in endpoints:
        try:
            payload = {
                "model": MODEL_NAME,
                "messages": msgs,
                "temperature": 0.3,
                "top_p": 0.9,
                "max_tokens": 200
            }
            r = requests.post(
                ep,
                headers={"Authorization": f"Bearer {key}", "Content-Type": "application/json"},
                json=payload,
                timeout=60
            )
            print(f"[DeepSeek-Translate] endpoint={ep} status={r.status_code} model={MODEL_NAME}")

            if r.status_code // 100 != 2:
                try:
                    err_detail = r.json()
                except Exception:
                    err_detail = {"raw": r.text[:500]}
                last_err = f"HTTP {r.status_code} @ {ep} (model={MODEL_NAME}) → {err_detail}"
                continue

            data = r.json()
            content = data.get("choices", [{}])[0].get("message", {}).get("content", "").strip()
            return html.escape(content)
        except Exception as e:
            last_err = f"{type(e).__name__}: {e}"

    return f"⚠️ 翻译失败：{last_err or '未知错误'}"

# ===================== 渲染函数 =====================
def render_main(result: dict) -> str:
    """
    主体渲染（不包含话术示例、不包含“我是老外”解释）。
    默认严格校验：若 unsure=True 则隐藏原文与篇名。
    """
    unsure = bool(result.get("unsure"))
    chap = html.escape(result.get("chapter", ""))
    quote = html.escape(result.get("quote", ""))
    insight = html.escape(result.get("insight", ""))
    action = html.escape(result.get("action", ""))

    # 默认“严格出处校验”：模型不确定时，隐藏原文与篇名
    show_quote = not unsure

    parts = ["<div style='font-size:22px;line-height:1.7em;'>"]
    if show_quote:
        badge = "📜" if not unsure else "❓"
        chap_line = f"《{chap}》" if chap else "（未标明篇名）"
        parts.append(f"{badge} {chap_line}<br><br>")
        parts.append(f"「{quote}」<br><br>")
    else:
        parts.append("⚠️ 为避免误引，已隐藏原文与篇名（模型不确定出典）。<br><br>")

    parts.append(f"<b>启示：</b>{insight}<br><br>")
    parts.append(f"<b>做法：</b>{action}<br><br>")
    parts.append("</div>")
    return "".join(parts)


def render_speech(result: dict) -> str:
    """单独渲染话术示例区域。"""
    speech = html.escape(result.get("speech", "").strip())
    if not speech:
        return (
            "<div style='margin-top:8px;font-size:16px;color:#777;'>"
            "当前没有可用的「话术示例」，请先点击「生成兵法」。"
            "</div>"
        )
    return (
        "<div style='margin-top:12px;padding:12px 14px;border:1px dashed #e2d7f2;"
        "border-radius:10px;background:#fbf8ff;font-size:18px;line-height:1.7em;'>"
        "<div style='font-weight:600;margin-bottom:6px;'>💬 话术示例：</div>"
        f"<div>{speech}</div>"
        "</div>"
    )


def render_explain_cn(result: dict) -> str:
    """为“我是老外”按钮渲染特别简单的中文解释。"""
    explain_cn = html.escape(result.get("explain_cn", "").strip())
    if not explain_cn:
        return (
            "<div style='margin-top:8px;font-size:16px;color:#777;'>"
            "暂时没有找到这句名言的简单解释，请先生成一次兵法再试。"
            "</div>"
        )
    return (
        "<div style='margin-top:12px;padding:12px 14px;border:1px solid #d7e6ff;"
        "border-radius:10px;background:#f5f8ff;font-size:18px;line-height:1.7em;'>"
        "<div style='font-weight:600;margin-bottom:6px;'>🌏 给外国朋友解释：</div>"
        f"<div>{explain_cn}</div>"
        "</div>"
    )


def render_explain_en(text_en: str) -> str:
    """渲染英文翻译结果。"""
    return (
        "<div style='margin-top:8px;padding:12px 14px;border:1px dashed #cddfea;"
        "border-radius:10px;background:#f6f9ff;font-size:18px;line-height:1.7em;'>"
        "<div style='font-weight:600;margin-bottom:6px;'>🔁 英文解释：</div>"
        f"<div>{text_en}</div>"
        "</div>"
    )

# ===================== UI =====================
theme = gr.themes.Soft(primary_hue="pink", neutral_hue="slate")
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;}
"""

with gr.Blocks(title=APP_TITLE, theme=theme, css=CSS) as demo:
    gr.HTML("<div class='header'>孙子你好</div>")
    with gr.Group(elem_classes=["card"]):
        q = gr.Textbox(
            label="你的问题",
            placeholder="例如：朋友借钱不还；老板让我加班；邻居太吵；甲方拖款…",
            lines=3
        )

        # 保存最近一次 DeepSeek 返回的结构化结果，用于“话术示例”和“我是老外”
        state_result = gr.State({})

        # 首页：只显示两个按钮
        with gr.Row():
            b1 = gr.Button("🎯 生成兵法", variant="primary")
            b3 = gr.Button("🧹 清空")

        # 下面这些按钮一开始不显示，生成兵法后再出现
        with gr.Row():
            btn_speech = gr.Button("💬 话术示例", visible=False)
            btn_foreign = gr.Button("🌏 我是老外", visible=False)
            btn_translate = gr.Button("🔁 翻译成英文", visible=False)

        out_main = gr.HTML()
        out_speech = gr.HTML()
        out_explain_cn = gr.HTML()
        out_explain_en = gr.HTML()

        # ---------- 回调函数 ----------
        def on_gen(user_q):
            # 返回顺序：
            # out_main, out_speech, out_explain_cn, out_explain_en, state_result,
            # btn_speech_update, btn_foreign_update, btn_translate_update
            if not (user_q or "").strip():
                empty_msg = (
                    "<div style='font-size:16px;color:#777;'>"
                    "请先在上面输入你的问题，比如：朋友借钱不还怎么办？</div>"
                )
                hide = gr.update(visible=False)
                return empty_msg, "", "", "", {}, hide, hide, hide

            res = call_deepseek(user_q)
            main_html = render_main(res)
            # 生成新结果时，显示「话术示例」「我是老外」，但先不显示「翻译成英文」
            return (
                main_html,
                "",
                "",
                "",
                res,
                gr.update(visible=True),   # btn_speech
                gr.update(visible=True),   # btn_foreign
                gr.update(visible=False),  # btn_translate
            )

        def on_clear():
            # 清空输入框和所有输出 & 状态，并隐藏所有二级按钮
            hide = gr.update(visible=False)
            return "", "", "", "", "", {}, hide, hide, hide

        def on_show_speech(res):
            if not isinstance(res, dict) or not res:
                return (
                    "<div style='margin-top:8px;font-size:16px;color:#777;'>"
                    "请先点击「生成兵法」，再查看话术示例。</div>"
                )
            return render_speech(res)

        def on_show_foreign(res):
            # 返回：out_explain_cn, btn_translate_update
            if not isinstance(res, dict) or not res:
                msg = (
                    "<div style='margin-top:8px;font-size:16px;color:#777;'>"
                    "请先点击「生成兵法」，再点「我是老外」。</div>"
                )
                return msg, gr.update(visible=False)
            return render_explain_cn(res), gr.update(visible=True)

        def on_translate(res):
            if not isinstance(res, dict) or not res:
                return (
                    "<div style='margin-top:8px;font-size:16px;color:#777;'>"
                    "请先生成兵法并点击「我是老外」后，再翻译成英文。</div>"
                )
            text_cn = res.get("explain_cn", "").strip()
            if not text_cn:
                return (
                    "<div style='margin-top:8px;font-size:16px;color:#777;'>"
                    "当前没有可以翻译的中文解释，请先点「我是老外」。</div>"
                )
            text_en = translate_to_en(text_cn)
            return render_explain_en(text_en)

        # ---------- 事件绑定 ----------
        b1.click(
            on_gen,
            inputs=[q],
            outputs=[
                out_main,
                out_speech,
                out_explain_cn,
                out_explain_en,
                state_result,
                btn_speech,
                btn_foreign,
                btn_translate,
            ]
        )

        b3.click(
            on_clear,
            inputs=None,
            outputs=[
                q,
                out_main,
                out_speech,
                out_explain_cn,
                out_explain_en,
                state_result,
                btn_speech,
                btn_foreign,
                btn_translate,
            ]
        )

        btn_speech.click(
            on_show_speech,
            inputs=[state_result],
            outputs=[out_speech]
        )

        btn_foreign.click(
            on_show_foreign,
            inputs=[state_result],
            outputs=[out_explain_cn, btn_translate]
        )

        btn_translate.click(
            on_translate,
            inputs=[state_result],
            outputs=[out_explain_en]
        )

if __name__ == "__main__":
    demo.launch()