Upload app.py

app.py

@@ -372,6 +372,63 @@ def inject_custom_css():
 
 inject_custom_css()
 
+
+def _render_action_buttons(content, msg_id):
+    """生成消息底部的复制/分享/播报按钮"""
+    # 转义内容用于 JavaScript
+    escaped = content.replace("\\", "\\\\").replace("`", "\\`").replace("$", "\\$")
+    return f'''
+    <div style="display:flex; gap:12px; margin-top:8px; padding-top:6px; border-top:1px solid #eee;">
+        <button onclick="copyText_{msg_id}()" style="background:none; border:none; cursor:pointer; color:#666; font-size:13px; padding:4px 8px; border-radius:4px;" onmouseover="this.style.background='#f0f0f0'" onmouseout="this.style.background='none'">📋 复制</button>
+        <button onclick="shareText_{msg_id}()" style="background:none; border:none; cursor:pointer; color:#666; font-size:13px; padding:4px 8px; border-radius:4px;" onmouseover="this.style.background='#f0f0f0'" onmouseout="this.style.background='none'">🔗 分享</button>
+        <button id="tts_btn_{msg_id}" onclick="speakText_{msg_id}()" style="background:none; border:none; cursor:pointer; color:#666; font-size:13px; padding:4px 8px; border-radius:4px;" onmouseover="this.style.background='#f0f0f0'" onmouseout="this.style.background='none'">🔊 播报</button>
+    </div>
+    <script>
+    const msgContent_{msg_id} = `{escaped}`;
+    let speaking_{msg_id} = false;
+    
+    function copyText_{msg_id}() {{
+        navigator.clipboard.writeText(msgContent_{msg_id}).then(() => {{
+            alert('已复制到剪贴板');
+        }}).catch(() => {{
+            alert('复制失败，请手动复制');
+        }});
+    }}
+    
+    function shareText_{msg_id}() {{
+        if (navigator.share) {{
+            navigator.share({{
+                title: '智答AI助手',
+                text: msgContent_{msg_id}
+            }}).catch(() => {{}});
+        }} else {{
+            navigator.clipboard.writeText(msgContent_{msg_id}).then(() => {{
+                alert('链接已复制，可粘贴分享');
+            }});
+        }}
+    }}
+    
+    function speakText_{msg_id}() {{
+        const btn = document.getElementById('tts_btn_{msg_id}');
+        if (speaking_{msg_id}) {{
+            window.speechSynthesis.cancel();
+            speaking_{msg_id} = false;
+            btn.textContent = '🔊 播报';
+        }} else {{
+            const utterance = new SpeechSynthesisUtterance(msgContent_{msg_id});
+            utterance.lang = 'zh-CN';
+            utterance.rate = 1.0;
+            utterance.onend = () => {{ speaking_{msg_id} = false; btn.textContent = '🔊 播报'; }};
+            utterance.onerror = () => {{ speaking_{msg_id} = false; btn.textContent = '🔊 播报'; }};
+            window.speechSynthesis.speak(utterance);
+            speaking_{msg_id} = true;
+            btn.textContent = '⏹️ 停止';
+        }}
+    }}
+    </script>
+    '''
+
+
 # 全局游客标识（session_state 在单次渲染中不变）
 IS_GUEST = not bool(st.session_state.get("current_user"))
 
@@ -1709,11 +1766,13 @@ def _chat_fragment():
         q = st.session_state.pop("_quick_question")
 
     with chat_box:
-        for m in st.session_state.messages:
+        for idx, m in enumerate(st.session_state.messages):
             with st.chat_message(m["role"]):
-                if m["role"] == "assistant" and m.get("meta"):
+                if m["role"] == "assistant":
+                    meta_html = f'\n\n<span style="color:#999;font-size:12px;">{m["meta"]}</span>' if m.get("meta") else ""
+                    action_btns = _render_action_buttons(m["content"], f"hist_{idx}")
                     st.markdown(
-                        m["content"] + f'\n\n<span style="color:#999;font-size:12px;">{m["meta"]}</span>',
+                        m["content"] + meta_html + action_btns,
                         unsafe_allow_html=True,
                     )
                 else:
@@ -1768,12 +1827,14 @@ def _chat_fragment():
                         llm_answer(q, relevant_docs, _model_name, _web_on, kb_mode=_kb_mode, source_files=source_files)
                     )
                     meta_info = st.session_state.get("last_meta", "")
-                    # meta 拼进 response_container，不创建独立DOM元素
-                    if meta_info:
-                        response_container.markdown(
-                            full_response + f'\n\n<span style="color:#999;font-size:12px;">{meta_info}</span>',
-                            unsafe_allow_html=True,
-                        )
+                    # 生成工具按钮（使用当前消息数作为ID）
+                    msg_idx = len(st.session_state.messages)
+                    action_btns = _render_action_buttons(full_response, f"new_{msg_idx}")
+                    meta_html = f'\n\n<span style="color:#999;font-size:12px;">{meta_info}</span>' if meta_info else ""
+                    response_container.markdown(
+                        full_response + meta_html + action_btns,
+                        unsafe_allow_html=True,
+                    )
                     st.session_state.messages.append(
                         {"role": "assistant", "content": full_response, "meta": meta_info}
                     )