Upload app.py

app.py

@@ -107,73 +107,34 @@ def inject_custom_css():
             }
             /* 减少标题上方空白 */
             .block-container { padding-top: 1.5rem !important; }
-
-            /* 固定标题区域 */
-            .main-title {
-                position: sticky !important;
-                top: 0 !important;
-                background: white !important;
-                z-index: 100 !important;
-                padding: 0.5rem 0 !important;
-            }
-
-            /* 固定搜索框在底部 */
-            [data-testid="stChatInput"] {
-                position: fixed !important;
-                bottom: 1rem !important;
-                left: 50% !important;
-                transform: translateX(-50%) !important;
-                width: 90% !important;
-                max-width: 800px !important;
-                z-index: 99 !important;
-                background: white !important;
-                padding: 0.5rem !important;
-                border-radius: 8px !important;
-            }
-
-            /* 给主内容区底部留出搜索框空间 */
-            .main .block-container {
-                padding-bottom: 5rem !important;
-            }
         </style>
     """, unsafe_allow_html=True)
 
 
 inject_custom_css()
-
-# =========================
-# 标题和欢迎页一起渲染（优化加载体验）
-# =========================
-# 标题区域 - 使用 main-title class 固定
 st.markdown(
     """
-    <div class="main-title">
-        <h1 style="white-space:nowrap; overflow:hidden; text-overflow:ellipsis; font-size:clamp(1.4rem, 5vw, 2.2rem); margin:0 0 0.5rem; text-align:center;">
-            🤖 智答 AI 助手
-        </h1>
-    </div>
+    <h1 style="white-space:nowrap; overflow:hidden; text-overflow:ellipsis; font-size:clamp(1.4rem, 5vw, 2.2rem); margin:0 0 0.5rem; text-align:center;">
+        🤖 智答 AI 助手
+    </h1>
     """,
     unsafe_allow_html=True,
 )
 
 # 欢迎页占位（登录后若有消息则清除）
 _welcome_hero = st.empty()
-
-# 立即渲染欢迎页（避免延迟显示）
-if not st.session_state.get("current_user") and not st.session_state.get("messages"):
+if not st.session_state.get("current_user"):
+    # 未登录：欢迎页（不含游客模式提示，避免闪现）
     _welcome_hero.markdown(
         """
-        <div style="display:flex; align-items:center; justify-content:center; min-height:45vh;">
+        <div style="display:flex; align-items:center; justify-content:center; min-height:50vh;">
             <div style="text-align:center; color:#888;">
-                <div style="font-size:48px; margin-bottom:12px;">🤖</div>
-                <div style="font-size:18px; font-weight:600; margin-bottom:8px;">欢迎使用智答 AI 助手</div>
-                <div style="font-size:14px; line-height:1.8;">
+                <div style="font-size:52px; margin-bottom:16px;">🤖</div>
+                <div style="font-size:20px; font-weight:600; margin-bottom:10px;">欢迎使用智答 AI 助手</div>
+                <div style="font-size:15px; line-height:2;">
                     🌐 <b>联网模式</b> —— 大模型 + 网络搜索，实时回答<br>
                     📚 <b>知识库模式</b> —— 上传文档，基于私有知识回答
                 </div>
