pqAutomationApp/app/views/panels/ai_image_panel.py

"""AI 图片对话面板：输入提示 → 生成 → 显示 → 列表切换/保存/删除。"""

from __future__ import annotations

import os
import sys
import threading
import time
import logging
import tkinter as tk
from tkinter import filedialog, messagebox, simpledialog


import ttkbootstrap as ttk
from PIL import Image, ImageTk

from app.services import ai_image as _svc

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from pqAutomationApp import PQAutomationApp


logger = logging.getLogger(__name__)


def _theme_colors():
    style = ttk.Style()
    colors = style.colors
    return {
        "bg": colors.bg,
        "fg": colors.fg,
        "muted": colors.secondary,
        "input_bg": colors.inputbg,
        "input_fg": colors.inputfg,
        "select_bg": colors.selectbg,
        "select_fg": colors.selectfg,
        "border": colors.border,
    }


def _apply_ai_image_list_style(self: "PQAutomationApp"):
    """刷新 AI 图片列表控件样式，尽量贴合当前主题色板。"""
    palette = _theme_colors()
    style = ttk.Style()
    style.configure(
        "AIImage.Treeview",
        background=palette["input_bg"],
        fieldbackground=palette["input_bg"],
        foreground=palette["input_fg"],
        bordercolor=palette["border"],
        lightcolor=palette["border"],
        darkcolor=palette["border"],
        rowheight=24,
        font=("微软雅黑", 9),
    )
    style.configure(
        "AIImage.Treeview.Heading",
        background=palette["bg"],
        foreground=palette["muted"],
        bordercolor=palette["border"],
        font=("微软雅黑", 9, "bold"),
    )
    style.map(
        "AIImage.Treeview",
        background=[("selected", palette["select_bg"])],
        foreground=[("selected", palette["select_fg"])],
    )


def _hide_tree_tooltip(self: "PQAutomationApp"):
    tip = getattr(self, "_ai_image_tooltip", None)
    if tip is None:
        return
    try:
        tip.withdraw()
    except Exception:
        pass
    self._ai_image_tooltip_item = ""


def _show_tree_tooltip(self: "PQAutomationApp", text: str, x_root: int, y_root: int, item_id: str):
    if not text:
        _hide_tree_tooltip(self)
        return
    tip = getattr(self, "_ai_image_tooltip", None)
    if tip is None:
        tip = tk.Toplevel(self.root)
        tip.withdraw()
        tip.overrideredirect(True)
        tip.transient(self.root)
        label = tk.Label(
            tip,
            text="",
            justify=tk.LEFT,
            anchor=tk.W,
            bg="#ffffff",
            fg="#1f2937",
            relief=tk.SOLID,
            bd=1,
            padx=8,
            pady=6,
            font=("微软雅黑", 9),
            wraplength=520,
        )
        label.pack(fill=tk.BOTH, expand=True)
        self._ai_image_tooltip = tip
        self._ai_image_tooltip_label = label
    else:
        label = getattr(self, "_ai_image_tooltip_label", None)
        if label is None:
            return

    self._ai_image_tooltip_item = item_id
    label.configure(text=text)
    tip.geometry(f"+{x_root + 14}+{y_root + 18}")
    tip.deiconify()
    tip.lift()


def _on_tree_motion(self: "PQAutomationApp", event):
    tree = self.ai_image_tree
    item_id = tree.identify_row(event.y)
    col = tree.identify_column(event.x)
    region = tree.identify("region", event.x, event.y)
    if not item_id or col != "#0" or region not in {"tree", "cell"}:
        _hide_tree_tooltip(self)
        return

    node_map = getattr(self, "_ai_image_node_map", None) or {}
    ridx = node_map.get(item_id)
    if ridx is None or ridx < 0 or ridx >= len(self.ai_image_records):
        _hide_tree_tooltip(self)
        return

    rec = self.ai_image_records[ridx]
    full_text = (rec.display_name or "").strip()
    if not full_text:
        _hide_tree_tooltip(self)
        return

    # 悬浮到任意图片标题项时都显示完整描述，避免因截断判定误差导致无响应。
    if getattr(self, "_ai_image_tooltip_item", "") == item_id:
        tip = getattr(self, "_ai_image_tooltip", None)
        if tip is not None:
            try:
                tip.geometry(f"+{event.x_root + 14}+{event.y_root + 18}")
            except Exception:
                pass
        return
    _show_tree_tooltip(self, full_text, event.x_root, event.y_root, item_id)





# ---------------- 面板创建 ----------------


