cutThenThink/examples/gui_integration_example.py

#!/usr/bin/env python3
"""
GUI 集成示例

演示如何在 GUI 中集成处理流程和结果展示
"""

import sys
import tkinter as tk
from tkinter import ttk, filedialog
from pathlib import Path
from typing import Optional

# 添加项目根目录到路径
project_root = Path(__file__).parent.parent
sys.path.insert(0, str(project_root))

from src.core.processor import ImageProcessor, ProcessCallback, ProcessResult, create_markdown_result
from src.config.settings import get_settings
from src.utils.logger import init_logger, get_logger
from src.gui.widgets import ResultWidget, MessageHandler, ProgressDialog


# 初始化日志
log_dir = project_root / "logs"
init_logger(log_dir=log_dir, level="INFO", colored_console=True)
logger = get_logger(__name__)


class GUIProcessCallback(ProcessCallback):
    """
    GUI 回调类
    """

    def __init__(self, progress_dialog: ProgressDialog, status_label: ttk.Label):
        super().__init__()
        self.progress_dialog = progress_dialog
        self.status_label = status_label

    def on_start(self, message: str = "开始处理"):
        self.status_label.config(text=message)

    def on_ocr_start(self, message: str = "开始 OCR 识别"):
        self.status_label.config(text=message)
        if self.progress_dialog and not self.progress_dialog.is_cancelled():
            self.progress_dialog.set_message(message)

    def on_ocr_complete(self, result):
        self.status_label.config(text=f"OCR 完成: {len(result.results)} 行")

    def on_ai_start(self, message: str = "开始 AI 分类"):
        self.status_label.config(text=message)
        if self.progress_dialog and not self.progress_dialog.is_cancelled():
            self.progress_dialog.set_message(message)

    def on_ai_complete(self, result):
        self.status_label.config(text=f"AI 完成: {result.category.value}")

    def on_save_start(self, message: str = "开始保存"):
        self.status_label.config(text=message)

    def on_save_complete(self, record_id: int):
        self.status_label.config(text=f"保存成功: ID={record_id}")

    def on_error(self, message: str, exception=None):
        self.status_label.config(text=f"错误: {message}")

    def on_complete(self, result: ProcessResult):
        if result.success:
            self.status_label.config(text=f"完成! 耗时 {result.process_time:.2f}s")
        else:
            self.status_label.config(text="处理失败")


