Upload app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import streamlit.components.v1 as components
 import json
 import time
 import logging
@@ -115,7 +116,7 @@ def inject_custom_css():
             .block-container {
                 position: fixed !important;
                 top: 7rem !important;
-                bottom: 5rem !important;
+                bottom: 7rem !important;
                 left: 0 !important;
                 right: 0 !important;
                 overflow-y: auto !important;
@@ -173,11 +174,6 @@ def inject_custom_css():
                 margin-bottom: 0.5rem !important;
             }
             
-            /* 主内容区域底部留出空间 */
-            .block-container {
-                bottom: 7rem !important;
-            }
-            
             /* 确保聊天容器可滚动 */
             [data-testid="stVerticalBlockBorderWrapper"] {
                 overflow-y: auto !important;
@@ -256,14 +252,16 @@ def inject_custom_css():
 
 inject_custom_css()
 
+# 全局游客标识（session_state 在单次渲染中不变）
+IS_GUEST = not bool(st.session_state.get("current_user"))
+
 # 标题（固定在顶部）- 游客始终显示提示
-_is_guest_init = not st.session_state.get("current_user")
 _guest_tip = """<div style="text-align:center; padding:4px 0; color:#666; font-size:13px;">
     🙋 当前为<b>游客模式</b>，可直接提问体验
-</div>""" if _is_guest_init else ""
+</div>""" if IS_GUEST else ""
 
 # 根据是否游客添加不同 class，用于 CSS 控制间距
-_title_class = "main-title main-title-guest" if _is_guest_init else "main-title main-title-user"
+_title_class = "main-title main-title-guest" if IS_GUEST else "main-title main-title-user"
 
 st.markdown(
     f"""
@@ -349,7 +347,10 @@ def _ensure_admin():
         logger.error(f"创建管理员失败: {e}")
 
 
-_ensure_admin()
+# 仅首次运行时检查管理员（避免每次刷新都查询）
+if not st.session_state.get("_admin_ensured"):
+    _ensure_admin()
+    st.session_state["_admin_ensured"] = True
 
 
 def _save_user(username, password_hash, role="user"):
@@ -434,8 +435,6 @@ if "current_role" not in st.session_state:
 if "auth_mode" not in st.session_state:
     st.session_state.auth_mode = "login"
 
-IS_GUEST = not bool(st.session_state.get("current_user"))
-
 with st.sidebar:
     # 未登录时不用 expander，直接显示登录表单（避免手动收起后无法自动展开）
     if IS_GUEST:
@@ -515,7 +514,6 @@ else:
 # 登录后自动收起侧边栏（通过 JS 模拟点击收起按钮）
 if not IS_GUEST and not st.session_state.get("_sidebar_collapsed_once"):
     st.session_state["_sidebar_collapsed_once"] = True
-    import streamlit.components.v1 as components
     components.html(
         """
         <script>
@@ -744,9 +742,6 @@ def _delete_uploaded_file_from_storage(scope, filename):
 
 def _clear_uploaded_files_storage(scope):
     """清空某个 scope 的所有上传文件。"""
-    files = _list_uploaded_files_db(scope)
-    paths = [f["storage_path"] for f in files] if files else []
-    # 从 uploaded_files 表拿 storage_path
     try:
         resp = _sb().table("uploaded_files").select("storage_path").eq("scope", scope).execute()
         paths = [row["storage_path"] for row in resp.data]
@@ -1223,8 +1218,6 @@ with st.sidebar:
                 pub_cnt = _count_chunks(PUBLIC_SCOPE)
                 st.text(f"📚 公共库切片数: {pub_cnt}")
 
-                # 统计所有 scope
-                resp = _sb().rpc("", {}).execute() if False else None  # placeholder
                 # 简单统计各用户私有库
                 for uname, _ in user_list:
                     cnt = _count_chunks(uname)
@@ -1470,7 +1463,6 @@ def _chat_fragment():
                 response_container = st.empty()
                 
                 # 滚动JS放在assistant块内部，不影响问题和回答之间的间距
-                import streamlit.components.v1 as components
                 components.html(
                     """<script>
                     setTimeout(function(){
@@ -1518,7 +1510,7 @@ def _chat_fragment():
                         _save_chat_message(CURRENT_USER, "assistant", full_response, meta_info)
                 except Exception as e:
                     logger.error(f"模型调用异常: {e}")
-                    container.error(f"❌ 抱歉，连接模型时出错了: {str(e)}")
+                    response_container.error(f"❌ 抱歉，连接模型时出错了: {str(e)}")
 
 
 # =========================