Upload app.py

app.py

@@ -1613,31 +1613,33 @@ def _chat_fragment():
 
 
 # =========================
-# 11. 搜索框下方的工具按钮
+# 11. 搜索框下方的工具按钮（使用 fragment 局部刷新）
 # =========================
-_current_model = st.session_state.get("sel_model", list(model_mapping.keys())[0])
-_web_status = st.session_state.get("sel_web", True)
-
-_model_short = _current_model.split("(")[0].strip()
-
-# 两个按钮并排
-_btn_cols = st.columns(2)
-with _btn_cols[0]:
-    with st.popover(_model_short, use_container_width=True):
-        st.caption("选择模型")
-        _model_list = list(_active_models.keys())
-        for m in _model_list:
-            _btn_type = "primary" if m == _current_model else "secondary"
-            if st.button(m, key=f"mdl_{m}", use_container_width=True, type=_btn_type):
-                st.session_state["sel_model"] = m
-
-with _btn_cols[1]:
-    # 游客模式：联网/直接回答；登录模式：联网/知识库
-    if IS_GUEST:
-        _web_label = "🌐 联网模式" if _web_status else "💬 直接回答"
-    else:
-        _web_label = "🌐 联网模式" if _web_status else "📚 知识库"
-    if st.button(_web_label, key="toggle_web", use_container_width=True):
-        st.session_state["sel_web"] = not _web_status
+@st.fragment
+def _toolbar_fragment():
+    _current_model = st.session_state.get("sel_model", list(model_mapping.keys())[0])
+    _web_status = st.session_state.get("sel_web", True)
+    _model_short = _current_model.split("(")[0].strip()
+    
+    _btn_cols = st.columns(2)
+    with _btn_cols[0]:
+        with st.popover(_model_short, use_container_width=True):
+            st.caption("选择模型")
+            _model_list = list(_active_models.keys())
+            for m in _model_list:
+                _btn_type = "primary" if m == _current_model else "secondary"
+                if st.button(m, key=f"mdl_{m}", use_container_width=True, type=_btn_type):
+                    if m != _current_model:
+                        st.session_state["sel_model"] = m
+
+    with _btn_cols[1]:
+        if IS_GUEST:
+            _web_label = "🌐 联网模式" if _web_status else "💬 直接回答"
+        else:
+            _web_label = "🌐 联网模式" if _web_status else "📚 知识库"
+        if st.button(_web_label, key="toggle_web", use_container_width=True):
+            st.session_state["sel_web"] = not _web_status
+
+_toolbar_fragment()
 
 _chat_fragment()