def create_ai_image_panel(self: "PQAutomationApp"):
    """创建 AI 图片对话面板，并注册到面板管理。"""
    frame = ttk.Frame(self.content_frame)
    self.ai_image_frame = frame

    # 内部状态
    self.ai_image_records = []          # list[AIImageRecord]
    self.ai_image_current = None        # AIImageRecord | None
    self.ai_image_photo = None          # 防止 ImageTk.PhotoImage 被 GC
    self._ai_image_requesting = False
    self._ai_image_progress_job = None
    self._ai_image_progress_phase = 0
    self._ai_image_cancel_event = None
    self._ai_image_request_seq = 0
    self._ai_image_active_seq = 0
    self._ai_image_node_map = {}
    self._ai_image_session_node_map = {}
    self._ai_image_all_records = []
    self._ai_image_search_var = tk.StringVar(value="")
    self._ai_image_count_var = tk.StringVar(value="0 张图片")
    self._ai_image_reloading = False
    self._ai_image_select_guard = False
    self._ai_image_list_loaded = False
    self._ai_image_tooltip = None
    self._ai_image_tooltip_label = None
    self._ai_image_tooltip_item = ""

    container = ttk.Frame(frame, padding=10)
    container.pack(fill=tk.BOTH, expand=True)

    # 左列：图片列表
    # 使用 grid + 权重，让右侧预览区优先占据剩余空间。
    container.columnconfigure(0, weight=0)
    container.columnconfigure(1, weight=1)
    container.rowconfigure(0, weight=1)
    palette = _theme_colors()

    left = ttk.Frame(container, width=360)
    left.grid(row=0, column=0, sticky=tk.NS, padx=(0, 10))
    left.grid_propagate(False)

    title_row = ttk.Frame(left)
    title_row.pack(fill=tk.X, pady=(0, 4))
    ttk.Label(title_row, text="历史图片", font=("微软雅黑", 10, "bold")).pack(side=tk.LEFT)
    ttk.Label(
        title_row,
        textvariable=self._ai_image_count_var,
        foreground=palette["muted"],
        font=("微软雅黑", 9),
    ).pack(side=tk.RIGHT)

    search_row = ttk.Frame(left)
    search_row.pack(fill=tk.X, pady=(0, 6))
    ttk.Label(
        search_row,
        text="搜索",
        foreground=palette["muted"],
        font=("微软雅黑", 9),
    ).pack(side=tk.LEFT, padx=(0, 6))
    self.ai_image_search_entry = ttk.Entry(
        search_row,
        textvariable=self._ai_image_search_var,
        bootstyle="secondary",
    )
    self.ai_image_search_entry.pack(side=tk.LEFT, fill=tk.X, expand=True)
    self.ai_image_search_entry.bind("<KeyRelease>", lambda e: reload_ai_image_list(self, auto_select_first=False))
    ttk.Button(
        search_row,
        text="清空",
        width=6,
        bootstyle="secondary-outline",
        command=lambda: _clear_list_search(self),
    ).pack(side=tk.LEFT, padx=(6, 0))

    list_wrap = ttk.Frame(left, padding=2)
    list_wrap.pack(fill=tk.BOTH, expand=True)
    list_wrap.columnconfigure(0, weight=1)
    list_wrap.rowconfigure(0, weight=1)

    _apply_ai_image_list_style(self)
    scroll = ttk.Scrollbar(list_wrap, orient=tk.VERTICAL)
    self.ai_image_tree = ttk.Treeview(
        list_wrap,
        columns=("time",),
        show="tree headings",
        selectmode="browse",
        style="AIImage.Treeview",
        yscrollcommand=scroll.set,
    )
    self.ai_image_tree.heading("#0", text="标题")
    self.ai_image_tree.heading("time", text="时间")
    self.ai_image_tree.column("#0", width=210, minwidth=140, anchor=tk.W)
    self.ai_image_tree.column("time", width=105, minwidth=90, stretch=False, anchor=tk.W)
    scroll.config(command=self.ai_image_tree.yview)
    self.ai_image_tree.grid(row=0, column=0, sticky=tk.NSEW)
    scroll.grid(row=0, column=1, sticky=tk.NS)
    self.ai_image_tree.bind("<<TreeviewSelect>>", lambda e: _on_list_select(self))
    self.ai_image_tree.bind("<Double-1>", lambda e: _on_tree_double_click(self, e))
    self.ai_image_tree.bind("<Motion>", lambda e: _on_tree_motion(self, e))
    self.ai_image_tree.bind("<Leave>", lambda e: _hide_tree_tooltip(self))
    # 右键菜单：发送到 UCD / 重命名 / 另存为 / 删除
    # 索引: 0=发送, 1=sep, 2=重命名, 3=另存为, 4=删除
    self.ai_image_menu = tk.Menu(self.root, tearoff=0)
    self.ai_image_menu.add_command(
        label="发送图片",
        command=lambda: _send_to_ucd(self),
    )
    self.ai_image_menu.add_separator()
    self.ai_image_menu.add_command(
        label="重命名…",
        command=lambda: _rename_current(self),
    )
    self.ai_image_menu.add_command(
        label="另存为…",
        command=lambda: _save_current(self),
    )
    self.ai_image_menu.add_command(
        label="删除",
        command=lambda: _delete_current(self),
    )
    self.ai_image_tree.bind(
        "<Button-3>",
        lambda e: _show_list_context_menu(self, e),
    )

    btn_row = ttk.Frame(left)
    btn_row.pack(fill=tk.X, pady=(6, 0))
    self.ai_image_send_ucd_btn = ttk.Button(
        btn_row, text="发送", bootstyle="info-outline", width=8,
        command=lambda: _send_to_ucd(self),
    )
    self.ai_image_send_ucd_btn.pack(side=tk.LEFT, padx=(0, 4))
    ttk.Button(
        btn_row, text="保存", bootstyle="success-outline", width=8,
        command=lambda: _save_current(self),
    ).pack(side=tk.LEFT, padx=(0, 4))
    ttk.Button(
        btn_row, text="删除", bootstyle="danger-outline", width=8,
        command=lambda: _delete_current(self),
    ).pack(side=tk.LEFT, padx=(0, 4))
    ttk.Button(
        btn_row, text="刷新", bootstyle="secondary-outline", width=8,
        command=lambda: reload_ai_image_list(self),
    ).pack(side=tk.LEFT)

    # 右列：预览 + 输入
    right = ttk.Frame(container)
    right.grid(row=0, column=1, sticky=tk.NSEW)

    preview_frame = ttk.LabelFrame(right, text="图片预览", padding=6)
    preview_frame.pack(fill=tk.BOTH, expand=True)

    self.ai_image_canvas = tk.Canvas(
        preview_frame, bg=palette["bg"], highlightthickness=0
    )
    self.ai_image_canvas.pack(fill=tk.BOTH, expand=True)
    self.ai_image_canvas.bind("<Configure>", lambda e: _redraw_preview(self))

    meta_row = ttk.Frame(right)
    meta_row.pack(fill=tk.X, pady=(4, 4))
    self.ai_image_meta_var = tk.StringVar(value="未选择图片")
    ttk.Label(
        meta_row, textvariable=self.ai_image_meta_var,
        foreground=palette["muted"], font=("微软雅黑", 9),
    ).pack(side=tk.LEFT)

    # 输入区
    input_frame = ttk.LabelFrame(right, text="提示输入（Ctrl+Enter 发送）", padding=6)
    input_frame.pack(fill=tk.X, pady=(4, 0))

    self.ai_image_input = tk.Text(
        input_frame,
        height=3,
        wrap=tk.WORD,
        bg=palette["input_bg"],
        fg=palette["input_fg"],
        insertbackground=palette["input_fg"],
    )
    self.ai_image_input.pack(fill=tk.X, side=tk.TOP)
    self.ai_image_input.bind("<Control-Return>", lambda e: (_send_prompt(self), "break"))

    send_row = ttk.Frame(input_frame)
    send_row.pack(fill=tk.X, pady=(4, 0))
    self.ai_image_status_var = tk.StringVar(value="就绪")
    ttk.Label(
        send_row, textvariable=self.ai_image_status_var,
        foreground=palette["muted"], font=("微软雅黑", 9),
    ).pack(side=tk.LEFT)
    self.ai_image_progress = ttk.Progressbar(
        send_row,
        mode="indeterminate",
        length=120,
        bootstyle="info-striped",
    )
    self.ai_image_progress.pack(side=tk.LEFT, padx=(8, 0))
    self.ai_image_progress.pack_forget()
    self.ai_image_send_btn = ttk.Button(
        send_row, text="发送", bootstyle="primary", width=10,
        command=lambda: _send_prompt(self),
    )
    self.ai_image_send_btn.pack(side=tk.RIGHT)
    self.ai_image_new_session_btn = ttk.Button(
        send_row, text="新对话", bootstyle="secondary-outline", width=10,
        command=lambda: _start_new_session(self),
    )
    self.ai_image_new_session_btn.pack(side=tk.RIGHT, padx=(0, 6))
    self.ai_image_stop_btn = ttk.Button(
        send_row, text="停止", bootstyle="danger-outline", width=10,
        command=lambda: _stop_request(self),
    )
    self.ai_image_stop_btn.pack(side=tk.RIGHT, padx=(0, 6))
    self.ai_image_stop_btn.configure(state=tk.DISABLED)

    # 注册面板
    self.register_panel("ai_image", frame, None, "ai_image_visible")
    self.ai_image_visible = False
    self.ai_image_meta_var.set("首次打开 AI 图片面板时加载缓存")


