Upload app.py

app.py

@@ -31,6 +31,8 @@ def inject_custom_css():
             [data-testid="stSidebarContent"] { padding-top: 1.5rem !important; }
             [data-testid="stVerticalBlock"] > div { gap: 0.8rem !important; }
             [data-testid="stSelectbox"] input { caret-color: transparent !important; }
+            /* 隐藏 text_input 的 press enter to apply 提示 */
+            [data-testid="stTextInput"] div[data-testid="InputInstructions"] { display: none !important; }
             [data-testid="stFileUploader"] section > div { display: none; }
             [data-testid="stFileUploaderDropzoneInstructions"] { display: none !important; }
             [data-testid="stFileUploader"] section::before {
@@ -54,7 +56,7 @@ def inject_custom_css():
             }
             /* 左上角侧边栏按钮：替换箭头为「☰ 菜单」 */
             [data-testid="stSidebarCollapsedControl"] {
-                top: 0.35rem !important;
+                top: 0.4rem !important;
                 left: 0.5rem !important;
             }
             [data-testid="stSidebarCollapsedControl"] button {
@@ -87,7 +89,7 @@ def inject_custom_css():
             /* 右上角三点菜单：替换为「⚙ 设置」按钮风格 */
             [data-testid="stMainMenu"] {
                 position: fixed !important;
-                top: 0.35rem !important;
+                top: 0.4rem !important;
                 right: 0.5rem !important;
             }
             [data-testid="stMainMenu"] button {
@@ -294,9 +296,12 @@ 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:
-    _is_logged_in = bool(st.session_state.get("current_user"))
-    with st.expander("🔑 账号", expanded=not _is_logged_in):
+    # 未登录时不用 expander，直接显示登录表单（避免手动收起后无法自动展开）
+    if IS_GUEST:
+        st.subheader("🔑 账号")
         if st.session_state.login_attempts >= MAX_LOGIN_ATTEMPTS:
             st.error("🚫 尝试次数过多，请刷新页面后重试。")
             st.stop()
@@ -318,28 +323,22 @@ with st.sidebar:
             if st.button("🔓 登录", use_container_width=True, type="primary", key="btn_login"):
                 if input_username == "" or input_password == "":
                     st.warning("请输入用户名和密码")
-                    st.stop()
-
-                ok, role, reason = verify_user(input_username, input_password)
-                if not ok:
-                    st.session_state.login_attempts += 1
-                    remaining = MAX_LOGIN_ATTEMPTS - st.session_state.login_attempts
-                    if reason == "not_found":
-                        st.warning(f"⚠️ 用户不存在，请先注册（剩余 {remaining} 次）")
-                    else:
-                        st.warning(f"⚠️ 密码错误（剩余 {remaining} 次）")
-                    st.stop()
+                elif st.session_state.login_attempts >= MAX_LOGIN_ATTEMPTS:
+                    pass
                 else:
-                    st.session_state.login_attempts = 0
-                    st.session_state.current_user = input_username
-                    st.session_state.current_role = role
-                    role_label = "管理员" if role == "admin" else "普通用户"
-                    st.success(f"✅ {input_username}（{role_label}）")
-                    st.rerun()
-            else:
-                # 未点击登录按钮时，检查是否已登录
-                if not st.session_state.get("current_user"):
-                    st.stop()
+                    ok, role, reason = verify_user(input_username, input_password)
+                    if not ok:
+                        st.session_state.login_attempts += 1
+                        remaining = MAX_LOGIN_ATTEMPTS - st.session_state.login_attempts
+                        if reason == "not_found":
+                            st.warning(f"⚠️ 用户不存在，请先注册（剩余 {remaining} 次）")
+                        else:
+                            st.warning(f"⚠️ 密码错误（剩余 {remaining} 次）")
+                    else:
+                        st.session_state.login_attempts = 0
+                        st.session_state.current_user = input_username
+                        st.session_state.current_role = role
+                        st.rerun()
 
         else:  # 注册
             reg_user = st.text_input("用户名", key="reg_user")
@@ -356,21 +355,30 @@ with st.sidebar:
                         st.session_state.current_user = reg_user
                         st.session_state.current_role = "user"
                         st.session_state.login_attempts = 0
-                        st.success(f"✅ 注册成功，已自动登录")
                         st.rerun()
                     else:
                         st.error(f"❌ {msg}")
-            st.stop()
-
-CURRENT_USER = st.session_state.current_user
-IS_ADMIN = st.session_state.current_role == "admin"
 
-# 登录成功：根据是否有历史消息决定欢迎页内容
-# （此时还未加载消息，先暂存 placeholder，加载后再决定）
-# _welcome_hero 会在聊天渲染区最终处理
+        st.divider()
+        st.caption("💡 未登录可直接提问（游客模式），登录后解锁全部功能")
+    else:
+        with st.expander("🔑 账号"):
+            role_label = "管理员" if st.session_state.current_role == "admin" else "普通用户"
+            st.success(f"✅ {st.session_state.current_user}（{role_label}）")
+
+# 设置全局用户标识
+if st.session_state.get("current_user"):
+    CURRENT_USER = st.session_state.current_user
+    IS_ADMIN = st.session_state.current_role == "admin"
+    IS_GUEST = False
+    _welcome_hero.empty()  # 登录成功，清除欢迎页
+else:
+    CURRENT_USER = "_guest_"
+    IS_ADMIN = False
+    IS_GUEST = True
 
 # 登录后自动收起侧边栏（通过 JS 模拟点击收起按钮）
-if not st.session_state.get("_sidebar_collapsed_once"):
+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(
@@ -891,6 +899,13 @@ model_mapping = {
     "🏢 百度文心 (官方)": "ernie-3.5-8k",
 }
 
+# 游客模式：屏蔽收费模型
+_GUEST_BLOCKED_MODELS = {"🛡️ DeepSeek (官方)", "🏢 百度文心 (官方)"}
+if IS_GUEST:
+    _active_models = {k: v for k, v in model_mapping.items() if k not in _GUEST_BLOCKED_MODELS}
+else:
+    _active_models = model_mapping
+
 
 with st.sidebar:
     st.caption("👇 展开下方面板管理知识库和设置")