class ProcessorDemoGUI:
    """
    处理器演示 GUI
    """

    def __init__(self):
        self.root = tk.Tk()
        self.root.title("CutThenThink - 处理流程整合演示")
        self.root.geometry("900x700")

        # 加载配置
        self.settings = get_settings()

        # 消息处理器
        self.message_handler = MessageHandler(self.root)

        # 处理器
        self.processor: Optional[ImageProcessor] = None

        # 创建 UI
        self._create_ui()

    def _create_ui(self):
        """创建 UI"""
        # 顶部工具栏
        toolbar = ttk.Frame(self.root, padding=10)
        toolbar.pack(side=tk.TOP, fill=tk.X)

        # 选择图片按钮
        self.select_button = ttk.Button(
            toolbar,
            text="📁 选择图片",
            command=self._on_select_image
        )
        self.select_button.pack(side=tk.LEFT, padx=5)

        # 处理按钮
        self.process_button = ttk.Button(
            toolbar,
            text="🚀 开始处理",
            command=self._on_process,
            state=tk.DISABLED
        )
        self.process_button.pack(side=tk.LEFT, padx=5)

        # 分隔符
        ttk.Separator(toolbar, orient=tk.VERTICAL).pack(side=tk.LEFT, fill=tk.Y, padx=10)

        # 当前图片标签
        ttk.Label(toolbar, text="当前图片:").pack(side=tk.LEFT, padx=5)
        self.image_label = ttk.Label(toolbar, text="未选择", foreground="gray")
        self.image_label.pack(side=tk.LEFT, padx=5)

        # 状态栏
        self.status_label = ttk.Label(
            self.root,
            text="就绪",
            relief=tk.SUNKEN,
            anchor=tk.W,
            padding=(5, 2)
        )
        self.status_label.pack(side=tk.BOTTOM, fill=tk.X)

        # 主内容区域（使用 PanedWindow 分割）
        paned = ttk.PanedWindow(self.root, orient=tk.HORIZONTAL)
        paned.pack(side=tk.TOP, fill=tk.BOTH, expand=True, padx=10, pady=10)

        # 左侧：结果展示
        left_frame = ttk.LabelFrame(paned, text="处理结果", padding=5)
        paned.add(left_frame, weight=2)

        self.result_widget = ResultWidget(left_frame)
        self.result_widget.pack(side=tk.TOP, fill=tk.BOTH, expand=True)

        # 右侧：日志
        right_frame = ttk.LabelFrame(paned, text="处理日志", padding=5)
        paned.add(right_frame, weight=1)

        self.log_text = tk.Text(
            right_frame,
            wrap=tk.WORD,
            font=("Consolas", 9),
            state=tk.DISABLED
        )
        self.log_text.pack(side=tk.LEFT, fill=tk.BOTH, expand=True)

        # 配置日志标签
        self.log_text.tag_config("INFO", foreground="#3498db")
        self.log_text.tag_config("WARNING", foreground="#f39c12")
        self.log_text.tag_config("ERROR", foreground="#e74c3c")
        self.log_text.tag_config("SUCCESS", foreground="#27ae60")

    def _log(self, level: str, message: str):
        """添加日志"""
        self.log_text.config(state=tk.NORMAL)
        self.log_text.insert(tk.END, f"{message}\n", level)
        self.log_text.see(tk.END)
        self.log_text.config(state=tk.DISABLED)
        self.root.update()

    def _on_select_image(self):
        """选择图片"""
        filename = filedialog.askopenfilename(
            title="选择图片",
            filetypes=[
                ("图片文件", "*.png *.jpg *.jpeg *.bmp *.gif"),
                ("所有文件", "*.*")
            ]
        )

        if filename:
            self.current_image_path = filename
            self.image_label.config(text=Path(filename).name)
            self.process_button.config(state=tk.NORMAL)
            self._log("INFO", f"已选择图片: {filename}")

    def _on_process(self):
        """处理图片"""
        if not hasattr(self, 'current_image_path'):
            return

        # 创建进度对话框
        progress_dialog = ProgressDialog(
            self.root,
            title="处理中",
            message="正在处理图片...",
            cancelable=True
        )

        # 创建回调
        callback = GUIProcessCallback(progress_dialog, self.status_label)

        # 创建处理器
        self.processor = ImageProcessor(
            ocr_config={
                'mode': self.settings.ocr.mode.value,
                'lang': self.settings.ocr.lang,
                'use_gpu': self.settings.ocr.use_gpu
            },
            ai_config=self.settings.ai,
            db_path=str(project_root / "data" / "cutnthink.db"),
            callback=callback
        )

        # 处理图片（在后台执行）
        self.root.after(100, lambda: self._do_process(progress_dialog))

    def _do_process(self, progress_dialog: ProgressDialog):
        """执行处理"""
        try:
            self._log("INFO", "开始处理...")

            # 处理
            result = self.processor.process_image(self.current_image_path)

            # 关闭进度对话框
            if not progress_dialog.is_cancelled():
                progress_dialog.close()

            # 显示结果
            if result.success:
                self.result_widget.set_result(result)
                self._log("SUCCESS", f"处理成功! 耗时 {result.process_time:.2f}s")
                self.message_handler.show_info(
                    "处理完成",
                    f"处理成功!\n耗时: {result.process_time:.2f}秒\n记录ID: {result.record_id}"
                )
            else:
                self._log("ERROR", f"处理失败: {result.error_message}")
                self.message_handler.show_error(
                    "处理失败",
                    result.error_message or "未知错误"
                )

        except Exception as e:
            progress_dialog.close()
            logger.error(f"处理失败: {e}", exc_info=True)
            self._log("ERROR", f"处理失败: {e}")
            self.message_handler.show_error("处理失败", str(e), exception=e)

    def run(self):
        """运行 GUI"""
        self.root.mainloop()


def main():
    """主函数"""
    app = ProcessorDemoGUI()
    app.run()


if __name__ == "__main__":
    main()