def toggle_ai_image_panel(self: "PQAutomationApp"):
    """切换 AI 图片面板显隐。"""
    self.show_panel("ai_image")
    _apply_ai_image_list_style(self)
    if not getattr(self, "_ai_image_list_loaded", False):
        logger.info("[AIImagePanel] 首次显示面板，开始加载列表")
        reload_ai_image_list(self)
        self._ai_image_list_loaded = True


def _get_app_base_dir(self: "PQAutomationApp") -> str:
    """返回应用根目录（settings 的上一级）。"""
    if getattr(self, "config_file", None):
        return os.path.dirname(os.path.dirname(self.config_file))
    if getattr(sys, "frozen", False):
        return os.path.dirname(sys.executable)
    return os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))


# ---------------- 列表 / 选中 ----------------


def reload_ai_image_list(self: "PQAutomationApp", auto_select_first=True):
    """重新扫描缓存并刷新列表。

    按 ``session_id`` 分组：每个会话用一行不可选的分隔头（``── 会话 #N · 时间 ──``），
    其下列出该轮生成的所有图片。会话按"最近使用"倒序，组内按时间倒序。
    auto_select_first: 是否自动选中第一张图片（默认 True）。
    """
    t0 = time.monotonic()
    if getattr(self, "_ai_image_reloading", False):
        logger.debug("[AIImagePanel] 忽略重入 reload 请求")
        return
    self._ai_image_reloading = True
    try:
        _apply_ai_image_list_style(self)
        all_records = _svc.list_records(base_dir=_get_app_base_dir(self))
        self._ai_image_all_records = list(all_records)
        self.ai_image_tree.delete(*self.ai_image_tree.get_children())
        self._ai_image_node_map = {}
        self._ai_image_session_node_map = {}

        keyword = (self._ai_image_search_var.get() or "").strip().lower()
        records = []
        for rec in all_records:
            if not keyword:
                records.append(rec)
                continue
            hay = "\n".join([
                rec.display_name or "",
                rec.prompt or "",
                os.path.basename(rec.image_path),
            ]).lower()
            if keyword in hay:
                records.append(rec)

        self.ai_image_records = records
        sessions = _svc.group_records_by_session(self.ai_image_records)
        flat = []
        current_sid = _svc.get_session_id()
        for idx, sess in enumerate(sessions, start=1):
            sid = sess["session_id"]
            is_current = sid and sid == current_sid
            header = _format_session_header(idx, sess, is_current=is_current)
            parent_id = self.ai_image_tree.insert(
                "",
                tk.END,
                text=header,
                values=("",),
                open=True,
            )
            self._ai_image_session_node_map[parent_id] = sid
            for rec in sess["records"]:
                label = _format_list_label(rec)
                created = (rec.created_at or "").replace("T", " ")[:16]
                node_id = self.ai_image_tree.insert(
                    parent_id,
                    tk.END,
                    text=label,
                    values=(created,),
                )
                self._ai_image_node_map[node_id] = len(flat)
                flat.append(rec)
        # 替换为按显示顺序展平后的列表，便于其它逻辑（前一张/后一张等）
        self.ai_image_records = flat
        total_count = len(all_records)
        visible_count = len(self.ai_image_records)
        if keyword:
            self._ai_image_count_var.set(f"{visible_count}/{total_count} 张")
        else:
            self._ai_image_count_var.set(f"{visible_count} 张图片")
        if self.ai_image_records and auto_select_first:
            first_id = ""
            for parent in self.ai_image_tree.get_children(""):
                children = self.ai_image_tree.get_children(parent)
                if children:
                    first_id = children[0]
                    break
            if first_id:
                _set_tree_selection(self, first_id)
        else:
            self.ai_image_current = None
            self.ai_image_photo = None
            self.ai_image_canvas.delete("all")
            self.ai_image_meta_var.set("暂无缓存图片" if not keyword else "当前筛选无结果")
        logger.info(
            "[AIImagePanel] reload 完成 total=%d visible=%d sessions=%d keyword=%r elapsed=%.3fs",
            total_count,
            visible_count,
            len(sessions),
            keyword,
            time.monotonic() - t0,
        )
        self._ai_image_list_loaded = True
    finally:
        self._ai_image_reloading = False


