198 lines
9.6 KiB
Python
198 lines
9.6 KiB
Python
import os, sys, subprocess
|
||
from urllib.request import urlopen
|
||
from urllib.parse import quote
|
||
if sys.stdout is None: sys.stdout = open(os.devnull, "w")
|
||
if sys.stderr is None: sys.stderr = open(os.devnull, "w")
|
||
try: sys.stdout.reconfigure(errors='replace')
|
||
except: pass
|
||
try: sys.stderr.reconfigure(errors='replace')
|
||
except: pass
|
||
script_dir = os.path.dirname(__file__)
|
||
sys.path.append(os.path.abspath(os.path.join(script_dir, '..')))
|
||
|
||
import streamlit as st
|
||
import time, json, re, threading, queue
|
||
from agentmain import GeneraticAgent
|
||
|
||
st.set_page_config(page_title="Cowork", layout="wide")
|
||
|
||
@st.cache_resource
|
||
def init():
|
||
agent = GeneraticAgent()
|
||
if agent.llmclient is None:
|
||
st.error("⚠️ 未配置任何可用的 LLM 接口，请设置mykey.py。")
|
||
st.stop()
|
||
else: threading.Thread(target=agent.run, daemon=True).start()
|
||
return agent
|
||
|
||
agent = init()
|
||
|
||
st.title("🖥️ Cowork")
|
||
|
||
if 'autonomous_enabled' not in st.session_state: st.session_state.autonomous_enabled = False
|
||
|
||
@st.fragment
|
||
def render_sidebar():
|
||
current_idx = agent.llm_no
|
||
st.caption(f"LLM Core: {current_idx}: {agent.get_llm_name()}", help="点击切换备用链路")
|
||
last_reply_time = st.session_state.get('last_reply_time', 0)
|
||
if last_reply_time > 0:
|
||
st.caption(f"空闲时间：{int(time.time()) - last_reply_time}秒", help="当超过30分钟未收到回复时，系统会自动任务")
|
||
if st.button("切换备用链路"):
|
||
agent.next_llm(); st.rerun(scope="fragment")
|
||
if st.button("强行停止任务"):
|
||
agent.abort(); st.toast("已发送停止信号"); st.rerun()
|
||
if st.button("重新注入工具"):
|
||
agent.llmclient.last_tools = ''
|
||
try:
|
||
hist_path = os.path.join(script_dir, '..', 'assets', 'tool_usable_history.json')
|
||
with open(hist_path, 'r', encoding='utf-8') as f: tool_hist = json.load(f)
|
||
agent.llmclient.backend.history.extend(tool_hist)
|
||
st.toast(f"已重新注入工具，追加了 {len(tool_hist)} 条示范记录")
|
||
except Exception as e: st.toast(f"注入工具示范失败: {e}")
|
||
if st.button("🐱 桌面宠物"):
|
||
kwargs = {'creationflags': 0x08} if sys.platform == 'win32' else {}
|
||
pet_script = os.path.join(script_dir, 'desktop_pet_v2.pyw')
|
||
if not os.path.exists(pet_script): pet_script = os.path.join(script_dir, 'desktop_pet.pyw')
|
||
subprocess.Popen([sys.executable, pet_script], **kwargs)
|
||
def _pet_req(q):
|
||
def _do():
|
||
try: urlopen(f'http://127.0.0.1:41983/?{q}', timeout=2)
|
||
except Exception: pass
|
||
threading.Thread(target=_do, daemon=True).start()
|
||
agent._pet_req = _pet_req
|
||
if not hasattr(agent, '_turn_end_hooks'): agent._turn_end_hooks = {}
|
||
def _pet_hook(ctx):
|
||
parts = [f"Turn {ctx.get('turn','?')}"]
|
||
if ctx.get('summary'): parts.append(ctx['summary'])
|
||
if ctx.get('exit_reason'): parts.append('任务已完成')
|
||
_pet_req(f'msg={quote(chr(10).join(parts))}')
|
||
if ctx.get('exit_reason'): _pet_req('state=idle')
|
||
agent._turn_end_hooks['pet'] = _pet_hook
|
||
st.toast("桌面宠物已启动")
|
||
|
||
st.divider()
|
||
if st.button("开始空闲自主行动"):
|
||
st.session_state.last_reply_time = int(time.time()) - 1800
|
||
st.toast("已将上次回复时间设为1800秒前"); st.rerun()
|
||
if st.session_state.autonomous_enabled:
|
||
if st.button("⏸️ 禁止自主行动"):
|
||
st.session_state.autonomous_enabled = False
|
||
st.toast("⏸️ 已禁止自主行动"); st.rerun()
|
||
st.caption("🟢 自主行动运行中，会在你离开它30分钟后自动进行")
|
||
else:
|
||
if st.button("▶️ 允许自主行动", type="primary"):
|
||
st.session_state.autonomous_enabled = True
|
||
st.toast("✅ 已允许自主行动"); st.rerun()
|
||
st.caption("🔴 自主行动已停止")
|
||
with st.sidebar: render_sidebar()
|
||
|
||
def fold_turns(text):
|
||
"""Return list of segments: [{'type':'text','content':...}, {'type':'fold','title':...,'content':...}]"""
|
||
parts = re.split(r'(\**LLM Running \(Turn \d+\) \.\.\.\*\**)', text)
|
||
if len(parts) < 4: return [{'type': 'text', 'content': text}]
|
||
segments = []
|
||
if parts[0].strip(): segments.append({'type': 'text', 'content': parts[0]})
|
||
turns = []
|
||
for i in range(1, len(parts), 2):
|
||
marker = parts[i]
|
||
content = parts[i+1] if i+1 < len(parts) else ''
|
||
turns.append((marker, content))
|
||
for idx, (marker, content) in enumerate(turns):
|
||
if idx < len(turns) - 1:
|
||
_c = re.sub(r'```.*?```|<thinking>.*?</thinking>', '', content, flags=re.DOTALL)
|
||
matches = re.findall(r'<summary>\s*((?:(?!<summary>).)*?)\s*</summary>', _c, re.DOTALL)
|
||
if matches:
|
||
title = matches[0].strip()
|
||
title = title.split('\n')[0]
|
||
if len(title) > 50: title = title[:50] + '...'
|
||
else: title = marker.strip('*')
|
||
segments.append({'type': 'fold', 'title': title, 'content': content})
|
||
else: segments.append({'type': 'text', 'content': marker + content})
|
||
return segments
|
||
def render_segments(segments, suffix=''):
|
||
# 整块重画：调用方用 slot.container() 包裹，保证 DOM 路径稳定、跨 rerun 对齐（消除"灰色重影"）。
|
||
# heartbeat 空转时 segments 不变 → Streamlit 后端 diff 无变化 → 前端零闪烁；
|
||
# 但 container/markdown 本身是 API 调用，StopException 仍会被抛出（abort 照常起作用）。
|
||
for seg in segments:
|
||
if seg['type'] == 'fold':
|
||
with st.expander(seg['title'], expanded=False): st.markdown(seg['content'])
|
||
else:
|
||
st.markdown(seg['content'] + suffix)
|
||
|
||
def agent_backend_stream(prompt):
|
||
display_queue = agent.put_task(prompt, source="user")
|
||
response = ''
|
||
try:
|
||
while True:
|
||
try: item = display_queue.get(timeout=1)
|
||
except queue.Empty:
|
||
yield response # heartbeat: let outer st.markdown() run → Streamlit checks StopException
|
||
continue
|
||
if 'next' in item:
|
||
response = item['next']; yield response
|
||
if 'done' in item:
|
||
yield item['done']; break
|
||
finally: agent.abort()
|
||
|
||
if "messages" not in st.session_state: st.session_state.messages = []
|
||
for msg in st.session_state.messages:
|
||
with st.chat_message(msg["role"]):
|
||
# 用 slot=st.empty() + with slot.container(): ... 的外壳，DOM 路径和流式渲染完全一致，跨 rerun 对齐
|
||
slot = st.empty()
|
||
with slot.container():
|
||
if msg["role"] == "assistant": render_segments(fold_turns(msg["content"]))
|
||
else: st.markdown(msg["content"])
|
||
|
||
# Scroll-height ghost fix: during streaming, expander open/close mid-animation can leave
|
||
# phantom height → scrollbar long but can't scroll to bottom. Periodically detect & reflow.
|
||
try:
|
||
from streamlit import iframe as _st_iframe # 1.56+
|
||
_embed_html = lambda html, **kw: _st_iframe(html, **{k: max(v, 1) if isinstance(v, int) else v for k, v in kw.items()})
|
||
except (ImportError, AttributeError):
|
||
from streamlit.components.v1 import html as _embed_html # ≤1.55
|
||
_js_scroll_fix = ("!function(){var p=window.parent;if(p.__sfx)return;p.__sfx=1;"
|
||
"var d=p.document;setInterval(function(){"
|
||
"var m=d.querySelector('section.main');if(!m)return;"
|
||
"var b=m.querySelector('.block-container');if(!b)return;"
|
||
"if(m.scrollHeight>b.scrollHeight+150){"
|
||
"m.style.overflow='hidden';void m.offsetHeight;m.style.overflow=''}"
|
||
"},3000)}()")
|
||
# IME composition fix (macOS only) - prevents Enter from submitting during CJK input
|
||
_js_ime_fix = ("" if os.name == 'nt' else
|
||
"!function(){if(window.parent.__imeFix)return;window.parent.__imeFix=1;"
|
||
"var d=window.parent.document,c=0;"
|
||
"d.addEventListener('compositionstart',()=>c=1,!0);"
|
||
"d.addEventListener('compositionend',()=>c=0,!0);"
|
||
"function f(){d.querySelectorAll('textarea[data-testid=stChatInputTextArea]')"
|
||
".forEach(t=>{t.__imeFix||(t.__imeFix=1,t.addEventListener('keydown',e=>{"
|
||
"e.key==='Enter'&&!e.shiftKey&&(e.isComposing||c||e.keyCode===229)&&"
|
||
"(e.stopImmediatePropagation(),e.preventDefault())},!0))})}"
|
||
"f();new MutationObserver(f).observe(d.body,{childList:1,subtree:1})}()")
|
||
_embed_html(f'<script>{_js_scroll_fix};{_js_ime_fix}</script>', height=0)
|
||
|
||
if prompt := st.chat_input("any task?"):
|
||
st.session_state.messages.append({"role": "user", "content": prompt})
|
||
if hasattr(agent, '_pet_req') and not prompt.startswith('/'): agent._pet_req('state=walk')
|
||
with st.chat_message("user"): st.markdown(prompt)
|
||
|
||
with st.chat_message("assistant"):
|
||
frozen = 0; live = st.empty(); response = ''
|
||
CURSOR = ' ▌'
|
||
for response in agent_backend_stream(prompt):
|
||
segs = fold_turns(response)
|
||
n_done = max(0, len(segs) - 1)
|
||
while frozen < n_done:
|
||
with live.container(): render_segments([segs[frozen]])
|
||
live = st.empty(); frozen += 1
|
||
with live.container(): render_segments([segs[-1]], suffix=CURSOR) # live 区域
|
||
segs = fold_turns(response)
|
||
for i in range(frozen, len(segs)):
|
||
with live.container(): render_segments([segs[i]])
|
||
if i < len(segs) - 1: live = st.empty()
|
||
st.session_state.messages.append({"role": "assistant", "content": response})
|
||
st.session_state.last_reply_time = int(time.time())
|
||
|
||
if st.session_state.autonomous_enabled:
|
||
st.markdown(f"""<div id="last-reply-time" style="display:none">{st.session_state.get('last_reply_time', int(time.time()))}</div>""", unsafe_allow_html=True)
|