Upload app.py

app.py

@@ -393,38 +393,6 @@ def inject_custom_css():
 
 inject_custom_css()
 
-# 移动端可见性变化检测 - 从后台切回时如果连接断开则自动刷新
-_visibility_js = """
-<script>
-(function() {
-    if (window._visibilityHandlerAdded) return;
-    window._visibilityHandlerAdded = true;
-    
-    let hiddenTime = 0;
-    document.addEventListener('visibilitychange', function() {
-        if (document.hidden) {
-            hiddenTime = Date.now();
-        } else {
-            // 从后台切回，检查是否超过10秒
-            const elapsed = Date.now() - hiddenTime;
-            if (elapsed > 10000) {
-                // 检查是否显示 Connecting 状态
-                setTimeout(function() {
-                    const statusEl = document.querySelector('[data-testid="stStatusWidget"]');
-                    const connectingEl = document.querySelector('.stConnectionStatus');
-                    if (statusEl || connectingEl) {
-                        console.log('检测到连接断开，自动刷新...');
-                        window.location.reload();
-                    }
-                }, 2000);
-            }
-        }
-    });
-})();
-</script>
-"""
-components.html(_visibility_js, height=0)
-
 
 def _render_action_buttons(content, msg_id):
     """生成消息底部的复制/分享/播报按钮（纯HTML部分）"""
@@ -1276,41 +1244,6 @@ def process_upload(uploaded_files, target_prefix, scope):
 # 6.5 聊天记录持久化（Supabase chat_history 表）
 # =========================
 import threading
-from queue import Queue, Empty
-
-
-def _stream_with_heartbeat(response_iter, heartbeat_interval=1.5):
-    """
-    包装流式响应迭代器，添加心跳机制防止WebSocket断开。
-    在后台线程读取API响应，主线程定时发送心跳（空字符串）。
-    heartbeat_interval: 心跳间隔（秒），移动端需要更短间隔
-    """
-    q = Queue()
-    done = object()  # 标记结束
-    
-    def reader():
-        try:
-            for chunk in response_iter:
-                q.put(chunk)
-        except Exception as e:
-            q.put(e)
-        finally:
-            q.put(done)
-    
-    t = threading.Thread(target=reader, daemon=True)
-    t.start()
-    
-    while True:
-        try:
-            item = q.get(timeout=heartbeat_interval)
-            if item is done:
-                break
-            if isinstance(item, Exception):
-                raise item
-            yield item
-        except Empty:
-            # 超时未收到数据，发送心跳保持连接
-            yield None
 
 
 def _async_run(fn, *args):
@@ -1784,12 +1717,9 @@ def llm_answer(query, context_docs, selected_display_name, web_enabled, kb_mode=
 
             full_text = ""
             has_content = False
-            # 使用心跳包装器防止WebSocket断开
-            for chunk in _stream_with_heartbeat(response):
-                if chunk is None:
-                    # 心跳信号，yield空字符串保持连接
-                    yield ""
-                elif chunk.choices and chunk.choices[0].delta.content:
+            # 直接迭代流式响应
+            for chunk in response:
+                if chunk.choices and chunk.choices[0].delta.content:
                     content = chunk.choices[0].delta.content
                     full_text += content
                     has_content = True
@@ -1852,12 +1782,9 @@ def llm_answer(query, context_docs, selected_display_name, web_enabled, kb_mode=
 
             full_text = ""
             has_content = False
-            # 使用心跳包装器防止WebSocket断开
-            for chunk in _stream_with_heartbeat(response):
-                if chunk is None:
-                    # 心跳信号，yield空字符串保持连接
-                    yield ""
-                elif chunk.choices and chunk.choices[0].delta.content:
+            # 直接迭代流式响应
+            for chunk in response:
+                if chunk.choices and chunk.choices[0].delta.content:
                     content = chunk.choices[0].delta.content
                     full_text += content
                     has_content = True