Upload app.py

app.py

@@ -1032,14 +1032,21 @@ def _check_and_sync():
 _check_and_sync()
 
 # --- 登录用户自动模式切换：根据知识库内容决定默认模式 ---
-if not IS_GUEST and "sel_web" not in st.session_state:
-    # 首次加载时，检查私有库是否有内容
+def _has_any_kb_content():
+    """检查是否有任何知识库内容（公共库或私有库）"""
+    pub_count = st.session_state.get("_sync_count_public")
+    if pub_count is None:
+        pub_count = _count_chunks(PUBLIC_SCOPE)
+        st.session_state["_sync_count_public"] = pub_count
     priv_count = st.session_state.get("_sync_count_private")
     if priv_count is None:
         priv_count = _count_chunks(PRIVATE_SCOPE)
         st.session_state["_sync_count_private"] = priv_count
-    # 有内容则默认知识库模式，否则联网模式
-    st.session_state["sel_web"] = (priv_count == 0)
+    return (pub_count + priv_count) > 0
+
+if not IS_GUEST and "sel_web" not in st.session_state:
+    # 首次加载时，有任何知识库内容则默认知识库模式，否则联网模式
+    st.session_state["sel_web"] = not _has_any_kb_content()
 
 
 # =========================
@@ -1375,6 +1382,9 @@ with st.sidebar:
                     _delete_uploaded_file_from_storage(PUBLIC_SCOPE, delete_target)
                     _list_uploaded_files_db.clear()
                     st.session_state["_sync_count_public"] = _count_chunks(PUBLIC_SCOPE)
+                    # 删除后如果所有知识库为空，自动切换到联网模式
+                    if not _has_any_kb_content():
+                        st.session_state["sel_web"] = True
                     st.toast(f"已删除 {delete_target}")
                     st.rerun()
                 
@@ -1403,6 +1413,8 @@ with st.sidebar:
                 # 检查是否有待处理的上传（通过 on_change 标记）
                 if pub_files and st.session_state.pop("_pending_upload_public", False):
                     process_upload(pub_files, "public", PUBLIC_SCOPE)
+                    # 上传文件后自动切换到知识库模式
+                    st.session_state["sel_web"] = False
 
                 if pub_chunk_count > 0 and len(pub_file_list) >= 2:
                     if st.button("🗑️ 清空公共库", use_container_width=True, type="secondary", key="clear_pub"):
@@ -1410,6 +1422,9 @@ with st.sidebar:
                         _clear_uploaded_files_storage(PUBLIC_SCOPE)
                         _list_uploaded_files_db.clear()
                         st.session_state["_sync_count_public"] = 0
+                        # 清空后如果所有知识库为空，自动切换到联网模式
+                        if not _has_any_kb_content():
+                            st.session_state["sel_web"] = True
                         st.toast("公共知识库已清空")
                         st.rerun()
             else:
@@ -1435,10 +1450,9 @@ with st.sidebar:
                     _delete_chunks_by_file(PRIVATE_SCOPE, delete_target)
                     _delete_uploaded_file_from_storage(PRIVATE_SCOPE, delete_target)
                     _list_uploaded_files_db.clear()
-                    new_count = _count_chunks(PRIVATE_SCOPE)
-                    st.session_state["_sync_count_private"] = new_count
-                    # 删除后如果私有库为空，自动切换到联网模式
-                    if new_count == 0:
+                    st.session_state["_sync_count_private"] = _count_chunks(PRIVATE_SCOPE)
+                    # 删除后如果所有知识库为空，自动切换到联网模式
+                    if not _has_any_kb_content():
                         st.session_state["sel_web"] = True
                     st.toast(f"已删除 {delete_target}")
                     st.rerun()
@@ -1472,8 +1486,9 @@ with st.sidebar:
                     _clear_all_chunks(PRIVATE_SCOPE)
                     _clear_uploaded_files_storage(PRIVATE_SCOPE)
                     st.session_state["_sync_count_private"] = 0
-                    # 清空后自动切换到联网模式
-                    st.session_state["sel_web"] = True
+                    # 清空后如果所有知识库为空，自动切换到联网模式
+                    if not _has_any_kb_content():
+                        st.session_state["sel_web"] = True
                     st.toast("私有知识库已清空")
                     st.rerun()