blackbird/BotOps/widgets/chat_view.py

"""聊天视图：用户/AI/系统消息 + AI 回复流式 Markdown 渲染。

服务端协议（aibox/aibox.py output_callback 原样透传 msg_type）：
  - chunk           增量片段（正式回答内容），直接 append 到 AI 消息
  - reasoning_start 思考开始信号，创建思考区域（在AI回答之前）
  - reasoning       思考内容增量片段，追加到思考区域
  - reasoning_end   思考结束信号
  - complete        收尾信号：chunk 流式是内容权威来源，complete 仅用于收尾；
                    仅当流式完全丢帧（未建 widget 且无内容）时，用 full_text 补建
"""
from textual.containers import VerticalScroll
from textual.widgets import Markdown, Static


class ChatView(VerticalScroll):
    """可滚动聊天区。AI 回复用单个 Markdown 部件增量 append，实现流式渲染。

    顺序保证：思考区域始终在 AI 正式回答之前。
    滚动策略：每条新消息挂载后调用 anchor() 锚定底部——之后内容增长
    （流式 append）容器会自动保持吸底；用户上翻时 Textual 自动脱锚，
    下一条新消息再重新锚定。
    """

    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self._ai_widget: Markdown | None = None
        self._reasoning_widget: Static | None = None
        self._ai_text = ""
        self._reasoning_text = ""
        self._ai_finalized = True  # 初始无活跃 AI 消息
        self._ai_prefab = False    # AI widget是预创建的空widget（可重排）

    @property
    def ai_text(self) -> str:
        return self._ai_text

    def _anchor_bottom(self):
        """锚定吸底（需在布局刷新后执行，否则滚到的是旧位置）。"""
        self.call_after_refresh(self.anchor)

    async def add_user(self, text: str):
        await self.mount(Static(f"👤 你: {text}", classes="msg-user", markup=False))
        self._anchor_bottom()

    async def add_system(self, text: str, css_class: str = "msg-system"):
        await self.mount(Static(text, classes=css_class, markup=False))
        self._anchor_bottom()

    async def start_ai(self):
        """预建一条 AI 消息（发送用户消息后即可调用，给出即时反馈）。
        
        注意：此时只创建空widget但不显示内容，如果后续先收到思考内容，
        会在start_reasoning()中重新排列顺序，确保思考在前。
        """
        if self._ai_widget is not None:
            return
        self._ai_text = ""
        self._ai_finalized = False
        self._ai_prefab = True
        self._ai_widget = Markdown("", classes="msg-ai")
        await self.mount(self._ai_widget)
        self._anchor_bottom()

    async def start_reasoning(self, hint: str = "🤔 思考中..."):
        """开始展示思考过程区域。
        
        如果AI widget已经预创建（空的），先移除它，等思考widget挂载后
        再重新挂载AI widget，保证思考在回答之前。
        """
        if self._reasoning_widget is not None:
            return
        
        # 如果AI widget已经存在且是空的预创建状态，重新排列顺序
        need_remount_ai = False
        ai_saved_text = self._ai_text
        if self._ai_widget is not None and self._ai_prefab:
            await self._ai_widget.remove()
            self._ai_widget = None
            need_remount_ai = True
        
        # 先挂载思考widget
        self._reasoning_text = ""
        self._reasoning_widget = Static(hint + "\n", classes="msg-reasoning", markup=False)
        await self.mount(self._reasoning_widget)
        
        # 如果之前有AI内容（理论上预创建时不会有，但保险处理），重新挂载AI widget
        if need_remount_ai:
            self._ai_widget = Markdown(ai_saved_text, classes="msg-ai")
            await self.mount(self._ai_widget)
        
        self._anchor_bottom()

    async def append_reasoning(self, delta: str):
        """追加思考内容增量。"""
        if not delta:
            return
        if self._reasoning_widget is None:
            await self.start_reasoning()
        self._reasoning_text += delta
        # 限制思考内容显示长度（最多显示最后 3000 字符，避免思考过长占满屏幕）
        display = self._reasoning_text
        if len(display) > 3000:
            display = "...(前略)\n" + display[-3000:]
        self._reasoning_widget.update(f"💭 思考中...\n{display}")
        self.call_after_refresh(self.scroll_end, animate=False)

    async def end_reasoning(self):
        """结束思考区域，更新提示文字。"""
        if self._reasoning_widget is not None and self._reasoning_text:
            # 思考结束后保持展示，更新标签为已完成
            display = self._reasoning_text
            if len(display) > 3000:
                display = "...(前略)\n" + display[-3000:]
            self._reasoning_widget.update(f"💭 思考过程（{len(self._reasoning_text)}字符）\n{display}\n{'─' * 30}")
        # 思考区域保留显示，不删除
        self._ai_prefab = False  # AI widget顺序已固定，不再重排

    async def append_ai(self, delta: str):
        """追加增量片段（服务端 chunk 为增量语义，直接拼接）。"""
        if not delta:
            return
        if self._ai_finalized:
            # 收尾后收到的延迟 chunk，丢弃以避免在上一条回复后追加脏数据
            return
        if self._ai_widget is None:
            # AI widget还没创建，先创建
            self._ai_text = ""
            self._ai_prefab = False
            self._ai_widget = Markdown("", classes="msg-ai")
            await self.mount(self._ai_widget)
        self._ai_prefab = False  # 开始有实际内容了
        self._ai_widget.append(delta)
        self._ai_text += delta
        # anchor 状态下容器会自动吸底；保险起见刷新后再滚一次
        self.call_after_refresh(self.scroll_end, animate=False)

    async def finish_ai(self, full_text: str = ""):
        """收尾一条 AI 消息。

        chunk 流式追加是内容权威来源，complete 只做收尾：
        - 已收尾则直接返回（防止 dialogue_ended + complete 双发导致重复挂载）
        - widget 存在且有流式内容：保持已渲染内容不变，仅清空引用
        - widget 存在但无内容（预建空框）且有 full_text：用 full_text 填充
        - widget 存在但无内容也无 full_text：移除空框
        - widget 不存在（流式丢帧/start_ai 未调用）且有 full_text：补建
        """
        if self._ai_finalized:
            return
        w = self._ai_widget
        self._ai_widget = None

        if w is not None:
            if self._ai_text:
                # 有流式内容，chunk 已完整渲染，直接保留，不做替换
                pass
            elif full_text:
                # 预建了空 widget 但没收到任何 chunk：用 full_text 填充
                w.update(full_text)
                self._ai_text = full_text
            else:
                # 预建后没有任何输出：移除空框
                await w.remove()
        else:
            # widget 不存在（start_ai 未调用），有 full_text 则补建
            if full_text:
                self._ai_text = full_text
                await self.mount(Markdown(full_text, classes="msg-ai"))
                self._anchor_bottom()

        self._ai_text = ""
        self._reasoning_widget = None
        self._reasoning_text = ""
        self._ai_finalized = True
        self._ai_prefab = False
        self.call_after_refresh(self.scroll_end, animate=False)

    async def reset(self):
        """清空聊天区（/clear）。"""
        self._ai_widget = None
        self._reasoning_widget = None
        self._ai_text = ""
        self._reasoning_text = ""
        self._ai_finalized = True
        self._ai_prefab = False
        await self.remove_children()