def _format_session_header(index: int, sess: dict, is_current: bool) -> str:
    started = (sess.get("started_at") or "").replace("T", " ")[:16]
    tag = "（当前）" if is_current else ""
    count = len(sess.get("records") or [])
    if sess.get("session_id"):
        return f"会话 #{index} · {started} · {count} 张 {tag}".strip()
    return f"未归类 · {started} · {count} 张"


def _format_list_label(rec: _svc.AIImageRecord) -> str:
    # 分辨率前缀：优先从 extra["size"] 取，其次从文件名猜
    size_tag = ""
    extra = rec.extra or {}
    if isinstance(extra, dict) and extra.get("size"):
        size_tag = f"[{extra['size']}] "

    name_line = rec.display_name.splitlines()[0] if rec.display_name else "(未命名)"
    # Treeview 标题列可变宽，给展示名预留较长空间
    max_name = 40 - len(size_tag)
    if max_name > 4 and len(name_line) > max_name:
        name_line = name_line[:max_name] + "…"
    return f"{size_tag}{name_line}"


def _clear_list_search(self: "PQAutomationApp"):
    if (self._ai_image_search_var.get() or "").strip() == "":
        return
    self._ai_image_search_var.set("")
    reload_ai_image_list(self, auto_select_first=False)
    _hide_tree_tooltip(self)