-                <div style="margin-top:12px; font-size:13px; color:#999;">
-                    🙋 当前为<b>游客模式</b>，可直接在下方提问体验
-                </div>
             </div>
         </div>
         """,
@@ -374,7 +335,6 @@ with st.sidebar:
                         st.session_state.login_attempts = 0
                         st.session_state.current_user = input_username
                         st.session_state.current_role = role
-                        st.toast(f"✅ 登录成功，欢迎 {input_username}！")
                         st.rerun()
 
         else:  # 注册
@@ -392,7 +352,6 @@ with st.sidebar:
                         st.session_state.current_user = reg_user
                         st.session_state.current_role = "user"
                         st.session_state.login_attempts = 0
-                        st.toast(f"✅ 注册成功，欢迎 {reg_user}！")
                         st.rerun()
                     else:
                         st.error(f"❌ {msg}")
@@ -724,9 +683,10 @@ SYSTEM_PROMPT = (
 )
 
 SYSTEM_PROMPT_WEB = (
-    "你是一个专业的AI助手，请基于互联网搜索获取的实时信息回答用户问题。"
-    "回答要准确、有条理、简洁，并尽可能提供最新的信息。"
-    "如果搜索结果不足以回答问题，请基于你的知识库作答并说明。"
+    "你是一个智能AI助手，擅长结合互联网搜索结果回答用户问题。"
+    "请根据搜索结果提供准确、有条理的回答。"
+    "如果搜索结果不足以回答问题，请结合你自身的知识进行补充。"
+    "回答要简洁实用，注明信息来源。"
 )
 
 
@@ -958,7 +918,8 @@ else:
 
 
 with st.sidebar:
-    st.caption("👇 展开下方面板管理知识库和设置")
+    if not IS_GUEST:
+        st.caption("👇 展开下方面板管理知识库和设置")
     # --- 模型设置（游客和登录用户均可用） ---
     with st.expander("⚙️ 模型设置"):
         _model_list = list(_active_models.keys())
@@ -1165,13 +1126,10 @@ with st.sidebar:
             hist_tab_new, hist_tab_history = st.tabs(["当前对话", "历史记录"])
 
             with hist_tab_new:
-                st.caption("清空当前对话（页面显示）")
+                st.caption("清空当前对话（数据库记录保留）")
                 if st.button("🧹 清空当前对话", use_container_width=True, type="secondary", key="btn_clear_chat"):
                     st.session_state.messages = []
-                    st.toast("当前对话已清空")
-                    st.rerun()
 
-                st.divider()
                 st.caption("清空所有历史记录（不可恢复）")
                 if st.button("🗑️ 清空全部记录", use_container_width=True, type="secondary", key="btn_clear_all_hist"):
                     _async_run(_clear_chat_history_db, CURRENT_USER)
@@ -1222,14 +1180,19 @@ def llm_answer(query, context_docs, selected_display_name, web_enabled):
     all_context = ""
     curr_time = datetime.now().strftime("%Y-%m-%d %H:%M")
 
-    if context_docs:
-        all_context += "【知识库资料】：\n" + "\n".join(context_docs) + "\n"
-
     if web_enabled:
+        # 联网模式：只用网络搜索，不用知识库
         search_res = web_search(query)
-        all_context += f"\n【互联网实时资料】：\n{search_res}"
+        all_context = f"【互联网搜索结果】：\n{search_res}"
+        prompt_content = f"当前时间：{curr_time}\n\n{all_context[:6500]}\n\n用户问题：{query}"
+        system_prompt = SYSTEM_PROMPT_WEB
+    else:
+        # 知识库模式：用知识库资料
+        if context_docs:
+            all_context = "【知识库资料】：\n" + "\n".join(context_docs)
+        prompt_content = f"当前时间：{curr_time}\n\n参考资料：\n{all_context[:6500]}\n\n用户问题：{query}"
+        system_prompt = SYSTEM_PROMPT
 
-    prompt_content = f"当前时间：{curr_time}\n\n参考资料：\n{all_context[:6500]}\n\n用户问题：{query}"
     input_tokens = estimate_tokens(prompt_content)
 
     or_client = get_or_client()
@@ -1258,10 +1221,8 @@ def llm_answer(query, context_docs, selected_display_name, web_enabled):
         if selected_id != p_id:
             retry_queue.append((p_client, p_id, f"💰 收费兜底-{p_label}"))
 
-    # 联网模式和知识库模式使用不同的 system prompt
-    _system_prompt = SYSTEM_PROMPT_WEB if web_enabled else SYSTEM_PROMPT
     messages = [
-        {"role": "system", "content": _system_prompt},
+        {"role": "system", "content": system_prompt},
         {"role": "user", "content": prompt_content},
     ]
 
@@ -1323,9 +1284,25 @@ if not st.session_state.get("messages"):
     else:
         st.session_state.messages = []
 
-# 有消息时清除欢迎页
+# 登录后处理欢迎页：有消息则清除，无消息则显示（不含登录提示）
 if st.session_state.messages:
     _welcome_hero.empty()
+else:
+    _welcome_hero.markdown(
+        """
+        <div style="display:flex; align-items:center; justify-content:center; min-height:50vh;">
+            <div style="text-align:center; color:#888;">
+                <div style="font-size:52px; margin-bottom:16px;">🤖</div>
+                <div style="font-size:20px; font-weight:600; margin-bottom:10px;">欢迎使用智答 AI 助手</div>
+                <div style="font-size:15px; line-height:2;">
+                    🌐 <b>联网模式</b> —— 大模型 + 网络搜索，实时回答<br>
+                    📚 <b>知识库模式</b> —— 上传文档，基于私有知识回答
+                </div>
+            </div>
+        </div>
+        """,
+        unsafe_allow_html=True,
+    )
 
 
 @st.fragment
@@ -1336,48 +1313,24 @@ def _chat_fragment():
     _top_k = st.session_state.get("sel_topk", 5)
     _threshold = st.session_state.get("sel_threshold", 0.25)
 
-    # 聊天区域：消息在上（固定高度可滚动），输入框在下（固定）
+    # 消息容器（可滚动） + 输入框固定在下方
+    # 有消息时使用固定高度实现滚动；无消息时不设高度避免空白
     _has_msgs = bool(st.session_state.messages)
-
-    # 消息区域 - 固定高度，可滚动
-    if _has_msgs:
-        # 固定高度的消息容器（可滚动）
-        msg_container = st.container(height=480)
-        with msg_container:
-            for m in st.session_state.messages:
-                with st.chat_message(m["role"]):
-                    st.markdown(m["content"])
-                    if "meta" in m:
-                        st.caption(m["meta"])
-
-        # 自动滚动到底部
-        st.markdown("""
-            u003cscriptu003e
-                (function() {
-                    const containers = document.querySelectorAll('[data-testid="stVerticalBlock"]');
-                    containers.forEach(c => {
-                        if (c.scrollHeight u003e c.clientHeight) {
-                            c.scrollTop = c.scrollHeight;
-                        }
-                    });
-                })();
-            u003c/scriptu003e
-        """, unsafe_allow_html=True)
-    else:
-        # 无消息时不占高度
-        msg_container = st.container()
-
-    # 搜索框固定在底部
+    chat_box = st.container(height=520) if _has_msgs else st.container()
     q = st.chat_input("输入问题...", key="chat_input_v3")
 
-    if q:
-        _welcome_hero.empty()  # 首次发消息时清除欢迎页
-        st.session_state.messages.append({"role": "user", "content": q})
-        if not IS_GUEST:
-            _save_chat_message(CURRENT_USER, "user", q)
-
-        # 立即显示用户消息
-        with msg_container:
+    with chat_box:
+        for m in st.session_state.messages:
+            with st.chat_message(m["role"]):
+                st.markdown(m["content"])
+                if "meta" in m:
+                    st.caption(m["meta"])
+
+        if q:
+            _welcome_hero.empty()  # 首次发消息时清除欢迎页
+            st.session_state.messages.append({"role": "user", "content": q})
+            if not IS_GUEST:
+                _save_chat_message(CURRENT_USER, "user", q)
             with st.chat_message("user"):
                 st.markdown(q)
 
@@ -1385,6 +1338,7 @@ def _chat_fragment():
                 container = st.empty()
 
                 # 联网模式：大模型 + 网络搜索，不查知识库
+                # 知识库模式：RAG 检索本地知识库
                 if _web_on:
                     container.markdown("*🌐 正在联网搜索...*")
                     relevant_docs = []