"""
桌面前端单文件版 – PySide6 聊天面板 + 悬浮按钮 thanks to GaoZhiCheng
依赖: pip install PySide6
可选: pip install markdown (Markdown 渲染)
用法: python frontends/qtapp.py
"""
from __future__ import annotations
import math, os, sys, json, glob, re, base64, time, threading
import queue as _queue
from datetime import datetime
from typing import Optional
from PySide6.QtWidgets import (
QWidget, QVBoxLayout, QHBoxLayout, QLabel, QPushButton,
QScrollArea, QFrame, QTextEdit, QStackedWidget,
QListWidget, QListWidgetItem, QSizePolicy, QFileDialog,
QSplitter, QTextBrowser, QApplication, QMessageBox,
QMenu, QLineEdit,
)
from PySide6.QtCore import (
Qt, QTimer, QPoint, QPointF, QByteArray, QSize,
Signal, QMetaObject, Q_ARG, QObject, QDateTime, QEvent,
)
from PySide6.QtGui import (
QPainter, QColor, QLinearGradient, QRadialGradient,
QPen, QPainterPath, QCursor, QFont, QIcon, QPixmap, QRegion,
)
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
from agentmain import GeneraticAgent
from chatapp_common import FILE_HINT, HELP_TEXT, clean_reply, build_done_text, format_restore
# ══════════════════════════════════════════════════════════════════════
# FloatingButton
# ══════════════════════════════════════════════════════════════════════
class FloatingButton(QWidget):
SIZE = 60 # circle diameter
MARGIN = 14 # extra space for glow
TOTAL = SIZE + MARGIN * 2
def __init__(self, chat_panel: QWidget):
super().__init__()
self.chat_panel = chat_panel
self._drag_origin_global: QPoint | None = None
self._drag_origin_win: QPoint | None = None
self._dragged = False
self._glow = 0.5
self._glow_dir = 1
self._hovering = False
self._hover_clock = 0.0
self._hover_strength = 0.0
self._flow_phase = 0.0
self._running = False
self._last_toggle_ms = 0 # debounce timestamp
# Window flags: frameless, always on top, no taskbar entry
self.setWindowFlags(
Qt.FramelessWindowHint
| Qt.WindowStaysOnTopHint
| Qt.Tool
)
self.setAttribute(Qt.WA_TranslucentBackground)
self.setFixedSize(self.TOTAL, self.TOTAL)
self.setCursor(QCursor(Qt.PointingHandCursor))
# Smooth animation (~30 fps)
self._timer = QTimer(self)
self._timer.timeout.connect(self._tick)
self._timer.start(33)
# Default position: bottom-right of the work area
scr = QApplication.primaryScreen().availableGeometry()
self.move(scr.right() - self.TOTAL - 20, scr.bottom() - self.TOTAL - 20)
# ── Animation ────────────────────────────────────────
def _tick(self):
# running status: green when model is actively responding
self._running = bool(
getattr(self.chat_panel, "_is_streaming", False)
or getattr(getattr(self.chat_panel, "agent", None), "is_running", False)
)
self._glow += self._glow_dir * 0.04
if self._glow >= 1.0:
self._glow, self._glow_dir = 1.0, -1
elif self._glow <= 0.0:
self._glow, self._glow_dir = 0.0, 1
target = 1.0 if self._hovering else 0.0
self._hover_strength += (target - self._hover_strength) * 0.20
self._hover_clock += 0.033
self._flow_phase += 0.16 + (0.06 if self._running else 0.0) + (0.05 if self._hovering else 0.0)
self.update()
# ── Painting ──────────────────────────────────────────
def paintEvent(self, _event):
p = QPainter(self)
p.setRenderHint(QPainter.Antialiasing)
m = self.MARGIN
r = self.SIZE // 2
cx = m + r
# Rhythmic spring bounce: one main hop + one lighter rebound per beat.
beat_t = self._hover_clock % 1.18
spring = 0.0
if beat_t < 0.70:
spring += max(0.0, math.exp(-5.2 * beat_t) * math.sin(15.5 * beat_t))
if beat_t > 0.20:
rt = beat_t - 0.20
spring += 0.52 * max(0.0, math.exp(-7.0 * rt) * math.sin(21.0 * rt))
idle_sway = 0.20 * math.sin(self._hover_clock * 2.1)
bounce = int(round((spring * 7.2 + idle_sway) * self._hover_strength))
cy = m + r - bounce
if self._running:
# running: #2DFFF5 -> #FFF878
g0 = QColor(45, 255, 245, 195)
g1 = QColor(255, 248, 120, 195)
glow_rgb = (96, 255, 216)
else:
# idle: #103CE7 -> #64E9FF
g0 = QColor(16, 60, 231, 195)
g1 = QColor(100, 233, 255, 195)
glow_rgb = (74, 170, 255)
# --- Outer glow rings (3 layers) ---
base_alpha = int(45 + 25 * self._glow)
for i, gr in enumerate([r + 10, r + 6, r + 2]):
g = QRadialGradient(QPointF(cx, cy), gr)
g.setColorAt(0.0, QColor(glow_rgb[0], glow_rgb[1], glow_rgb[2], max(0, base_alpha - i * 14)))
g.setColorAt(1.0, QColor(glow_rgb[0], glow_rgb[1], glow_rgb[2], 0))
p.setBrush(g)
p.setPen(Qt.NoPen)
p.drawEllipse(int(cx - gr), int(cy - gr), int(gr * 2), int(gr * 2))
# --- Frosted glass disc behind main circle ---
frost = QRadialGradient(QPointF(cx, cy), r)
frost.setColorAt(0.0, QColor(30, 30, 45, 140))
frost.setColorAt(0.85, QColor(20, 20, 32, 160))
frost.setColorAt(1.0, QColor(14, 14, 20, 100))
p.setBrush(frost)
p.setPen(Qt.NoPen)
p.drawEllipse(cx - r, cy - r, r * 2, r * 2)
# --- Main circle (flowing state gradient) ---
spin = self._flow_phase
dx = math.cos(spin) * r
dy = math.sin(spin) * r
grad = QLinearGradient(cx - dx, cy - dy, cx + dx, cy + dy)
grad.setColorAt(0.0, g0)
grad.setColorAt(1.0, g1)
p.setBrush(grad)
p.setPen(QPen(QColor(255, 255, 255, 50), 1.5))
p.drawEllipse(cx - r, cy - r, r * 2, r * 2)
# --- Flowing glass streaks ---
clip = QPainterPath()
clip.addEllipse(float(cx - r), float(cy - r), float(r * 2), float(r * 2))
p.setClipPath(clip)
flow_shift = math.sin(self._flow_phase * 0.85) * (r * 0.7)
streak1 = QLinearGradient(cx - r + flow_shift, cy - r, cx + r + flow_shift, cy + r)
streak1.setColorAt(0.00, QColor(255, 255, 255, 0))
streak1.setColorAt(0.45, QColor(255, 255, 255, 42))
streak1.setColorAt(0.52, QColor(255, 255, 255, 78))
streak1.setColorAt(0.60, QColor(255, 255, 255, 24))
streak1.setColorAt(1.00, QColor(255, 255, 255, 0))
p.setBrush(streak1)
p.setPen(Qt.NoPen)
p.drawEllipse(cx - r, cy - r, r * 2, r * 2)
flow_shift_2 = math.cos(self._flow_phase * 1.2) * (r * 0.5)
streak2 = QLinearGradient(cx - r, cy + flow_shift_2, cx + r, cy - flow_shift_2)
streak2.setColorAt(0.00, QColor(255, 255, 255, 0))
streak2.setColorAt(0.35, QColor(255, 255, 255, 16))
streak2.setColorAt(0.50, QColor(255, 255, 255, 46))
streak2.setColorAt(0.65, QColor(255, 255, 255, 16))
streak2.setColorAt(1.00, QColor(255, 255, 255, 0))
p.setBrush(streak2)
p.drawEllipse(cx - r, cy - r, r * 2, r * 2)
# --- Top highlight ---
hl = QLinearGradient(cx, cy - r, cx, cy)
hl.setColorAt(0.0, QColor(255, 255, 255, 72))
hl.setColorAt(1.0, QColor(255, 255, 255, 0))
p.setBrush(hl)
p.drawRect(cx - r, cy - r, r * 2, r)
p.setClipping(False)
# --- Bot icon ---
p.setPen(QPen(QColor(255, 255, 255, 220), 1.8))
p.setBrush(Qt.NoBrush)
# Head
p.drawRoundedRect(cx - 9, cy - 6, 18, 12, 2, 2)
# Eyes
p.setBrush(QColor(255, 255, 255, 220))
p.setPen(Qt.NoPen)
p.drawEllipse(cx - 6, cy - 3, 4, 4)
p.drawEllipse(cx + 2, cy - 3, 4, 4)
# Antenna stem
p.setPen(QPen(QColor(255, 255, 255, 220), 1.8))
p.drawLine(cx, cy - 6, cx, cy - 10)
# Antenna tip
p.setBrush(QColor(255, 255, 255, 190))
p.setPen(Qt.NoPen)
p.drawEllipse(cx - 2, cy - 13, 4, 4)
def enterEvent(self, event):
self._hovering = True
self.update()
super().enterEvent(event)
def leaveEvent(self, event):
self._hovering = False
self.update()
super().leaveEvent(event)
# ── Mouse events (drag + click) ───────────────────────
def mousePressEvent(self, event):
if event.button() == Qt.LeftButton:
self._drag_origin_global = event.globalPosition().toPoint()
self._drag_origin_win = self.pos()
self._dragged = False
def mouseMoveEvent(self, event):
if event.buttons() == Qt.LeftButton and self._drag_origin_global:
delta = event.globalPosition().toPoint() - self._drag_origin_global
if abs(delta.x()) > 5 or abs(delta.y()) > 5:
self._dragged = True
if self._dragged:
new = self._drag_origin_win + delta
scr = QApplication.primaryScreen().availableGeometry()
new.setX(max(scr.left(), min(new.x(), scr.right() - self.width())))
new.setY(max(scr.top(), min(new.y(), scr.bottom() - self.height())))
self.move(new)
def mouseDoubleClickEvent(self, event):
# Qt sends Press→Release→DoubleClick→Release on double-click.
# The first Release already toggled the panel; swallow the DoubleClick
# so the second Release does NOT trigger a second toggle.
self._dragged = True # mark as "dragged" → Release will be ignored
event.accept()
def mouseReleaseEvent(self, event):
if event.button() == Qt.LeftButton:
if not self._dragged:
self._toggle()
self._dragged = False
self._drag_origin_global = None
# ── Toggle panel ──────────────────────────────────────
def _toggle(self):
now = QDateTime.currentMSecsSinceEpoch()
if now - self._last_toggle_ms < 500: # 500 ms debounce
return
self._last_toggle_ms = now
if self.chat_panel.isVisible():
self.chat_panel.hide()
else:
self._position_panel()
self.chat_panel.show()
self.chat_panel.raise_()
self.chat_panel.activateWindow()
def _position_panel(self):
scr = QApplication.primaryScreen().availableGeometry()
btn = self.geometry()
pw = self.chat_panel.width()
ph = self.chat_panel.height()
# Prefer left of button, bottom-aligned
x = btn.left() - pw - 12
y = btn.bottom() - ph
x = max(scr.left() + 10, min(x, scr.right() - pw - 10))
y = max(scr.top() + 10, min(y, scr.bottom() - ph - 10))
self.chat_panel.move(x, y)
# ══════════════════════════════════════════════════════════════════════
# ChatPanel
# ══════════════════════════════════════════════════════════════════════
# ── constants ─────────────────────────────────────────────────────────────────
HISTORY_FILE = "memory/chat_history.json"
TEXT_FILE_EXTS = {
".txt", ".md", ".py", ".json", ".csv", ".yaml", ".yml",
".log", ".ini", ".toml", ".xml", ".html", ".js", ".ts", ".sql",
}
MAX_INLINE_CHARS = 6000
MAX_UPLOAD_BYTES = 10 * 1024 * 1024 # 10 MB
AUTO_IDLE_THRESHOLD = 1800 # seconds before autonomous trigger
AUTO_COOLDOWN = 120 # seconds between triggers
C = {
"bg": QColor(14, 14, 18),
"panel": QColor(20, 20, 24, 248),
"border": QColor(45, 45, 50),
"accent": "#7c3aed",
"text": "#e4e4e7",
"muted": "#71717a",
"user_g0": QColor(79, 70, 229),
"user_g1": QColor(124, 58, 237),
"asst_bg": QColor(39, 39, 42, 210),
"asst_bdr": QColor(63, 63, 70),
"send_g0": QColor(220, 38, 38),
"send_g1": QColor(239, 68, 68),
"green": "#22c55e",
"hover_bg": "rgba(63,63,70,0.6)",
"accent_bg":"rgba(124,58,237,0.25)",
"accent_bdr":"rgba(124,58,237,0.5)",
}
SCROLLBAR_STYLE = """
QScrollBar:vertical { width: 5px; background: transparent; border: none; }
QScrollBar::handle:vertical {
background: rgba(255,255,255,0.12); border-radius: 2px; min-height: 20px;
}
QScrollBar::add-line:vertical, QScrollBar::sub-line:vertical { height: 0; }
QScrollBar::add-page:vertical, QScrollBar::sub-page:vertical { background: none; }
"""
_SVG_COPY = '")
in_code = not in_code
continue
if in_code:
html.append(raw.replace("&", "&").replace("<", "<").replace(">", ">"))
continue
line = raw
line = re.sub(r"`([^`]+)`", r"\1", line)
line = re.sub(r"\*\*(.+?)\*\*", r"\1 ", line)
line = re.sub(r"\*(.+?)\*", r"\1 ", line)
line = re.sub(r"\[([^\]]+)\]\(([^)]+)\)", r'\1 ', line)
if re.match(r"^#{1,6}\s", line):
lvl = len(line.split()[0])
line = f"{line[lvl:].strip()} "
elif re.match(r"^-{3,}$|^_{3,}$|^\*{3,}$", line.strip()):
line = "")
in_ul = True
line = f"{content} "
else:
if in_ul:
html.append(" ")
in_ul = False
line = f"{line}
" if line.strip() else ""
html.append(line)
if in_code:
html.append("{kw_esc} ')
self._label.setText(highlighted)
self._label.adjustSize()
return 0 # plain text, keyword at top
else:
from PySide6.QtGui import QTextDocument, QTextCursor, QTextCharFormat
doc = self._label.document()
cursor = QTextCursor(doc)
flags = QTextDocument.FindFlags(0)
fmt = QTextCharFormat()
fmt.setBackground(QColor(251, 191, 36, 90))
fmt.setForeground(QColor(251, 191, 36))
keyword_y = None
while True:
cursor = doc.find(keyword, cursor, flags)
if cursor.isNull():
break
cursor.mergeCharFormat(fmt)
if keyword_y is None:
keyword_y = self._label.cursorRect(cursor).y()
self._adjust_browser_height()
return keyword_y
def clear_highlight(self):
if self._role == "user":
self._label.setText(self._text)
self._label.adjustSize()
else:
self._label.setHtml(self._render_with_folds(self._text))
self._adjust_browser_height()
def _parse_foldable_blocks(self, text: str):
"""解析文本为可折叠块，返回 [(type, title_or_None, content), ...]"""
import re
lines = text.split('\n')
blocks = []
current_type = "normal"
current_title = None
current_lines = []
for line in lines:
# 检查是否是折叠块开始
llm_match = re.match(r'^\s*\*\*LLM Running \(Turn \d+\) \.\.\.\*\*\s*$', line)
tool_match = re.match(r'^\s*🛠️\s*Tool:', line)
tool_compact_match = re.match(r'^\s*🛠️\s+\w+\(', line)
is_foldable_start = llm_match or tool_match or tool_compact_match
if is_foldable_start:
if current_lines:
blocks.append((current_type, current_title, '\n'.join(current_lines)))
title = line.strip()
if llm_match:
title = line.strip().replace('**', '')
current_type = "foldable"
current_title = title
current_lines = [line]
else:
current_lines.append(line)
if current_lines:
blocks.append((current_type, current_title, '\n'.join(current_lines)))
return blocks
def _auto_fold_new_blocks(self, text: str):
"""将新出现的折叠块加入 _folded_ids（仅在此处修改集合）"""
for _, title, _ in self._parse_foldable_blocks(text):
if title is not None and title not in self._folded_ids:
self._folded_ids.add(title)
def _render_with_folds(self, text: str) -> str:
"""渲染文本为带折叠的 HTML（纯渲染，不修改 _folded_ids）"""
from urllib.parse import quote
blocks = self._parse_foldable_blocks(text)
html_parts = []
for i, (block_type, title, content) in enumerate(blocks):
if block_type == "normal":
html_parts.append(f'{_md_to_html(content)}
')
else:
safe_title = quote(title, safe='')
display_title = title.replace('**', '')
if title in self._folded_ids:
# 折叠状态：只显示标题 + 展开链接
html_parts.append(
f''
)
else:
# 展开状态：显示标题 + 折叠链接 + 内容
html_parts.append(
f''
)
return '\n'.join(html_parts)
def _toggle_fold(self, title):
"""折叠/展开切换"""
if title in self._folded_ids:
self._folded_ids.remove(title)
else:
self._folded_ids.add(title)
self._label.setHtml(self._render_with_folds(self._text))
self._adjust_browser_height()
class _TabButton(QPushButton):
_STYLE = """
QPushButton {{
background: transparent; color: {muted};
border: none; border-radius: 8px;
padding: 0 14px; font-size: 12px; font-weight: 700;
}}
QPushButton:hover {{
background: {hover_bg}; color: {text};
}}
QPushButton:checked {{
background: {accent}; color: white;
}}
""".format(muted=C["muted"], text=C["text"], hover_bg=C["hover_bg"], accent=C["accent"])
def __init__(self, text: str, parent=None):
super().__init__(text, parent)
self.setCheckable(True)
self.setFixedHeight(30)
self.setStyleSheet(self._STYLE)
def _action_btn(label: str, color: str, icon: QIcon | None = None) -> QPushButton:
btn = QPushButton(label)
if icon and not icon.isNull():
btn.setIcon(icon)
btn.setIconSize(QSize(16, 16))
btn.setFixedHeight(36)
btn.setStyleSheet(f"""
QPushButton {{
background: rgba(35,35,40,0.8); color: {C['text']};
border: 1px solid {C['border'].name()};
border-left: 3px solid {color};
border-radius: 8px; padding: 0 14px;
font-size: 13px; font-weight: 700; text-align: left;
}}
QPushButton:hover {{ background: rgba(55,55,62,0.9); }}
QPushButton:checked {{ color: {color}; background: rgba(35,35,40,0.95); }}
""")
return btn
# ── Main panel ────────────────────────────────────────────────────────────────
class ChatPanel(QWidget):
"""Frameless always-on-top chat window."""
def __init__(self, agent):
super().__init__()
self.agent = agent
# session state
self._messages: list[dict] = []
self._session = {"id": _make_session_id(), "title": "新对话", "messages": []}
self._history: list[dict] = _load_history()
self._pending_files: list[dict] = [] # {'name','type','raw'}
self._settings_health_checked = False
# streaming state
self._display_queue: Optional[_queue.Queue] = None
self._streaming_row: Optional[_MsgRow] = None
self._streaming_text = ""
self._user_scrolled_up = False
self._poll_timer = QTimer(self)
self._poll_timer.timeout.connect(self._poll_queue)
# autonomous mode
self.autonomous_enabled = False
self.last_reply_time = time.time()
self.setWindowFlags(
Qt.FramelessWindowHint | Qt.WindowStaysOnTopHint | Qt.Tool
)
self.setAttribute(Qt.WA_TranslucentBackground)
self.resize(530, 700)
# drag state (title bar)
self._drag_pos: Optional[QPoint] = None
self._build_ui()
def paintEvent(self, _event):
p = QPainter(self)
p.setRenderHint(QPainter.Antialiasing)
path = QPainterPath()
path.addRect(0.5, 0.5, self.width() - 1.0, self.height() - 1.0)
grad = QLinearGradient(0, 0, 0, self.height())
grad.setColorAt(0.0, QColor(20, 20, 28, 255))
grad.setColorAt(1.0, QColor(10, 10, 14, 255))
p.fillPath(path, grad)
def resizeEvent(self, event):
path = QPainterPath()
path.addRect(0, 0, float(self.width()), float(self.height()))
self.setMask(QRegion(path.toFillPolygon().toPolygon()))
super().resizeEvent(event)
# ── UI construction ───────────────────────────────────────────────────────
def _build_ui(self):
root = QVBoxLayout(self)
root.setContentsMargins(0, 0, 0, 0)
root.setSpacing(0)
root.addWidget(self._build_titlebar())
root.addWidget(_Separator())
root.addWidget(self._build_tabbar())
root.addWidget(_Separator())
self._stack = QStackedWidget()
self._stack.setStyleSheet("background: transparent;")
self._stack.addWidget(self._build_chat_page()) # 0
self._stack.addWidget(self._build_history_page()) # 1
self._stack.addWidget(self._build_sop_page()) # 2
self._stack.addWidget(self._build_settings_page())# 3
root.addWidget(self._stack)
root.addWidget(self._build_statusbar())
# Now that _stack exists, activate the first tab
self._switch_tab(0)
# ── title bar ─────────────────────────────────────────────────────────────
def _build_titlebar(self) -> QWidget:
bar = QWidget()
bar.setFixedHeight(48)
bar.setStyleSheet("background: transparent;")
bar.setCursor(QCursor(Qt.SizeAllCursor))
ly = QHBoxLayout(bar)
ly.setContentsMargins(16, 0, 10, 0)
ly.setSpacing(8)
# Search button
search_btn = QPushButton()
search_btn.setIcon(_svg_icon("search", _SVG_SEARCH, "#a1a1aa"))
search_btn.setIconSize(QSize(16, 16))
search_btn.setFixedSize(26, 26)
search_btn.setCursor(QCursor(Qt.PointingHandCursor))
search_btn.setStyleSheet("""
QPushButton { background: transparent; border: none; border-radius: 13px; }
QPushButton:hover { background: rgba(63,63,70,0.6); }
""")
search_btn.clicked.connect(self._toggle_search)
self._search_btn = search_btn
ly.addWidget(search_btn)
# Search widget (hidden by default)
self._search_widget = QWidget()
self._search_widget.hide()
sw_ly = QHBoxLayout(self._search_widget)
sw_ly.setContentsMargins(0, 0, 0, 0)
sw_ly.setSpacing(6)
self._search_input = QLineEdit()
self._search_input.setPlaceholderText("搜索当前对话和历史...")
self._search_input.setFixedHeight(26)
self._search_input.setStyleSheet(f"""
QLineEdit {{
background: rgba(32,32,38,0.9);
border: 1px solid {C['border'].name()};
border-radius: 13px;
color: {C['text']};
font-size: 13px;
padding: 0 10px;
}}
QLineEdit::placeholder {{ color: {C['muted']}; }}
""")
self._search_input.setFixedWidth(200)
self._search_input.textChanged.connect(self._on_search_changed)
self._search_input.installEventFilter(self)
sw_ly.addWidget(self._search_input)
close_search = QPushButton("×")
close_search.setFixedSize(26, 26)
close_search.setCursor(QCursor(Qt.PointingHandCursor))
close_search.setStyleSheet("""
QPushButton { background: transparent; color: #71717a; border: none; font-size: 16px; }
QPushButton:hover { color: #a1a1aa; }
""")
close_search.clicked.connect(self._hide_search)
sw_ly.addWidget(close_search)
ly.addWidget(self._search_widget)
ly.addStretch()
# Minimize button
mini = QPushButton("\uE949")
mini.setFixedSize(26, 26)
mini.setCursor(QCursor(Qt.PointingHandCursor))
mini.setStyleSheet("""
QPushButton { background: rgba(63,63,70,0.6); color: #a1a1aa;
border: none; border-radius: 13px; font-family: "Segoe MDL2 Assets"; font-size: 9px; }
QPushButton:hover { background: rgba(63,63,70,0.9); color: white; }
""")
mini.clicked.connect(self.hide)
ly.addWidget(mini)
# Maximize button
maxi = QPushButton("\uE739")
maxi.setFixedSize(26, 26)
maxi.setCursor(QCursor(Qt.PointingHandCursor))
maxi.setStyleSheet("""
QPushButton { background: rgba(63,63,70,0.6); color: #a1a1aa;
border: none; border-radius: 13px; font-family: "Segoe MDL2 Assets"; font-size: 9px; }
QPushButton:hover { background: rgba(63,63,70,0.9); color: white; }
""")
maxi.clicked.connect(self._toggle_maximize)
self._maxi_btn = maxi
ly.addWidget(maxi)
# Close button
close = QPushButton("\uE8BB")
close.setFixedSize(26, 26)
close.setCursor(QCursor(Qt.PointingHandCursor))
close.setStyleSheet("""
QPushButton { background: rgba(63,63,70,0.6); color: #a1a1aa;
border: none; border-radius: 13px; font-family: "Segoe MDL2 Assets"; font-size: 9px; }
QPushButton:hover { background: rgba(220,38,38,0.85); color: white; }
""")
close.clicked.connect(lambda: (self.close(), QApplication.instance().quit()))
ly.addWidget(close)
# Drag
bar.mousePressEvent = self._tb_press
bar.mouseMoveEvent = self._tb_move
bar.mouseReleaseEvent = self._tb_release
return bar
def _toggle_search(self):
if hasattr(self, "_search_visible") and self._search_visible:
self._hide_search()
else:
self._show_search()
def _show_search(self):
self._search_visible = True
self._search_btn.setFixedSize(0, 0)
self._search_widget.show()
self._search_input.setFocus()
self._search_input.selectAll()
def _hide_search(self):
self._search_visible = False
self._search_btn.setFixedSize(26, 26)
self._search_widget.hide()
self._search_input.clear()
self._clear_all_highlights()
if self._stack.currentIndex() == 1:
self._reset_history_items_style()
def _hide_search_if_no_focus(self):
if not self._search_input.hasFocus():
self._hide_search()
def _on_search_changed(self, text):
if not text.strip():
self._clear_all_highlights()
return
keyword = text.strip()
current_tab = self._stack.currentIndex()
if current_tab == 0:
self._search_current_chat(keyword)
elif current_tab == 1:
self._search_history(keyword)
def _clear_all_highlights(self):
for i in range(self._msg_layout.count() - 1):
w = self._msg_layout.itemAt(i).widget()
if isinstance(w, _MsgRow):
w.clear_highlight()
def _search_current_chat(self, keyword: str):
first_found = None
first_keyword_y = None
for i in range(self._msg_layout.count() - 1):
w = self._msg_layout.itemAt(i).widget()
if isinstance(w, _MsgRow):
if keyword.lower() in w._text.lower():
kw_y = w.highlight(keyword)
if first_found is None:
first_found = w
first_keyword_y = kw_y
else:
w.clear_highlight()
# 滚动到第一个匹配项（使用关键词在文档内的实际位置）
if first_found:
self._scroll_to_widget(first_found, first_keyword_y or 0)
def _scroll_to_widget(self, w, keyword_y=0):
self._user_scrolled_up = True
self._msg_container.layout().activate()
QApplication.processEvents()
sb = self._scroll.verticalScrollBar()
vp_h = self._scroll.viewport().height()
keyword_screen_y = w.y() + keyword_y
target = keyword_screen_y - vp_h // 3
target = max(0, min(target, sb.maximum()))
sb.setValue(target)
QApplication.processEvents()
self._scroll.viewport().repaint()
def _search_history(self, keyword: str):
kw_lower = keyword.lower()
for i in range(self._hist_list.count()):
item = self._hist_list.item(i)
session = item.data(Qt.UserRole)
messages = session.get("messages", []) if session else []
content_text = " ".join([m.get("content", "") for m in messages if isinstance(m.get("content"), str)])
match = kw_lower in content_text.lower()
item.setHidden(not match)
if match:
item.setBackground(QColor(251, 191, 36, 50))
item.setForeground(QColor(251, 191, 36))
else:
item.setBackground(QColor(0, 0, 0, 0))
item.setForeground(QColor(255, 255, 255))
def _reset_history_items_style(self):
for i in range(self._hist_list.count()):
item = self._hist_list.item(i)
item.setHidden(False)
item.setBackground(QColor(0, 0, 0, 0))
item.setForeground(QColor(255, 255, 255))
w = self._hist_list.itemWidget(item)
if w:
w.setStyleSheet(
f"background: rgba(35,35,42,0.6); color: {C['text']};"
" border: 1px solid #3f3f46; border-radius: 8px;"
" padding: 8px 12px; margin: 2px 0;"
)
def _tb_press(self, e):
if e.button() == Qt.LeftButton:
self._drag_pos = e.globalPosition().toPoint() - self.pos()
def _tb_move(self, e):
if e.buttons() == Qt.LeftButton and self._drag_pos is not None:
self.move(e.globalPosition().toPoint() - self._drag_pos)
def _tb_release(self, _e):
self._drag_pos = None
def _toggle_maximize(self):
if self.isMaximized():
self.showNormal()
self._maxi_btn.setText("☐")
else:
self.showMaximized()
self._maxi_btn.setText("❐")
# ── status bar ─────────────────────────────────────────────────────────────
def _build_statusbar(self) -> QWidget:
bar = QWidget()
bar.setFixedHeight(24)
bar.setStyleSheet("background: transparent;")
ly = QHBoxLayout(bar)
ly.setContentsMargins(16, 0, 10, 0)
ly.setSpacing(8)
# Status dot
dot = QLabel("●")
dot.setStyleSheet(f"color: {C['green']}; font-size: 9px;")
dot.setFixedWidth(12)
ly.addWidget(dot)
# Model name (clickable to show model list)
self._model_badge = QLabel(self._model_name())
self._model_badge.setStyleSheet("color: #a1a1aa; font-size: 11px;")
self._model_badge.setCursor(QCursor(Qt.PointingHandCursor))
self._model_badge.mousePressEvent = lambda e: self._show_model_menu(e)
ly.addWidget(self._model_badge)
self._streaming_badge = _StreamingBadge()
ly.addWidget(self._streaming_badge)
ly.addStretch()
return bar
def _show_model_menu(self, _e):
menu = QMenu(self._model_badge)
menu.setStyleSheet(f"""
QMenu {{
background: {C['panel'].name()};
border: 1px solid {C['border'].name()};
padding: 4px 0;
}}
QMenu::item {{
color: {C['text']};
padding: 6px 20px 6px 12px;
font-size: 12px;
}}
QMenu::item:selected {{
background: {C['hover_bg']};
}}
""")
for i, client in enumerate(self.agent.llmclients):
name = getattr(client, 'name', None) or "未知"
act = menu.addAction(f"{name} #{i + 1}")
act.triggered.connect(lambda _, idx=i: self._do_switch_to(idx))
menu.exec(QCursor.pos())
# ── tab bar ───────────────────────────────────────────────────────────────
def _build_tabbar(self) -> QWidget:
bar = QWidget()
bar.setFixedHeight(40)
bar.setStyleSheet("background: rgba(10,10,14,0.6);")
ly = QHBoxLayout(bar)
ly.setContentsMargins(12, 5, 12, 5)
ly.setSpacing(4)
self._tabs: list[_TabButton] = []
tab_defs = [
(_SVG_CHAT, "对话"),
(_SVG_CLOCK, "历史"),
(_SVG_BOOK, "SOP"),
(_SVG_GEAR, "设置"),
]
for i, (svg, text) in enumerate(tab_defs):
btn = _TabButton(text)
btn.setIcon(_svg_icon(text, svg, "#b0b0b8"))
btn.setIconSize(QSize(14, 14))
btn.clicked.connect(lambda _checked, idx=i: self._switch_tab(idx))
ly.addWidget(btn)
self._tabs.append(btn)
ly.addStretch()
new_btn = QPushButton("新对话")
new_btn.setIcon(_svg_icon("plus", _SVG_PLUS, "#a78bfa"))
new_btn.setIconSize(QSize(12, 12))
new_btn.setFixedHeight(27)
new_btn.setStyleSheet(f"""
QPushButton {{ background: rgba(124,58,237,0.18); color: #a78bfa;
border: 1px solid rgba(124,58,237,0.3); border-radius: 7px;
padding: 0 10px; font-size: 12px; font-weight: 700; }}
QPushButton:hover {{ background: rgba(124,58,237,0.35); color: white; }}
""")
new_btn.clicked.connect(self._new_session)
ly.addWidget(new_btn)
# NOTE: _switch_tab(0) is called in _build_ui() after _stack is created
return bar
def _switch_tab(self, idx: int):
self._stack.setCurrentIndex(idx)
for i, btn in enumerate(self._tabs):
btn.setChecked(i == idx)
# 切换标签时关闭搜索框
if hasattr(self, '_search_visible') and self._search_visible:
self._hide_search()
if idx == 1:
self._refresh_history()
if idx == 2:
self._refresh_sop()
if idx == 3:
self._refresh_model_rows_style()
if not self._settings_health_checked:
self._start_health_checks()
self._settings_health_checked = True
# ── chat page ─────────────────────────────────────────────────────────────
def _build_chat_page(self) -> QWidget:
page = QWidget()
page.setStyleSheet("background: transparent;")
ly = QVBoxLayout(page)
ly.setContentsMargins(0, 0, 0, 0)
ly.setSpacing(0)
# ── message scroll area ──
self._scroll = QScrollArea()
self._scroll.setWidgetResizable(True)
self._scroll.setFrameShape(QFrame.NoFrame)
self._scroll.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self._scroll.setStyleSheet(f"QScrollArea {{ background: transparent; border: none; }} {SCROLLBAR_STYLE}")
self._msg_container = QWidget()
self._msg_container.setStyleSheet("background: transparent;")
self._msg_layout = QVBoxLayout(self._msg_container)
self._msg_layout.setContentsMargins(0, 12, 0, 12)
self._msg_layout.setSpacing(4)
self._msg_layout.addStretch()
self._scroll.setWidget(self._msg_container)
self._scroll.verticalScrollBar().valueChanged.connect(self._on_scroll)
# ── scroll navigation buttons (centered at bottom of message area) ──
scroll_wrapper = QWidget()
scroll_wrapper.setStyleSheet("background: transparent;")
wrap_ly = QVBoxLayout(scroll_wrapper)
wrap_ly.setContentsMargins(0, 0, 0, 0)
wrap_ly.setSpacing(0)
wrap_ly.addWidget(self._scroll)
self._nav_widget = QWidget()
self._nav_widget.setFixedSize(68, 28)
self._nav_widget.setStyleSheet("background: transparent; border: none;")
nav_ly = QHBoxLayout(self._nav_widget)
nav_ly.setContentsMargins(6, 2, 6, 2)
nav_ly.setSpacing(8)
self._nav_up = QPushButton("∧")
self._nav_up.setFixedWidth(26)
self._nav_up.setCursor(QCursor(Qt.PointingHandCursor))
self._nav_up.setStyleSheet("""
QPushButton { background: transparent; color: #71717a; border: none; font-size: 14px; }
QPushButton:hover { color: #a1a1aa; }
QPushButton:disabled { color: #27272a; }
""")
self._nav_up.clicked.connect(self._scroll_to_top)
self._nav_down = QPushButton("∨")
self._nav_down.setFixedWidth(26)
self._nav_down.setCursor(QCursor(Qt.PointingHandCursor))
self._nav_down.setStyleSheet("""
QPushButton { background: transparent; color: #71717a; border: none; font-size: 14px; }
QPushButton:hover { color: #a1a1aa; }
QPushButton:disabled { color: #27272a; }
""")
self._nav_down.clicked.connect(self._scroll_to_bottom)
nav_ly.addWidget(self._nav_up)
nav_ly.addWidget(self._nav_down)
wrap_ly.addWidget(self._nav_widget, 0, Qt.AlignHCenter | Qt.AlignBottom)
self._nav_widget.setContentsMargins(0, 0, 0, 8)
self._nav_widget.hide()
ly.addWidget(scroll_wrapper, 1)
ly.addWidget(_Separator())
# ── input area ──
ly.addWidget(self._build_input_area())
QTimer.singleShot(200, self._update_nav_visibility)
return page
def _build_input_area(self) -> QWidget:
wrap = QWidget()
wrap.setStyleSheet("background: transparent;")
ly = QVBoxLayout(wrap)
ly.setContentsMargins(20, 6, 20, 0)
ly.setSpacing(0)
self._chips_row = QWidget()
self._chips_row.setStyleSheet("background: transparent;")
self._chips_ly = QHBoxLayout(self._chips_row)
self._chips_ly.setContentsMargins(0, 0, 0, 6)
self._chips_ly.setSpacing(6)
self._chips_row.hide()
ly.addWidget(self._chips_row)
card = QWidget()
card.setStyleSheet(f"""
QWidget#inputCard {{
background: rgba(32,32,38,0.85);
border: 1px solid {C['border'].name()};
border-radius: 16px;
}}
QWidget#inputCard:focus-within {{
border-color: rgba(124,58,237,0.55);
}}
""")
card.setObjectName("inputCard")
card_ly = QVBoxLayout(card)
card_ly.setContentsMargins(14, 10, 10, 10)
card_ly.setSpacing(6)
class _PlainTextEdit(QTextEdit):
def insertFromMimeData(self, source):
text = source.text() or source.data("text/plain")
if text:
self.insertPlainText(text)
self._input = _PlainTextEdit()
self._input.setAutoFormatting(QTextEdit.AutoNone)
self._input.setFixedHeight(64)
self._input.setPlaceholderText("给助手发送消息... Enter发送，Shift+Enter换行")
self._input.setStyleSheet(f"""
QTextEdit {{
background: transparent; color: {C['text']};
border: none; padding: 0; font-size: 14px;
selection-background-color: rgba(124,58,237,0.4);
}}
""")
self._input.installEventFilter(self)
self._input.textChanged.connect(self._on_text_changed)
card_ly.addWidget(self._input)
bottom = QHBoxLayout()
bottom.setSpacing(6)
attach = QPushButton()
attach.setIcon(_svg_icon("clip", _SVG_CLIP, "#a1a1aa"))
attach.setIconSize(QSize(17, 17))
attach.setFixedSize(30, 30)
attach.setToolTip("上传附件")
attach.setCursor(QCursor(Qt.PointingHandCursor))
attach.setStyleSheet("""
QPushButton { background: transparent; border: none; border-radius: 15px; }
QPushButton:hover { background: rgba(63,63,70,0.6); }
""")
attach.clicked.connect(self._attach_files)
bottom.addWidget(attach)
self._char_lbl = QLabel("0 / 2000")
self._char_lbl.setStyleSheet(f"color: {C['muted']}; font-size: 11px;")
bottom.addWidget(self._char_lbl)
self._token_lbl = QLabel("")
self._token_lbl.setStyleSheet(f"color: {C['muted']}; font-size: 11px; margin-left: 10px;")
bottom.addWidget(self._token_lbl)
bottom.addStretch()
self._is_streaming = False
self._send_btn = QPushButton()
self._send_btn.setFixedSize(34, 34)
self._send_btn.setCursor(QCursor(Qt.PointingHandCursor))
self._send_btn.clicked.connect(self._on_send_btn_click)
self._set_send_mode()
bottom.addWidget(self._send_btn)
card_ly.addLayout(bottom)
ly.addWidget(card)
return wrap
# ── history page ──────────────────────────────────────────────────────────
def _build_history_page(self) -> QWidget:
page = QWidget()
page.setStyleSheet("background: transparent;")
ly = QVBoxLayout(page)
ly.setContentsMargins(12, 12, 12, 12)
ly.setSpacing(8)
header = QHBoxLayout()
lbl = QLabel("历史记录")
lbl.setStyleSheet("color: #f4f4f5; font-weight: 600; font-size: 14px;")
header.addWidget(lbl)
header.addStretch()
restore_btn = QPushButton("恢复会话")
restore_btn.setStyleSheet(self._small_btn_style(C["accent"]))
restore_btn.clicked.connect(self._restore_selected)
header.addWidget(restore_btn)
del_btn = QPushButton("删除")
del_btn.setStyleSheet(self._small_btn_style("#dc2626"))
del_btn.clicked.connect(self._delete_selected)
header.addWidget(del_btn)
ly.addLayout(header)
self._hist_list = QListWidget()
self._hist_list.setStyleSheet(f"""
QListWidget {{ background: transparent; border: none; outline: none; }}
QListWidget::item {{
background: rgba(35,35,42,0.6); color: {C['text']};
border: 1px solid {C['border'].name()}; border-radius: 8px;
padding: 8px 12px; margin: 2px 0;
}}
QListWidget::item:hover {{ background: rgba(55,55,65,0.8);
border-color: rgba(124,58,237,0.4); }}
QListWidget::item:selected {{ background: {C["accent_bg"]};
border-color: rgba(124,58,237,0.6); }}
{SCROLLBAR_STYLE}
""")
self._hist_list.itemDoubleClicked.connect(self._restore_selected)
ly.addWidget(self._hist_list)
return page
# ── SOP page ──────────────────────────────────────────────────────────────
def _build_sop_page(self) -> QWidget:
page = QWidget()
page.setStyleSheet("background: transparent;")
ly = QVBoxLayout(page)
ly.setContentsMargins(0, 0, 0, 0)
splitter = QSplitter(Qt.Horizontal)
self._sop_list = QListWidget()
self._sop_list.setMaximumWidth(175)
self._sop_list.setStyleSheet(f"""
QListWidget {{ background: rgba(10,10,14,0.7); border: none;
border-right: 1px solid {C['border'].name()}; outline: none; }}
QListWidget::item {{ color: {C['muted']}; padding: 7px 10px;
border-radius: 4px; margin: 1px 4px; }}
QListWidget::item:hover {{ background: rgba(55,55,65,0.7); color: {C['text']}; }}
QListWidget::item:selected {{ background: rgba(124,58,237,0.28); color: white; }}
{SCROLLBAR_STYLE}
""")
self._sop_list.currentItemChanged.connect(self._load_sop)
splitter.addWidget(self._sop_list)
self._sop_viewer = QTextBrowser()
self._sop_viewer.setOpenExternalLinks(True)
self._sop_viewer.document().setDefaultStyleSheet(_MD_CSS)
self._sop_viewer.setStyleSheet(f"""
QTextBrowser {{ background: transparent; color: {C['text']};
border: none; padding: 10px 14px;
font-family: "Arial", "Microsoft YaHei", sans-serif;
font-size: 13px; }}
{SCROLLBAR_STYLE}
""")
splitter.addWidget(self._sop_viewer)
splitter.setSizes([165, 340])
ly.addWidget(splitter)
return page
# ── settings page ─────────────────────────────────────────────────────────
def _build_settings_page(self) -> QWidget:
page = QWidget()
page.setStyleSheet("background: transparent;")
ly = QVBoxLayout(page)
ly.setContentsMargins(16, 16, 16, 16)
ly.setSpacing(8)
lbl = QLabel("控制面板")
lbl.setStyleSheet("color: #f4f4f5; font-weight: 600; font-size: 14px;")
ly.addWidget(lbl)
self._model_info = QLabel(f"当前模型：{self._model_name()} (#{self.agent.llm_no})")
self._model_info.setStyleSheet(f"color: {C['muted']}; font-size: 12px;")
ly.addWidget(self._model_info)
ly.addSpacing(4)
model_hdr = QLabel("模型列表")
model_hdr.setStyleSheet("color: #d4d4d8; font-weight: 600; font-size: 13px;")
ly.addWidget(model_hdr)
self._model_rows_container = QWidget()
self._model_rows_container.setStyleSheet("background: transparent;")
self._model_rows_layout = QVBoxLayout(self._model_rows_container)
self._model_rows_layout.setContentsMargins(0, 0, 0, 0)
self._model_rows_layout.setSpacing(3)
ly.addWidget(self._model_rows_container)
self._model_row_widgets: list[dict] = []
self._health_results: dict[int, bool | None] = {}
self._build_model_rows()
ly.addSpacing(6)
for (lbl_text, color, handler, svg) in [
("重置提示词", "#059669", self._do_reset_prompt, _SVG_RESET),
("保存当前会话","#0ea5e9", self._do_save, _SVG_SAVE),
("清空对话", "#78716c", self._do_clear, _SVG_TRASH),
]:
b = _action_btn(lbl_text, color, _svg_icon(lbl_text, svg))
b.clicked.connect(handler)
ly.addWidget(b)
ly.addSpacing(10)
sep = QLabel("自主行动")
sep.setStyleSheet("color: #f4f4f5; font-weight: 600; font-size: 13px;")
ly.addWidget(sep)
self._auto_btn = _action_btn(f"开启自主行动 (idle > {AUTO_IDLE_THRESHOLD // 60} min 自动触发)", "#f59e0b",
_svg_icon("bolt", _SVG_BOLT))
self._auto_btn.setCheckable(True)
self._auto_btn.clicked.connect(self._do_toggle_auto)
ly.addWidget(self._auto_btn)
trigger_btn = _action_btn("立即触发一次", "#f59e0b",
_svg_icon("play", _SVG_PLAY))
trigger_btn.clicked.connect(self._do_trigger_auto)
ly.addWidget(trigger_btn)
ly.addStretch()
return page
# ── model list ────────────────────────────────────────────────────────────
_MODEL_ROW_STYLE = (
"QPushButton { background: rgba(39,39,42,0.7); color: #e4e4e7;"
" border: 1px solid #3f3f46; border-radius: 8px;"
" padding: 6px 10px; font-size: 12px; font-weight: 700; text-align: left; }"
" QPushButton:hover { background: rgba(63,63,70,0.8); }"
)
_MODEL_ROW_ACTIVE = (
"QPushButton { background: rgba(124,58,237,0.25); color: #c4b5fd;"
" border: 1px solid rgba(124,58,237,0.5); border-radius: 8px;"
" padding: 6px 10px; font-size: 12px; font-weight: 700; text-align: left; }"
" QPushButton:hover { background: rgba(124,58,237,0.35); }"
)
def _build_model_rows(self):
while self._model_rows_layout.count():
w = self._model_rows_layout.takeAt(0).widget()
if w:
w.deleteLater()
self._model_row_widgets.clear()
for idx, tc in enumerate(self.agent.llmclients):
b = tc.backend
name = f"{type(b).__name__}/{b.model}"
is_current = idx == self.agent.llm_no
row = QWidget()
row.setStyleSheet("background: transparent;")
rlay = QHBoxLayout(row)
rlay.setContentsMargins(0, 0, 0, 0)
rlay.setSpacing(6)
dot = QLabel("●")
dot.setFixedWidth(14)
dot.setAlignment(Qt.AlignCenter)
dot.setStyleSheet("color: #71717a; font-size: 11px;")
rlay.addWidget(dot)
btn = QPushButton(f" #{idx} {name}")
btn.setCursor(QCursor(Qt.PointingHandCursor))
btn.setStyleSheet(self._MODEL_ROW_ACTIVE if is_current else self._MODEL_ROW_STYLE)
btn.clicked.connect(lambda checked, i=idx: self._do_switch_to(i))
rlay.addWidget(btn, 1)
self._model_rows_layout.addWidget(row)
self._model_row_widgets.append({"dot": dot, "btn": btn, "idx": idx})
def _refresh_model_rows_style(self):
for entry in self._model_row_widgets:
is_current = entry["idx"] == self.agent.llm_no
entry["btn"].setStyleSheet(
self._MODEL_ROW_ACTIVE if is_current else self._MODEL_ROW_STYLE
)
status = self._health_results.get(entry["idx"])
if status is True:
entry["dot"].setStyleSheet("color: #22c55e; font-size: 11px;")
elif status is False:
entry["dot"].setStyleSheet("color: #ef4444; font-size: 11px;")
else:
entry["dot"].setStyleSheet("color: #71717a; font-size: 11px;")
def _do_switch_to(self, idx: int):
if idx == self.agent.llm_no:
return
self.agent.next_llm(n=idx)
name = self._model_name()
self._model_badge.setText(name)
self._model_info.setText(f"当前模型：{name} (#{self.agent.llm_no})")
self._add_system_notice(f"已切换至 {name}，对话上下文已保留")
self._refresh_model_rows_style()
def _start_health_checks(self):
self._health_results.clear()
self._health_pending = 0
self._health_result_queue = _queue.Queue()
for entry in self._model_row_widgets:
entry["dot"].setStyleSheet("color: #71717a; font-size: 11px;")
entry["dot"].setText("◌")
for idx, tc in enumerate(self.agent.llmclients):
self._health_pending += 1
t = threading.Thread(target=self._check_backend, args=(idx, tc.backend), daemon=True)
t.start()
if not hasattr(self, '_health_poll_timer'):
self._health_poll_timer = QTimer(self)
self._health_poll_timer.timeout.connect(self._poll_health_results)
self._health_poll_timer.start(500)
def _poll_health_results(self):
while True:
try:
idx, ok = self._health_result_queue.get_nowait()
self._health_results[idx] = ok
except _queue.Empty:
break
self._refresh_model_rows_style()
if len(self._health_results) >= self._health_pending:
self._health_poll_timer.stop()
def _check_backend(self, idx: int, backend):
ok = False
try:
reply = backend.ask("你好")
# 兼容生成器函数（NativeClaudeSession.ask是生成器）
if hasattr(reply, '__iter__') and not isinstance(reply, str):
reply = ''.join(str(b) for b in reply if isinstance(b, str))
text = str(reply).strip() if reply else ""
ok = len(text) > 0 and not text.startswith("Error") and not text.startswith("[")
print(f"[HealthCheck] Backend #{idx} {type(backend).__name__}/{backend.model}: {'OK' if ok else 'FAIL'} -> {text[:60]}")
except Exception as e:
print(f"[HealthCheck] Backend #{idx} {type(backend).__name__}/{backend.model}: ERROR -> {e}")
ok = False
if hasattr(backend, 'raw_msgs') and backend.raw_msgs:
backend.raw_msgs = [m for m in backend.raw_msgs if m.get("prompt") != "你好"]
self._health_result_queue.put((idx, ok))
# ── event filter (Enter key in text edit, Escape to close search) ──────────
def eventFilter(self, obj, event):
if event.type() == QEvent.KeyPress:
if obj is self._search_input and event.key() == Qt.Key_Escape:
self._hide_search()
return True
if obj is self._input and event.key() in (Qt.Key_Return, Qt.Key_Enter):
if not (event.modifiers() & Qt.ShiftModifier):
self._handle_send()
return True
# 搜索框失焦时关闭搜索
if event.type() == QEvent.FocusOut and obj is self._search_input:
# 延迟关闭，等待点击事件处理完毕
QTimer.singleShot(50, self._hide_search_if_no_focus)
return super().eventFilter(obj, event)
def _on_text_changed(self):
n = len(self._input.toPlainText())
self._char_lbl.setText(f"{n} / 2000")
# ── file attachment ────────────────────────────────────────────────────────
def _attach_files(self):
paths, _ = QFileDialog.getOpenFileNames(
self, "选择附件", "",
"All Files (*);;"
"Images (*.png *.jpg *.jpeg *.gif *.webp *.bmp);;"
"Text (*.txt *.md *.py *.json *.csv *.yaml *.yml *.log *.js *.ts *.sql)",
)
for path in paths:
name = os.path.basename(path)
if any(f["name"] == name for f in self._pending_files):
continue
ext = os.path.splitext(path)[1].lower()
img_exts = {".png", ".jpg", ".jpeg", ".gif", ".webp", ".bmp"}
mime = (f"image/{ext[1:]}" if ext in img_exts else
"text/plain" if ext in TEXT_FILE_EXTS else
"application/octet-stream")
try:
with open(path, "rb") as fh:
raw = fh.read()
if len(raw) > MAX_UPLOAD_BYTES:
print(f"[Attach] 文件过大，已跳过: {name} ({len(raw)} bytes)")
continue
self._pending_files.append({"name": name, "type": mime, "raw": raw})
except Exception as e:
print(f"[Attach] Failed to read {path}: {e}")
self._refresh_chips()
def _refresh_chips(self):
while self._chips_ly.count():
item = self._chips_ly.takeAt(0)
if item.widget():
item.widget().deleteLater()
if not self._pending_files:
self._chips_row.hide()
return
for f in self._pending_files:
chip = QLabel(f['name'])
chip.setStyleSheet(f"""
QLabel {{ background: rgba(55,55,65,0.7); color: {C['text']};
border: 1px solid {C['border'].name()}; border-radius: 6px;
padding: 3px 8px; font-size: 11px; }}
""")
self._chips_ly.addWidget(chip)
self._chips_ly.addStretch()
self._chips_row.show()
# ── send / streaming ───────────────────────────────────────────────────────
_SEND_BTN_STYLE = """
QPushButton { background: #e4e4e7; border: none; border-radius: 17px; }
QPushButton:hover { background: #f4f4f5; }
QPushButton:pressed { background: #d4d4d8; }
"""
_STOP_BTN_STYLE = """
QPushButton { background: rgba(239,68,68,0.85); border: none; border-radius: 17px; }
QPushButton:hover { background: rgba(248,113,113,0.9); }
QPushButton:pressed { background: rgba(220,38,38,0.9); }
"""
def _set_send_mode(self):
self._is_streaming = False
self._send_btn.setText("")
self._send_btn.setIcon(_svg_icon("send_arrow", _SVG_SEND, "#18181b"))
self._send_btn.setIconSize(QSize(18, 18))
self._send_btn.setStyleSheet(self._SEND_BTN_STYLE)
def _set_stop_mode(self):
self._is_streaming = True
self._send_btn.setText("")
self._send_btn.setIcon(_svg_icon("stop_circle", _SVG_STOP, "#ffffff"))
self._send_btn.setIconSize(QSize(16, 16))
self._send_btn.setStyleSheet(self._STOP_BTN_STYLE)
def _on_send_btn_click(self):
if self._is_streaming:
self._do_stop()
else:
self._handle_send()
def _handle_send(self):
text = self._input.toPlainText().strip()
files = self._pending_files.copy()
if not text and not files:
return
if text.startswith("/"):
self._input.clear()
self._pending_files.clear()
self._refresh_chips()
self._handle_command(text)
return
prompt = text or "请分析我上传的附件。"
full_prompt, display_prompt, _ = _build_prompt_with_uploads(prompt, files)
# Clear input state
self._input.clear()
self._pending_files.clear()
self._refresh_chips()
# Update session title
if self._session["title"] == "新对话" and prompt:
self._session["title"] = prompt[:20] + ("..." if len(prompt) > 20 else "")
from datetime import datetime
now_iso = datetime.now().isoformat()
user_idx = len(self._messages)
self._messages.append({"role": "user", "content": display_prompt, "created_at": now_iso})
self._add_msg_row(
"user",
display_prompt,
created_at=now_iso,
on_delete=lambda idx=user_idx: self._delete_message(idx),
on_rewrite=lambda idx=user_idx: self._rewrite_message(idx)
)
self._update_token_usage()
# Start streaming — reset scroll lock so new output auto-scrolls
self._user_scrolled_up = False
self._streaming_text = ""
# The streaming row will be replaced when done, it doesn't need deletion/export
self._streaming_row = self._add_msg_row("assistant", "▌")
self._streaming_row.set_finished(False)
self._set_stop_mode()
self._streaming_badge.show()
self._display_queue = self.agent.put_task(f"{FILE_HINT}\n\n{full_prompt}", source="user")
self._poll_timer.start(40)
def _handle_command(self, cmd: str):
parts = cmd.split()
op = parts[0].lower() if parts else ""
if op == "/help":
self._add_system_notice(HELP_TEXT)
elif op == "/stop":
self._do_stop()
self._add_system_notice("⏹️ 已停止")
elif op == "/status":
llm = self._model_name()
state = "🔴 运行中" if self.agent.is_running else "🟢 空闲"
self._add_system_notice(f"状态: {state}\nLLM: [{self.agent.llm_no}] {llm}")
elif op == "/llm":
if not self.agent.llmclient:
self._add_system_notice("❌ 当前没有可用的 LLM 配置")
elif len(parts) > 1:
try:
idx = int(parts[1])
self._do_switch_to(idx)
except Exception:
self._add_system_notice(f"用法: /llm <0-{len(self.agent.llmclients) - 1}>")
else:
lines = [f"{'→' if i == self.agent.llm_no else ' '} [{i}] {getattr(c, 'name', type(c.backend).__name__ + '/' + c.backend.model)}"
for i, c in enumerate(self.agent.llmclients)]
self._add_system_notice("LLMs:\n" + "\n".join(lines))
elif op == "/restore":
restored_info, err = format_restore()
if err:
self._add_system_notice(err)
else:
restored, fname, count = restored_info
self.agent.abort()
self.agent.history.extend(restored)
self._add_system_notice(f"✅ 已恢复 {count} 轮对话\n来源: {fname}")
elif op == "/new":
self._do_clear()
self._add_system_notice("✅ 已开启新对话")
else:
self._add_system_notice(f"未知命令: {cmd}\n{HELP_TEXT}")
def _poll_queue(self):
if not self._display_queue:
return
try:
while True:
item = self._display_queue.get_nowait()
if not isinstance(item, dict) or ("next" not in item and "done" not in item):
print(f"[Queue] 跳过异常项: {item}")
continue
if "next" in item:
self._streaming_text = item["next"]
if self._streaming_row:
self._streaming_row.set_text(self._streaming_text + " ▌")
self._update_token_usage()
self._scroll_bottom()
if "done" in item:
final = item["done"]
from datetime import datetime
now_iso = datetime.now().isoformat()
# Remove the temporary streaming row
if self._streaming_row:
# Find its position in the layout to replace it
idx = self._msg_layout.indexOf(self._streaming_row)
self._streaming_row.deleteLater()
self._streaming_row = None
# Add the final message with proper buttons
assist_idx = len(self._messages)
self._messages.append({"role": "assistant", "content": final, "created_at": now_iso})
# Insert at the same position where the streaming row was, or before the stretch
insert_pos = idx if idx >= 0 else self._msg_layout.count() - 1
row = _MsgRow(
final,
"assistant",
on_resend=self._regenerate_response,
on_delete=lambda idx=assist_idx: self._delete_message(idx),
on_rewrite=None,
created_at=now_iso
)
# 自动展开最后一个 LLM Running 块，方便用户直接看到结果
for _, title, _ in reversed(row._parse_foldable_blocks(final)):
if title is not None and title in row._folded_ids and 'LLM Running' in title:
row._folded_ids.remove(title)
row._label.setHtml(row._render_with_folds(final))
row._adjust_browser_height()
break
self._msg_layout.insertWidget(insert_pos, row)
self._poll_timer.stop()
self._set_send_mode()
self._streaming_badge.hide()
self.last_reply_time = time.time()
self._update_token_usage()
self._scroll_bottom()
self._auto_save()
break
except _queue.Empty:
pass
def _add_msg_row(self, role: str, text: str, created_at: str = None, on_delete=None, on_rewrite=None) -> _MsgRow:
row = _MsgRow(
text,
role,
on_resend=self._regenerate_response if role != "user" else None,
on_delete=on_delete,
on_rewrite=on_rewrite,
created_at=created_at
)
self._msg_layout.insertWidget(self._msg_layout.count() - 1, row)
self._scroll_bottom()
return row
def _regenerate_response(self):
"""Resend the last user message to regenerate the assistant response."""
if self._is_streaming:
return
for msg in reversed(self._messages):
if msg["role"] == "user":
self._input.setPlainText(msg["content"])
self._handle_send()
break
def _delete_message(self, index: int):
"""Delete the message at the given index."""
if index < 0 or index >= len(self._messages):
return
# Remove from data
self._messages.pop(index)
# Rebuild all rows to ensure on_delete indices are correct
self._rebuild_messages()
# Update
self._update_token_usage()
self._auto_save()
def _rewrite_message(self, index: int):
"""Rewrite the user message at the given index."""
if index < 0 or index >= len(self._messages):
return
if self._messages[index]["role"] != "user":
return
# Get the content and fill it into the input
content = self._messages[index]["content"]
self._input.setPlainText(content)
# Remove this message and everything after it
self._messages = self._messages[:index]
# Rebuild UI
self._rebuild_messages()
self._update_token_usage()
self._auto_save()
def _on_scroll(self, value):
sb = self._scroll.verticalScrollBar()
self._user_scrolled_up = value < sb.maximum() - 30
self._update_nav_visibility()
def _update_nav_visibility(self):
sb = self._scroll.verticalScrollBar()
max_val = sb.maximum()
vp_h = self._scroll.viewport().height()
total_h = max_val + vp_h
show_nav = max_val > 0 and total_h >= vp_h * 1.5
if show_nav:
self._nav_widget.show()
self._nav_up.setEnabled(sb.value() > 2)
self._nav_down.setEnabled(max_val > 0 and sb.value() < max_val - 2)
else:
self._nav_widget.hide()
def _scroll_to_top(self):
self._user_scrolled_up = True
self._scroll.verticalScrollBar().setValue(0)
def _scroll_to_bottom(self):
self._user_scrolled_up = False
QTimer.singleShot(60, lambda: self._scroll.verticalScrollBar().setValue(
self._scroll.verticalScrollBar().maximum()
))
def _scroll_bottom(self):
if self._user_scrolled_up:
return
QTimer.singleShot(60, lambda: self._scroll.verticalScrollBar().setValue(
self._scroll.verticalScrollBar().maximum()
))
# ── inject (autonomous mode) ───────────────────────────────────────────────
def inject_message(self, text: str):
"""Programmatically send a message (called by idle monitor)."""
self._input.setPlainText(text)
self._handle_send()
# ── history ────────────────────────────────────────────────────────────────
def _refresh_history(self):
self._history = _load_history()
self._hist_list.clear()
for s in reversed(self._history[-20:]):
n = len(s.get("messages", []))
item = QListWidgetItem(f" {s.get('title','未命名')} ({n} 条)")
item.setData(Qt.UserRole, s)
self._hist_list.addItem(item)
def _restore_selected(self, item=None):
item = item or self._hist_list.currentItem()
if not item:
return
s = item.data(Qt.UserRole)
if s:
self._session = s.copy()
self._messages = s.get("messages", []).copy()
self._rebuild_messages()
self._switch_tab(0)
self._update_token_usage()
search_text = self._search_input.text().strip()
if search_text:
QTimer.singleShot(50, lambda: self._search_current_chat(search_text))
def _delete_selected(self):
item = self._hist_list.currentItem()
if not item:
return
s = item.data(Qt.UserRole)
if s:
self._history = [h for h in self._history if h.get("id") != s.get("id")]
_save_history(self._history)
self._refresh_history()
def _rebuild_messages(self):
while self._msg_layout.count() > 1:
it = self._msg_layout.takeAt(0)
if it.widget():
it.widget().deleteLater()
for i, m in enumerate(self._messages):
rewrite_cb = (lambda idx=i: self._rewrite_message(idx)) if m["role"] == "user" else None
self._add_msg_row(
m["role"],
m["content"],
created_at=m.get("created_at"),
on_delete=lambda idx=i: self._delete_message(idx),
on_rewrite=rewrite_cb
)
self._update_token_usage()
def _update_token_usage(self):
in_chars = sum(len(m.get("content", "")) for m in self._messages if m.get("role") == "user")
out_chars = sum(len(m.get("content", "")) for m in self._messages if m.get("role") == "assistant")
if getattr(self, "_is_streaming", False) and getattr(self, "_streaming_text", ""):
out_chars += len(self._streaming_text)
in_tokens = int(in_chars / 2.5)
out_tokens = int(out_chars / 2.5)
if in_tokens == 0 and out_tokens == 0:
self._token_lbl.setText("")
else:
self._token_lbl.setText(f"| 会话上下文消耗: 入 {in_tokens} 出 {out_tokens} tokens")
# ── SOP ────────────────────────────────────────────────────────────────────
def _refresh_sop(self):
self._sop_list.clear()
file_icon = _svg_icon("sop_file_item", _SVG_FILE, C["muted"])
for path in sorted(glob.glob(os.path.join(os.path.dirname(os.path.dirname(__file__)), "memory", "*.md"))):
name = os.path.basename(path)
size = os.path.getsize(path)
it = QListWidgetItem(name)
it.setIcon(file_icon)
it.setData(Qt.UserRole, path)
it.setToolTip(f"{size:,} 字节")
self._sop_list.addItem(it)
def _load_sop(self, item):
if not item:
return
path = item.data(Qt.UserRole)
try:
with open(path, "r", encoding="utf-8") as f:
self._sop_viewer.setHtml(_md_to_html(f.read()))
except Exception as e:
self._sop_viewer.setPlainText(f"读取失败: {e}")
# ── settings actions ───────────────────────────────────────────────────────
def _model_name(self) -> str:
if self.agent.llmclient is None:
return "未配置"
try:
return self.agent.get_llm_name()
except Exception:
return "未知"
def _add_system_notice(self, text: str):
"""Insert a small centered notice label (not tracked as a message)."""
lbl = QLabel(text)
lbl.setWordWrap(True)
lbl.setAlignment(Qt.AlignCenter)
lbl.setStyleSheet(
"QLabel { background: transparent; color: #71717a;"
" border: none; padding: 6px 20px; font-size: 12px; }"
)
self._msg_layout.insertWidget(self._msg_layout.count() - 1, lbl)
self._scroll_bottom()
def _do_stop(self):
self.agent.abort()
self._poll_timer.stop()
self._set_send_mode()
self._streaming_badge.hide()
if self._streaming_row:
self._streaming_row.set_text(self._streaming_text or "（已停止）")
self._streaming_row.set_finished(True)
self._streaming_row = None
self._update_token_usage()
def _do_reset_prompt(self):
if self.agent.llmclient and hasattr(self.agent.llmclient, 'last_tools'):
self.agent.llmclient.last_tools = ""
def _auto_save(self):
if not self._messages:
return
if self._session.get("title") == "新对话":
first_user = next(
(m["content"] for m in self._messages if m["role"] == "user"), ""
)
if first_user:
self._session["title"] = first_user[:30].replace("\n", " ")
self._do_save()
def _do_save(self):
if not self._messages:
return
self._session["messages"] = self._messages.copy()
self._session["updatedAt"] = datetime.now().isoformat()
self._history = _load_history()
for i, s in enumerate(self._history):
if s.get("id") == self._session["id"]:
self._history[i] = self._session.copy()
break
else:
self._history.append(self._session.copy())
_save_history(self._history)
def _do_clear(self):
self._messages.clear()
self._session = {"id": _make_session_id(), "title": "新对话", "messages": []}
self._rebuild_messages()
self._switch_tab(0)
self._update_token_usage()
def _new_session(self):
if self._messages:
self._do_save()
self._do_clear()
def _do_toggle_auto(self):
self.autonomous_enabled = not self.autonomous_enabled
self._auto_btn.setChecked(self.autonomous_enabled)
lbl = "暂停自主行动" if self.autonomous_enabled else "开启自主行动 (idle > 30 min 自动触发)"
self._auto_btn.setText(lbl)
def _do_trigger_auto(self):
self.inject_message(
"[AUTO]🤖 用户触发了自主行动，请阅读自动化sop，选择并执行一项有价值的任务。"
)
# ── helpers ────────────────────────────────────────────────────────────────
@staticmethod
def _small_btn_style(color: str) -> str:
return (
f"QPushButton {{ background: {color}; color: white; border: none;"
f" border-radius: 7px; padding: 4px 12px; font-size: 12px; font-weight: 600; }}"
f"QPushButton:hover {{ opacity: 0.85; }}"
)
# ══════════════════════════════════════════════════════════════════════
# Entry Point
# ══════════════════════════════════════════════════════════════════════
def main():
# High-DPI support
QApplication.setHighDpiScaleFactorRoundingPolicy(
Qt.HighDpiScaleFactorRoundingPolicy.PassThrough
)
app = QApplication(sys.argv)
app.setQuitOnLastWindowClosed(False)
app.setApplicationName("GenericAgent")
# Font
font = QFont()
# Keep English glyphs in Arial; Chinese falls back to Microsoft YaHei.
try:
font.setFamilies(["Arial", "Microsoft YaHei"])
except Exception:
font.setFamily("Microsoft YaHei")
font.setPointSize(10)
app.setFont(font)
# ── Agent initialisation ──────────────────────────────
agent = GeneraticAgent()
if agent.llmclient is None:
QMessageBox.critical(
None,
"未配置 LLM",
"未在 mykey.py 中发现任何可用的 LLM 接口配置，\n程序将在无 LLM 模式下运行。",
)
else:
threading.Thread(target=agent.run, daemon=True).start()
# ── Windows ───────────────────────────────────────────
panel = ChatPanel(agent)
button = FloatingButton(panel)
button.show()
# Position panel next to button and show it on first launch
button._position_panel()
panel.show()
scr = QApplication.primaryScreen().availableGeometry()
print(f"[GenericAgent] 启动成功")
print(f" 屏幕分辨率: {scr.width()}x{scr.height()}")
print(f" 悬浮按钮: ({button.x()}, {button.y()})")
print(f" 聊天面板: ({panel.x()}, {panel.y()})")
print(f" 关闭面板后可点击右下角发光按钮重新打开")
# ── Idle monitor (autonomous mode) ────────────────────
_last_trigger = 0.0
def idle_check():
nonlocal _last_trigger
if not panel.autonomous_enabled:
return
now = time.time()
if now - _last_trigger < AUTO_COOLDOWN:
return
idle = now - panel.last_reply_time
if idle > AUTO_IDLE_THRESHOLD:
_last_trigger = now
panel.inject_message(
"[AUTO]🤖 用户已经离开超过30分钟，作为自主智能体，请阅读自动化sop，执行自动任务。"
)
idle_timer = QTimer()
idle_timer.timeout.connect(idle_check)
idle_timer.start(5000) # check every 5 s
sys.exit(app.exec())
if __name__ == "__main__":
main()