def _set_tree_selection(self: "PQAutomationApp", item_id: str):
    if not item_id:
        return
    _hide_tree_tooltip(self)
    try:
        current = self.ai_image_tree.selection()
        if current and current[0] == item_id:
            node_map = getattr(self, "_ai_image_node_map", None) or {}
            ridx = node_map.get(item_id)
            if ridx is not None and 0 <= ridx < len(self.ai_image_records):
                _select_record(self, self.ai_image_records[ridx])
            return
        self.ai_image_tree.selection_set(item_id)
        self.ai_image_tree.focus(item_id)
        self.ai_image_tree.see(item_id)
    except Exception:
        return
    node_map = getattr(self, "_ai_image_node_map", None) or {}
    ridx = node_map.get(item_id)
    if ridx is not None and 0 <= ridx < len(self.ai_image_records):
        _select_record(self, self.ai_image_records[ridx])


def _find_tree_item_by_record_id(self: "PQAutomationApp", record_id: str) -> str:
    if not record_id:
        return ""
    node_map = getattr(self, "_ai_image_node_map", None) or {}
    for item_id, ridx in node_map.items():
        if ridx is None or ridx >= len(self.ai_image_records):
            continue
        rec = self.ai_image_records[ridx]
        if rec.id == record_id:
            return item_id
    return ""


def _on_list_select(self: "PQAutomationApp"):
    if getattr(self, "_ai_image_reloading", False):
        return
    if getattr(self, "_ai_image_select_guard", False):
        logger.debug("[AIImagePanel] 忽略重入选择事件")
        return
    sel = self.ai_image_tree.selection()
    if not sel:
        return
    self._ai_image_select_guard = True
    try:
        item_id = sel[0]
        node_map = getattr(self, "_ai_image_node_map", None) or {}
        ridx = node_map.get(item_id)
        if ridx is None:
            session_id = _session_id_for_item(self, item_id)
            if session_id:
                logger.info("[AIImagePanel] 选中会话头 sid=%s", session_id[:8])
                _switch_to_session(self, session_id, show_message=False, refresh_list=False)
            return
        if 0 <= ridx < len(self.ai_image_records):
            rec = self.ai_image_records[ridx]
            if rec.session_id:
                _switch_to_session(
                    self,
                    rec.session_id,
                    show_message=False,
                    refresh_list=False,
                )
            _select_record(self, rec)
            logger.debug("[AIImagePanel] 选中图片 id=%s sid=%s", rec.id, (rec.session_id or "")[:8])
    finally:
        self._ai_image_select_guard = False


def _on_tree_double_click(self: "PQAutomationApp", event):
    """双击会话头时只切换会话并展开；双击记录保持默认行为。"""
    item_id = self.ai_image_tree.identify_row(event.y)
    if not item_id:
        return
    node_map = getattr(self, "_ai_image_node_map", None) or {}
    if item_id in node_map:
        return
    sid = _session_id_for_item(self, item_id)
    if sid:
        _switch_to_session(self, sid, show_message=False, refresh_list=False)


def _select_record(self: "PQAutomationApp", rec: _svc.AIImageRecord):
    self.ai_image_current = rec
    self.ai_image_meta_var.set(
        f"{os.path.basename(rec.image_path)} | {rec.created_at}"
    )
    _redraw_preview(self)


# ---------------- 预览绘制 ----------------


