shuiyu67
0aaca9c63d
- 版本号统一更新至 v8.0.2（README 中英 badge / about.txt / 中央 VERSION / TUI banner / LSP·MCP clientInfo.version） - CHANGELOG 新增 v8.0.2 条目，记录两个 hotfix： -c81914a加载动画在非 TTY 环境不再污染输出流（isatty 守卫） -1c52482非 TTY 环境保留一句随机情话保人味（过滤 URL 行）
135 lines
5.3 KiB
Python
135 lines
5.3 KiB
Python
"""
|
|
v8.0 Rich 面板式 TUI + 可截取快照出口
|
|
======================================
|
|
- 通过 cli.tui_output_callback = RichTUI.push 接收逐行输出，渲染为 rich 面板
|
|
- Console(record=True) 快照出口: export_text / export_svg / export_html
|
|
→ AI 开发 TUI 时能把渲染结果读回自验，补上“看不见渲染”的反馈闭环
|
|
|
|
设计取舍（与 7.0 重构前 textual TUI 的区别）:
|
|
- 不用 textual 的异步全屏 App，避免与同步 input() 主循环冲突；
|
|
- 采用 rich 顺序面板流（一问一答往下渲染），天然支持 record=True 快照；
|
|
- 这正是“让 AI 能看见自己写的 TUI”的关键能力。
|
|
"""
|
|
import os
|
|
from rich.console import Console
|
|
from rich.panel import Panel
|
|
from rich.text import Text
|
|
from rich.box import ROUNDED
|
|
|
|
SNAP_DIR = os.path.join(
|
|
os.path.dirname(os.path.dirname(os.path.abspath(__file__))), ".tui_snapshots"
|
|
)
|
|
|
|
|
|
class RichTUI:
|
|
"""rich 面板式 TUI：聊天面板 + 状态栏 + 快照出口。"""
|
|
|
|
def __init__(self, cli):
|
|
self.cli = cli
|
|
# record=True 是关键：渲染过程被记录，可随时导出文本/SVG/HTML 自验
|
|
self.console = Console(record=True, width=100)
|
|
self.version = "8.0.2"
|
|
self.engine = ""
|
|
self.model = ""
|
|
self.mode = "普通"
|
|
self.used_tokens = None
|
|
self.context_window = None
|
|
self.msg_count = 0
|
|
self._log = []
|
|
|
|
# ── 状态设置（由 cli 在切换引擎/压缩后刷新）──
|
|
def set_status(self, engine=None, model=None, mode=None,
|
|
used_tokens=None, context_window=None, msg_count=None):
|
|
if engine is not None:
|
|
self.engine = engine
|
|
if model is not None:
|
|
self.model = model
|
|
if mode is not None:
|
|
self.mode = mode
|
|
if used_tokens is not None:
|
|
self.used_tokens = used_tokens
|
|
if context_window is not None:
|
|
self.context_window = context_window
|
|
if msg_count is not None:
|
|
self.msg_count = msg_count
|
|
|
|
def banner(self):
|
|
self.console.print(Panel(
|
|
Text.from_markup(f"[bold cyan]小狸 Pro-CLI[/] [yellow]v{self.version}[/] · Rich TUI"),
|
|
subtitle="[dim]自然语言对话 · /exit 返回 CLI · /snapshot 导出快照[/]",
|
|
box=ROUNDED, border_style="cyan"))
|
|
self.console.print()
|
|
|
|
def _bar(self):
|
|
if self.used_tokens and self.context_window:
|
|
ratio = min(1.0, self.used_tokens / self.context_window)
|
|
filled = int(ratio * 10)
|
|
bar = "▓" * filled + "░" * (10 - filled)
|
|
tok = f"{self.used_tokens}/{self.context_window} {bar}"
|
|
elif self.used_tokens:
|
|
tok = f"{self.used_tokens} tokens"
|
|
else:
|
|
tok = "-"
|
|
return (f"[dim]引擎 {self.engine or '-'} · 模型 {self.model or '-'} · "
|
|
f"模式 {self.mode} · {tok} · 消息 {self.msg_count}[/]")
|
|
|
|
# ── 回调入口：DisplayMixin._output 把每行输出投到这里 ──
|
|
def push(self, message):
|
|
m = (message or "").strip("\n")
|
|
if not m:
|
|
return
|
|
self._log.append(m)
|
|
if len(self._log) > 300:
|
|
self._log = self._log[-300:]
|
|
# 原始 message 已由 DisplayMixin 产出合法 rich markup，直接打印；
|
|
# 个别工具结果含裸方括号可能破坏 markup，失败则降级为纯文本。
|
|
try:
|
|
self.console.print(m)
|
|
except Exception:
|
|
self.console.print(m, markup=False)
|
|
|
|
# ── 快照出口（AI 自验渲染的关键能力）──
|
|
def snapshot_text(self):
|
|
return self.console.export_text(clear=False)
|
|
|
|
def snapshot_svg(self, title="小狸 Pro-CLI v8.0 Rich TUI"):
|
|
return self.console.export_svg(title=title, clear=False)
|
|
|
|
def snapshot_html(self, title="小狸 Pro-CLI v8.0 Rich TUI"):
|
|
# rich 新版 export_html 不接受 title 参数，标题作为注释写入
|
|
html = self.console.export_html(clear=False)
|
|
return f"<!-- {title} -->\n{html}"
|
|
|
|
def save_snapshot(self, stem):
|
|
os.makedirs(os.path.dirname(stem), exist_ok=True)
|
|
with open(stem + ".txt", "w", encoding="utf-8") as f:
|
|
f.write(self.snapshot_text())
|
|
with open(stem + ".svg", "w", encoding="utf-8") as f:
|
|
f.write(self.snapshot_svg())
|
|
with open(stem + ".html", "w", encoding="utf-8") as f:
|
|
f.write(self.snapshot_html())
|
|
return stem + ".txt", stem + ".svg", stem + ".html"
|
|
|
|
# ── 主循环 ──
|
|
def run(self):
|
|
self.banner()
|
|
try:
|
|
import readline # 可选：输入历史补全
|
|
except Exception:
|
|
pass
|
|
while True:
|
|
try:
|
|
ui = input("> ").strip()
|
|
except (EOFError, KeyboardInterrupt):
|
|
break
|
|
if ui in ("/exit", "/quit"):
|
|
break
|
|
if ui == "/snapshot":
|
|
paths = self.save_snapshot(os.path.join(SNAP_DIR, "live"))
|
|
print(f"[快照已保存] {paths[2]}")
|
|
continue
|
|
if ui:
|
|
self.cli.process_conversation(ui)
|
|
self.msg_count = len(self.cli.shared_conversation_history)
|
|
self.console.print(Text.from_markup(self._bar()))
|