Update app.py

app.py

@@ -109,7 +109,7 @@ FEWSHOT = [
 ]
 
 # ===================== 调用 DeepSeek：主分析 =====================
-def call_deepseek(user_question: str, alt: bool = False) -> dict:
+def call_deepseek(user_question: str) -> dict:
     """自动尝试两个端点，返回结构化兵法解析结果。"""
     key = os.getenv("DEEPSEEK_API_KEY", "").strip()
     if not key:
@@ -148,7 +148,7 @@ def call_deepseek(user_question: str, alt: bool = False) -> dict:
             payload = {
                 "model": MODEL_NAME,
                 "messages": msgs,
-                "temperature": 0.9 if alt else 0.7,
+                "temperature": 0.7,
                 "top_p": 0.9,
                 "max_tokens": 450
             }
@@ -258,7 +258,7 @@ def translate_to_en(text_cn: str) -> str:
 def render_main(result: dict) -> str:
     """
     主体渲染（不包含话术示例、不包含“我是老外”解释）。
-    严格校验：若 unsure=True 则隐藏原文与篇名。
+    默认严格校验：若 unsure=True 则隐藏原文与篇名。
     """
     unsure = bool(result.get("unsure"))
     chap = html.escape(result.get("chapter", ""))
@@ -266,7 +266,7 @@ def render_main(result: dict) -> str:
     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;'>"]
@@ -350,15 +350,16 @@ with gr.Blocks(title=APP_TITLE, theme=theme, css=CSS) as demo:
         # 保存最近一次 DeepSeek 返回的结构化结果，用于“话术示例”和“我是老外”
         state_result = gr.State({})
 
+        # 首页：只显示两个按钮
         with gr.Row():
             b1 = gr.Button("🎯 生成兵法", variant="primary")
-            b2 = gr.Button("♻️ 换一种说法")
             b3 = gr.Button("🧹 清空")
 
+        # 下面这些按钮一开始不显示，生成兵法后再出现
         with gr.Row():
-            btn_speech = gr.Button("💬 话术示例")
-            btn_foreign = gr.Button("🌏 我是老外")
-            btn_translate = gr.Button("🔁 翻译成英文")
+            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()
@@ -367,31 +368,35 @@ with gr.Blocks(title=APP_TITLE, theme=theme, css=CSS) as demo:
 
         # ---------- 回调函数 ----------
         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>"
                 )
-                return empty_msg, "", "", "", {}
-            res = call_deepseek(user_q, alt=False)
-            main_html = render_main(res)
-            # 生成新结果时，清空话术示例和解释区域
-            return main_html, "", "", "", res
+                hide = gr.update(visible=False)
+                return empty_msg, "", "", "", {}, hide, hide, hide
 
-        def on_alt(user_q, cur_state):
-            if not (user_q or "").strip():
-                empty_msg = (
-                    "<div style='font-size:16px;color:#777;'>"
-                    "请先在上面输入你的问题，比如：朋友借钱不还怎么办？</div>"
-                )
-                return empty_msg, "", "", "", {}
-            res = call_deepseek(user_q, alt=True)
+            res = call_deepseek(user_q)
             main_html = render_main(res)
-            return main_html, "", "", "", 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():
-            # 清空输入框和所有输出 & 状态
-            return "", "", "", "", "", {}
+            # 清空输入框和所有输出 & 状态，并隐藏所有二级按钮
+            hide = gr.update(visible=False)
+            return "", "", "", "", "", {}, hide, hide, hide
 
         def on_show_speech(res):
             if not isinstance(res, dict) or not res:
@@ -402,12 +407,14 @@ with gr.Blocks(title=APP_TITLE, theme=theme, css=CSS) as demo:
             return render_speech(res)
 
         def on_show_foreign(res):
+            # 返回：out_explain_cn, btn_translate_update
             if not isinstance(res, dict) or not res:
-                return (
+                msg = (
                     "<div style='margin-top:8px;font-size:16px;color:#777;'>"
                     "请先点击「生成兵法」，再点「我是老外」。</div>"
                 )
-            return render_explain_cn(res)
+                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:
@@ -428,17 +435,32 @@ with gr.Blocks(title=APP_TITLE, theme=theme, css=CSS) as demo:
         b1.click(
             on_gen,
             inputs=[q],
-            outputs=[out_main, out_speech, out_explain_cn, out_explain_en, state_result]
-        )
-        b2.click(
-            on_alt,
-            inputs=[q, state_result],
-            outputs=[out_main, out_speech, out_explain_cn, out_explain_en, state_result]
+            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]
+            outputs=[
+                q,
+                out_main,
+                out_speech,
+                out_explain_cn,
+                out_explain_en,
+                state_result,
+                btn_speech,
+                btn_foreign,
+                btn_translate,
+            ]
         )
 
         btn_speech.click(
@@ -450,7 +472,7 @@ with gr.Blocks(title=APP_TITLE, theme=theme, css=CSS) as demo:
         btn_foreign.click(
             on_show_foreign,
             inputs=[state_result],
-            outputs=[out_explain_cn]
+            outputs=[out_explain_cn, btn_translate]
         )
 
         btn_translate.click(