def _redraw_preview(self: "PQAutomationApp"):
    rec = getattr(self, "ai_image_current", None)
    canvas = self.ai_image_canvas
    palette = _theme_colors()
    canvas.delete("all")
    if rec is None or not os.path.isfile(rec.image_path):
        return
    cw = canvas.winfo_width() or 1
    ch = canvas.winfo_height() or 1
    try:
        img = Image.open(rec.image_path)
        img.load()
    except Exception as exc:
        canvas.create_text(cw // 2, ch // 2, text=f"加载失败: {exc}", fill=palette["select_bg"])
        return
    iw, ih = img.size
    scale = min(cw / iw, ch / ih, 1.0)
    nw, nh = max(1, int(iw * scale)), max(1, int(ih * scale))
    img_resized = img.resize((nw, nh), Image.LANCZOS)
    self.ai_image_photo = ImageTk.PhotoImage(img_resized)
    canvas.create_image(cw // 2, ch // 2, image=self.ai_image_photo, anchor="center")


# ---------------- 发送 / 保存 / 删除 ----------------


def _start_new_session(self: "PQAutomationApp"):
    """开启新的对话会话，后续生成将使用新的 session_id。"""
    if getattr(self, "_ai_image_requesting", False):
        messagebox.showinfo("提示", "请等待当前请求完成")
        return
    _svc.reset_session()
    self.ai_image_status_var.set("已开启新对话")
    # 新对话创建后不要自动选中历史图片，否则会立即把 session 切回旧会话。
    reload_ai_image_list(self, auto_select_first=False)
    try:
        self.ai_image_tree.selection_remove(self.ai_image_tree.selection())
    except Exception:
        pass
    self.ai_image_current = None
    self.ai_image_photo = None
    self.ai_image_canvas.delete("all")
    self.ai_image_meta_var.set("新对话已开启，等待生成图片")


def _session_id_for_item(self: "PQAutomationApp", item_id: str) -> str:
    session_map = getattr(self, "_ai_image_session_node_map", None) or {}
    parent = item_id
    if parent not in session_map:
        try:
            parent = self.ai_image_tree.parent(item_id)
        except Exception:
            parent = ""
    return (session_map.get(parent) or "") if parent else ""


def _switch_to_session(
    self: "PQAutomationApp",
    session_id: str,
    show_message: bool = True,
    target_record_id: str = "",
    refresh_list: bool = True,
):
    sid = (session_id or "").strip()
    if not sid:
        return
    if sid == _svc.get_session_id():
        return
    _svc.set_session_id(sid)
    logger.info(
        "[AIImagePanel] 切换会话 sid=%s refresh=%s target=%s",
        sid[:8],
        refresh_list,
        target_record_id[:8] if target_record_id else "",
    )
    if refresh_list:
        reload_ai_image_list(self)
    if target_record_id:
        item_id = _find_tree_item_by_record_id(self, target_record_id)
        if item_id:
            _set_tree_selection(self, item_id)
    self.ai_image_status_var.set("已切换到历史对话")
    if show_message:
        messagebox.showinfo("提示", "已切换到所选历史对话")


def _update_request_progress(self: "PQAutomationApp"):
    if not getattr(self, "_ai_image_requesting", False):
        self._ai_image_progress_job = None
        return
    self.ai_image_status_var.set("正在生成图片…")
    self._ai_image_progress_phase += 1
    self._ai_image_progress_job = self.root.after(900, lambda: _update_request_progress(self))


def _send_prompt(self: "PQAutomationApp"):
    if getattr(self, "_ai_image_requesting", False):
        return
    prompt = self.ai_image_input.get("1.0", tk.END).strip()
    if not prompt:
        messagebox.showinfo("提示", "请输入内容")
        return

    self._ai_image_request_seq += 1
    req_seq = self._ai_image_request_seq
    self._ai_image_active_seq = req_seq
    self._ai_image_cancel_event = threading.Event()
    _set_requesting(self, True)
    is_remote_url = _svc.is_remote_image_url(prompt)
    self.ai_image_status_var.set("下载中…" if is_remote_url else "后端处理中…")

    def _success(record):
        self.root.after(0, lambda: _on_request_done(self, record, None, req_seq))

    def _error(exc):
        self.root.after(0, lambda: _on_request_done(self, None, exc, req_seq))

    if is_remote_url:
        _svc.import_image_from_url_async(
            prompt,
            on_success=_success,
            on_error=_error,
            base_dir=_get_app_base_dir(self),
            cancel_event=self._ai_image_cancel_event,
        )
        return

    _svc.request_image_async(
        prompt,
        on_success=_success,
        on_error=_error,
        base_dir=_get_app_base_dir(self),
        cancel_event=self._ai_image_cancel_event,
    )


def _set_requesting(self: "PQAutomationApp", flag: bool):
    self._ai_image_requesting = flag
    try:
        self.ai_image_send_btn.configure(state=tk.DISABLED if flag else tk.NORMAL)
        self.ai_image_new_session_btn.configure(state=tk.DISABLED if flag else tk.NORMAL)
        self.ai_image_stop_btn.configure(state=tk.NORMAL if flag else tk.DISABLED)
    except Exception:
        pass
    if flag:
        self._ai_image_progress_phase = 0
        self.ai_image_progress.pack(side=tk.LEFT, padx=(8, 0))
        self.ai_image_progress.start(10)
        if self._ai_image_progress_job is not None:
            self.root.after_cancel(self._ai_image_progress_job)
        self._ai_image_progress_job = self.root.after(0, lambda: _update_request_progress(self))
    else:
        if self._ai_image_progress_job is not None:
            self.root.after_cancel(self._ai_image_progress_job)
            self._ai_image_progress_job = None
        try:
            self.ai_image_progress.stop()
            self.ai_image_progress.pack_forget()
        except Exception:
            pass


def _on_request_done(self: "PQAutomationApp", record, exc, req_seq):
    # 旧请求回调（例如用户已点击停止后）直接忽略
    if req_seq != getattr(self, "_ai_image_active_seq", 0):
        return
    self._ai_image_active_seq = 0
    self._ai_image_cancel_event = None
    _set_requesting(self, False)
    if exc is not None:
        self.ai_image_status_var.set(f"失败: {exc}")
        messagebox.showerror("生成失败", str(exc))
        return
    self.ai_image_status_var.set("完成")
    self.ai_image_input.delete("1.0", tk.END)
    reload_ai_image_list(self)
    if record is not None:
        logger.info("[AIImagePanel] 生成完成并入列 id=%s sid=%s", record.id, (record.session_id or "")[:8])
    if record is not None and self.ai_image_records:
        item_id = _find_tree_item_by_record_id(self, record.id)
        if item_id:
            _set_tree_selection(self, item_id)


def _stop_request(self: "PQAutomationApp"):
    """停止当前生成任务（协作取消：屏蔽后续回调并恢复 UI）。"""
    if not getattr(self, "_ai_image_requesting", False):
        return
    event = getattr(self, "_ai_image_cancel_event", None)
    if event is not None:
        event.set()
    self._ai_image_active_seq = 0
    _set_requesting(self, False)
    self.ai_image_status_var.set("已停止生成")


def _save_current(self: "PQAutomationApp"):
    rec = getattr(self, "ai_image_current", None)
    if rec is None:
        messagebox.showinfo("提示", "请先选择一张图片")
        return
    ext = os.path.splitext(rec.image_path)[1] or ".png"
    dest = filedialog.asksaveasfilename(
        title="另存为",
        defaultextension=ext,
        initialfile=os.path.basename(rec.image_path),
        filetypes=[("图片", "*.png;*.jpg;*.jpeg;*.bmp;*.webp"), ("所有文件", "*.*")],
    )
    if not dest:
        return
    try:
        _svc.export_record(rec, dest)
        messagebox.showinfo("成功", f"已保存到:\n{dest}")
    except Exception as exc:
        messagebox.showerror("保存失败", str(exc))


def _delete_current(self: "PQAutomationApp"):
    rec = getattr(self, "ai_image_current", None)
    if rec is None:
        messagebox.showinfo("提示", "请先选择一张图片")
        return
    if not messagebox.askyesno("确认删除", f"确定删除这张缓存图片吗？\n{os.path.basename(rec.image_path)}"):
        return
    _svc.delete_record(rec)
    reload_ai_image_list(self)


def _rename_current(self: "PQAutomationApp"):
    """弹窗让用户修改当前记录的展示标题（保存到侧车 ``title`` 字段，原始 prompt 不变）。"""
    rec = getattr(self, "ai_image_current", None)
    if rec is None:
        messagebox.showinfo("提示", "请先选择一张图片")
        return

    current = rec.title or rec.display_name
    new_name = simpledialog.askstring(
        "重命名",
        "修改显示标题（留空可恢复使用原始提示词）：",
        initialvalue=current,
        parent=self.root,
    )
    if new_name is None:   # 取消
        return
    new_name = new_name.strip()
    if new_name == (rec.title or ""):
        return

    if not _svc.update_record_title(rec, new_name):
        messagebox.showerror("保存失败", "无法更新元数据，请检查文件权限。")
        return

    target_id = rec.id
    reload_ai_image_list(self)
    item_id = _find_tree_item_by_record_id(self, target_id)
    if item_id:
        _set_tree_selection(self, item_id)


# ---------------- 发送到 UCD ----------------


def _show_list_context_menu(self: "PQAutomationApp", event):
    """在图片列表上显示右键菜单，并根据状态启用/禁用项。"""
    _hide_tree_tooltip(self)
    try:
        item_id = self.ai_image_tree.identify_row(event.y)
    except Exception:
        item_id = ""

    node_map = getattr(self, "_ai_image_node_map", None) or {}
    ridx = node_map.get(item_id)
    if ridx is not None and 0 <= ridx < len(self.ai_image_records):
        _set_tree_selection(self, item_id)
        _select_record(self, self.ai_image_records[ridx])
    elif item_id:
        sid = _session_id_for_item(self, item_id)
        if sid:
            _switch_to_session(self, sid, show_message=False, refresh_list=False)

    has_selection = self.ai_image_current is not None
    ucd = getattr(self, "ucd", None)
    can_send = (
        has_selection
        and ucd is not None
        and getattr(ucd, "status", False)
    )
    try:
        self.ai_image_menu.entryconfigure(
            0, state=("normal" if can_send else "disabled")
        )
        self.ai_image_menu.entryconfigure(
            2, state=("normal" if has_selection else "disabled")
        )
        self.ai_image_menu.entryconfigure(
            3, state=("normal" if has_selection else "disabled")
        )
        self.ai_image_menu.entryconfigure(
            4, state=("normal" if has_selection else "disabled")
        )
        self.ai_image_menu.tk_popup(event.x_root, event.y_root)
    finally:
        self.ai_image_menu.grab_release()


def _send_to_ucd(self: "PQAutomationApp"):
    """把当前选中的 AI 图片通过 UCD 发送到显示设备。"""
    rec = getattr(self, "ai_image_current", None)
    if rec is None:
        messagebox.showinfo("提示", "请先选择一张图片")
        return
    if not os.path.isfile(rec.image_path):
        messagebox.showerror("错误", f"图片文件不存在:\n{rec.image_path}")
        return
    ucd = getattr(self, "ucd", None)
    if ucd is None or not getattr(ucd, "status", False):
        messagebox.showwarning("警告", "请先连接 UCD323 设备")
        return

    image_path = rec.image_path
    send_path = image_path
    log = getattr(self, "log_gui", None)

    # 发送前检查分辨率：建议与当前 UCD 输出分辨率一致。
    try:
        target_w, target_h = self.signal_service.current_resolution()
    except Exception:
        target_w, target_h = 3840, 2160
    try:
        with Image.open(image_path) as _img:
            src_w, src_h = _img.size
    except Exception as exc:
        messagebox.showerror("错误", f"无法读取图片尺寸:\n{exc}")
        return

    if log is not None:
        log.log(
            f"UCD分辨率: {target_w}x{target_h} | 图片分辨率: {src_w}x{src_h}",
            level="info",
        )

    if (src_w, src_h) != (target_w, target_h):
        if not messagebox.askyesno(
            "分辨率不匹配",
            (
                f"当前 UCD 分辨率: {target_w}x{target_h}\n"
                f"图片分辨率: {src_w}x{src_h}\n\n"
                "是否自动缩放后再发送？"
            ),
        ):
            if log is not None:
                log.log("用户取消发送：图片分辨率与 UCD 不一致", level="warning")
            return

        try:
            send_path = _build_ucd_resized_image(image_path, target_w, target_h)
            if log is not None:
                log.log(
                    f"已生成匹配分辨率副本: {os.path.basename(send_path)}",
                    level="info",
                )
        except Exception as exc:
            messagebox.showerror("错误", f"自动缩放失败:\n{exc}")
            return

    if log is not None:
        log.log(
            f"发送 AI 图片到 UCD: {os.path.basename(send_path)}",
            level="info",
        )
    if hasattr(self, "ai_image_status_var"):
        self.ai_image_status_var.set("发送中…")

    def _worker():
        err = None
        try:
            self.signal_service.send_image(send_path)
            ok = True
        except Exception as exc:
            ok = False
            err = str(exc)

        def _done():
            if ok:
                if log is not None:
                    log.log(
                        f"图片已发送到 UCD: {os.path.basename(send_path)}",
                        level="success",
                    )
                if hasattr(self, "ai_image_status_var"):
                    self.ai_image_status_var.set("已发送到 UCD")
            else:
                msg = f"UCD 发送失败: {err}" if err else "UCD 发送失败"
                if log is not None:
                    log.log(msg, level="error")
                if hasattr(self, "ai_image_status_var"):
                    self.ai_image_status_var.set("UCD 发送失败")

        try:
            self.root.after(0, _done)
        except Exception:
            pass

    threading.Thread(target=_worker, daemon=True).start()


def _build_ucd_resized_image(image_path: str, target_w: int, target_h: int) -> str:
    """生成与 UCD 分辨率匹配的临时副本（缓存目录内）。"""
    base_dir = os.path.dirname(image_path)
    base_name = os.path.splitext(os.path.basename(image_path))[0]
    out_name = f"{base_name}_{target_w}x{target_h}_ucd.png"
    out_path = os.path.join(base_dir, out_name)
    with Image.open(image_path) as img:
        resized = img.convert("RGB").resize((target_w, target_h), Image.LANCZOS)
        resized.save(out_path, format="PNG")
    return out_path


class AIImagePanelMixin:
    """由 tools/refactor_to_mixins.py 自动生成。
    把本模块的自由函数挂到 PQAutomationApp 上，便于 F12 跳转与类型推断。
    """
    create_ai_image_panel = create_ai_image_panel
    toggle_ai_image_panel = toggle_ai_image_panel
    _get_app_base_dir = _get_app_base_dir
    reload_ai_image_list = reload_ai_image_list
    _on_list_select = _on_list_select
    _select_record = _select_record
    _redraw_preview = _redraw_preview
    _start_new_session = _start_new_session
    _session_id_for_item = _session_id_for_item
    _switch_to_session = _switch_to_session
    _update_request_progress = _update_request_progress
    _send_prompt = _send_prompt
    _set_requesting = _set_requesting
    _on_request_done = _on_request_done
    _stop_request = _stop_request
    _save_current = _save_current
    _delete_current = _delete_current
    _rename_current = _rename_current
    _show_list_context_menu = _show_list_context_menu
    _send_to_ucd = _send_to_ucd