477 lines
24 KiB
Python
477 lines
24 KiB
Python
import tkinter as tk
|
||
from tkinter import ttk, filedialog, messagebox
|
||
import threading
|
||
import queue
|
||
import os
|
||
from typing import List
|
||
import concurrent.futures
|
||
import logging
|
||
from config_manager import ConfigManager
|
||
from api_services import ApiService, DEFAULT_LLM_PROMPT_TEMPLATE
|
||
|
||
# 配置日志记录器
|
||
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
|
||
|
||
class AboutDialog(tk.Toplevel):
|
||
"""“关于”对话框，展示应用信息，支持滚动查看。"""
|
||
def __init__(self, parent, config_manager: ConfigManager):
|
||
super().__init__(parent)
|
||
self.transient(parent)
|
||
self.title("关于 AI 作文批改助手")
|
||
# 设置一个适合滚动的默认窗口尺寸
|
||
self.geometry("450x400")
|
||
|
||
# 主框架，用于容纳文本和滚动条
|
||
main_frame = ttk.Frame(self)
|
||
main_frame.pack(fill=tk.BOTH, expand=True, padx=10, pady=10)
|
||
main_frame.grid_rowconfigure(0, weight=1)
|
||
main_frame.grid_columnconfigure(0, weight=1)
|
||
|
||
# 使用Text控件以支持长文本和滚动条
|
||
text_widget = tk.Text(main_frame, wrap="word", relief="flat", spacing1=5, spacing3=5)
|
||
text_widget.grid(row=0, column=0, sticky="nsew")
|
||
|
||
# 创建并关联垂直滚动条
|
||
scrollbar = ttk.Scrollbar(main_frame, orient="vertical", command=text_widget.yview)
|
||
scrollbar.grid(row=0, column=1, sticky="ns")
|
||
text_widget.config(yscrollcommand=scrollbar.set)
|
||
|
||
vlm_in = config_manager.get('UsageVlmInput', 0)
|
||
vlm_out = config_manager.get('UsageVlmOutput', 0)
|
||
llm_in = config_manager.get('UsageLlmInput', 0)
|
||
llm_out = config_manager.get('UsageLlmOutput', 0)
|
||
|
||
about_text = f"""
|
||
欢迎使用 AI 作文批改助手！这是一款专为教育工作者和学生设计的智能工具，利用前沿的人工智能技术，提供高效、精准、个性化的英文作文批改体验。
|
||
|
||
✨ 核心特色:
|
||
|
||
- **双AI引擎架构:** 采用创新的两步式处理流程。首先由专业的视觉语言模型(VLM)进行高精度手写文字识别(OCR)和专业的书写质量评估；然后由强大的大语言模型(LLM)结合识别文本、作文题目和书写评分，进行深度内容分析和专业批改。
|
||
|
||
- **极致灵活性:**
|
||
* **服务独立配置:** VLM和LLM支持完全独立的API服务地址、密钥和模型名称，轻松适配各种AI服务提供商（兼容OpenAI格式）
|
||
* **智能评分调节:** 书写质量"敏感度因子"可自由调整，适应不同年级和评分标准要求
|
||
* **Prompt完全开放:** 核心批改指令模板完全可自定义，支持调整评分标准、总分设置和反馈风格
|
||
|
||
- **高效并发处理:** 内置多线程并发引擎，支持批量处理任意数量的图片，大幅提升批改效率，最大并发数可配置
|
||
|
||
- **企业级安全保障:** 所有API密钥均采用军事级加密算法存储，确保您的账户信息安全
|
||
|
||
- **专业评分体系:** 针对高考英语作文场景设计，支持应用文(15分制)和读后续写(25分制)两种评分标准
|
||
|
||
📋 使用指南:
|
||
1. **首次设置:** 点击"设置"，配置VLM和LLM服务的URL、API密钥和模型
|
||
2. **输入题目:** 在主界面文本框中输入本次批改的作文题目
|
||
3. **选择图片:** 点击"选择图片"，可多选需要批改的作文图片
|
||
4. **开始批改:** 点击"开始批改"，程序自动进行并发处理
|
||
5. **查看报告:** 处理完成后，Markdown和HTML格式的详细批改报告将保存在原图片目录
|
||
|
||
🎯 输出格式:
|
||
- Markdown源文件（可编辑）
|
||
- HTML可视化报告（美观易读）
|
||
- 详细的语法错误分析
|
||
- 专业的写作建议
|
||
- 精准的分数评估
|
||
|
||
作者: Eric_Terminal
|
||
项目地址: https://github.com/Eric-Terminal/Pro_llm_correct
|
||
版本: 3.1
|
||
|
||
---
|
||
历史Token使用统计:
|
||
- VLM 输入Token: {vlm_in:,}
|
||
- VLM 输出Token: {vlm_out:,}
|
||
- LLM 输入Token: {llm_in:,}
|
||
- LLM 输出Token: {llm_out:,}
|
||
"""
|
||
|
||
text_widget.insert("1.0", about_text)
|
||
# 将文本设置为只读，防止用户修改
|
||
text_widget.config(state="disabled")
|
||
|
||
# 放置“关闭”按钮的框架
|
||
btn_frame = ttk.Frame(main_frame)
|
||
btn_frame.grid(row=1, column=0, columnspan=2, sticky="e", pady=(10, 0))
|
||
close_button = ttk.Button(btn_frame, text="关闭", command=self.destroy)
|
||
close_button.pack()
|
||
|
||
self.protocol("WM_DELETE_WINDOW", self.destroy)
|
||
self.grab_set()
|
||
self.wait_window(self)
|
||
|
||
|
||
class SettingsDialog(tk.Toplevel):
|
||
""""设置"对话框，允许用户配置VLM、LLM服务及其他应用参数。"""
|
||
def __init__(self, parent, config_manager: ConfigManager):
|
||
super().__init__(parent)
|
||
self.transient(parent)
|
||
self.title("设置")
|
||
self.result = None
|
||
self.config_manager = config_manager
|
||
|
||
# 为VLM和LLM服务分别创建Tkinter字符串变量
|
||
# 使用config_manager.get()方法获取解密后的值用于显示
|
||
self.vlm_url = tk.StringVar(value=config_manager.get("VlmUrl", "https://api.siliconflow.cn/v1"))
|
||
self.vlm_api_key = tk.StringVar(value=config_manager.get("VlmApiKey", ""))
|
||
self.vlm_model = tk.StringVar(value=config_manager.get("VlmModel", "Pro/THUDM/GLM-4.1V-9B-Thinking"))
|
||
self.llm_url = tk.StringVar(value=config_manager.get("LlmUrl", "https://api.siliconflow.cn/v1"))
|
||
self.llm_api_key = tk.StringVar(value=config_manager.get("LlmApiKey", ""))
|
||
self.llm_model = tk.StringVar(value=config_manager.get("LlmModel", "moonshotai/Kimi-K2-Instruct"))
|
||
self.sensitivity_factor = tk.StringVar(value=config_manager.get("SensitivityFactor", "1.5"))
|
||
self.max_workers = tk.StringVar(value=config_manager.get("MaxWorkers", "4"))
|
||
self.max_retries = tk.StringVar(value=config_manager.get("MaxRetries", "3"))
|
||
self.retry_delay = tk.StringVar(value=config_manager.get("RetryDelay", "5"))
|
||
self.save_markdown = tk.BooleanVar(value=config_manager.get("SaveMarkdown", True))
|
||
self.render_markdown = tk.BooleanVar(value=config_manager.get("RenderMarkdown", True))
|
||
|
||
# 智能加载Prompt模板：优先使用用户自定义模板，否则使用默认模板
|
||
user_template = config_manager.get("LlmPromptTemplate")
|
||
self.llm_prompt_template_str = user_template if user_template else DEFAULT_LLM_PROMPT_TEMPLATE
|
||
|
||
|
||
frame = ttk.Frame(self, padding="10")
|
||
frame.grid(row=0, column=0, sticky=(tk.W, tk.E, tk.N, tk.S))
|
||
frame.grid_columnconfigure(0, weight=1)
|
||
|
||
# VLM服务设置区域
|
||
vlm_frame = ttk.LabelFrame(frame, text="VLM (视觉模型) 设置", padding="10")
|
||
vlm_frame.grid(row=0, column=0, sticky=(tk.W, tk.E), pady=5)
|
||
vlm_frame.grid_columnconfigure(1, weight=1)
|
||
ttk.Label(vlm_frame, text="VLM URL:").grid(column=0, row=0, sticky=tk.W, pady=2)
|
||
ttk.Entry(vlm_frame, textvariable=self.vlm_url, width=40).grid(column=1, row=0, sticky=(tk.W, tk.E))
|
||
ttk.Label(vlm_frame, text="VLM API Key:").grid(column=0, row=1, sticky=tk.W, pady=2)
|
||
ttk.Entry(vlm_frame, textvariable=self.vlm_api_key, width=40).grid(column=1, row=1, sticky=(tk.W, tk.E))
|
||
ttk.Label(vlm_frame, text="VLM 模型:").grid(column=0, row=2, sticky=tk.W, pady=2)
|
||
ttk.Entry(vlm_frame, textvariable=self.vlm_model, width=40).grid(column=1, row=2, sticky=(tk.W, tk.E))
|
||
|
||
# LLM服务设置区域
|
||
llm_frame = ttk.LabelFrame(frame, text="LLM (语言模型) 设置", padding="10")
|
||
llm_frame.grid(row=1, column=0, sticky=(tk.W, tk.E), pady=5)
|
||
llm_frame.grid_columnconfigure(1, weight=1)
|
||
ttk.Label(llm_frame, text="LLM URL:").grid(column=0, row=0, sticky=tk.W, pady=2)
|
||
ttk.Entry(llm_frame, textvariable=self.llm_url, width=40).grid(column=1, row=0, sticky=(tk.W, tk.E))
|
||
ttk.Label(llm_frame, text="LLM API Key:").grid(column=0, row=1, sticky=tk.W, pady=2)
|
||
ttk.Entry(llm_frame, textvariable=self.llm_api_key, width=40).grid(column=1, row=1, sticky=(tk.W, tk.E))
|
||
ttk.Label(llm_frame, text="LLM 模型:").grid(column=0, row=2, sticky=tk.W, pady=2)
|
||
ttk.Entry(llm_frame, textvariable=self.llm_model, width=40).grid(column=1, row=2, sticky=(tk.W, tk.E))
|
||
|
||
# 其他应用参数设置区域
|
||
other_frame = ttk.LabelFrame(frame, text="其他设置", padding="10")
|
||
other_frame.grid(row=2, column=0, sticky=(tk.W, tk.E), pady=5)
|
||
other_frame.grid_columnconfigure(1, weight=1)
|
||
ttk.Label(other_frame, text="手写打分敏感度:").grid(column=0, row=0, sticky=tk.W, pady=2)
|
||
ttk.Entry(other_frame, textvariable=self.sensitivity_factor, width=40).grid(column=1, row=0, sticky=(tk.W, tk.E))
|
||
ttk.Label(other_frame, text="最大并发数:").grid(column=0, row=1, sticky=tk.W, pady=2)
|
||
ttk.Entry(other_frame, textvariable=self.max_workers, width=40).grid(column=1, row=1, sticky=(tk.W, tk.E))
|
||
ttk.Label(other_frame, text="最大重试次数:").grid(column=0, row=2, sticky=tk.W, pady=2)
|
||
ttk.Entry(other_frame, textvariable=self.max_retries, width=40).grid(column=1, row=2, sticky=(tk.W, tk.E))
|
||
ttk.Label(other_frame, text="重试延迟(秒):").grid(column=0, row=3, sticky=tk.W, pady=2)
|
||
ttk.Entry(other_frame, textvariable=self.retry_delay, width=40).grid(column=1, row=3, sticky=(tk.W, tk.E))
|
||
ttk.Label(other_frame, text="保存Markdown文件:").grid(column=0, row=4, sticky=tk.W, pady=2)
|
||
ttk.Checkbutton(other_frame, variable=self.save_markdown).grid(column=1, row=4, sticky=tk.W)
|
||
ttk.Label(other_frame, text="渲染HTML报告:").grid(column=0, row=5, sticky=tk.W, pady=2)
|
||
ttk.Checkbutton(other_frame, variable=self.render_markdown).grid(column=1, row=5, sticky=tk.W)
|
||
|
||
# LLM Prompt模板编辑区域
|
||
prompt_frame = ttk.LabelFrame(frame, text="LLM Prompt 模板 (可在此修改，请勿修改{}占位符内容导致程序参数无法正常传递，通常情况下修改总分即可)", padding="10")
|
||
prompt_frame.grid(row=3, column=0, sticky=(tk.W, tk.E), pady=5)
|
||
prompt_frame.grid_columnconfigure(0, weight=1)
|
||
prompt_frame.grid_rowconfigure(0, weight=1)
|
||
|
||
self.llm_prompt_text = tk.Text(prompt_frame, height=10, wrap="word")
|
||
self.llm_prompt_text.grid(row=0, column=0, sticky="nsew")
|
||
prompt_scrollbar = ttk.Scrollbar(prompt_frame, orient="vertical", command=self.llm_prompt_text.yview)
|
||
prompt_scrollbar.grid(row=0, column=1, sticky="ns")
|
||
self.llm_prompt_text.config(yscrollcommand=prompt_scrollbar.set)
|
||
self.llm_prompt_text.insert("1.0", self.llm_prompt_template_str)
|
||
|
||
btn_frame = ttk.Frame(frame)
|
||
btn_frame.grid(row=4, column=0, sticky=tk.E, pady=10)
|
||
ttk.Button(btn_frame, text="确定", command=self.on_ok).pack(side=tk.LEFT, padx=5)
|
||
ttk.Button(btn_frame, text="关闭", command=self.on_close).pack(side=tk.LEFT)
|
||
|
||
self.protocol("WM_DELETE_WINDOW", self.on_close)
|
||
self.grab_set()
|
||
self.wait_window(self)
|
||
|
||
def on_ok(self):
|
||
self.result = {
|
||
"VlmUrl": self.vlm_url.get(),
|
||
"VlmApiKey": self.vlm_api_key.get(),
|
||
"VlmModel": self.vlm_model.get(),
|
||
"LlmUrl": self.llm_url.get(),
|
||
"LlmApiKey": self.llm_api_key.get(),
|
||
"LlmModel": self.llm_model.get(),
|
||
"SensitivityFactor": self.sensitivity_factor.get(),
|
||
"MaxWorkers": self.max_workers.get(),
|
||
"MaxRetries": self.max_retries.get(),
|
||
"RetryDelay": self.retry_delay.get(),
|
||
"SaveMarkdown": self.save_markdown.get(),
|
||
"RenderMarkdown": self.render_markdown.get(),
|
||
"LlmPromptTemplate": self.llm_prompt_text.get("1.0", "end-1c")
|
||
}
|
||
|
||
# 如果用户修改后的模板与默认模板内容一致，则不写入配置文件，以使用默认值
|
||
if self.result["LlmPromptTemplate"].strip() == DEFAULT_LLM_PROMPT_TEMPLATE.strip():
|
||
self.result["LlmPromptTemplate"] = None # 使用 None 作为信号，表示应移除此配置项
|
||
|
||
self.destroy()
|
||
|
||
def on_close(self):
|
||
self.result = None
|
||
self.destroy()
|
||
|
||
|
||
class MainApp:
|
||
"""应用主窗口类，负责构建UI界面、处理用户交互和协调后台服务。"""
|
||
def __init__(self, root: tk.Tk, config_manager: ConfigManager):
|
||
self.root = root
|
||
self.config_manager = config_manager
|
||
self.ui_queue = queue.Queue()
|
||
self.api_service = ApiService(config_manager, self.ui_queue)
|
||
|
||
self.file_paths: List[str] = []
|
||
self.is_file_selected = False
|
||
self.topic_input = None
|
||
self.processed_count = 0
|
||
self.lock = threading.Lock()
|
||
|
||
self._setup_ui()
|
||
self._initialize_config()
|
||
self.root.after(100, self._process_ui_queue)
|
||
|
||
def _setup_ui(self):
|
||
"""初始化和布局主窗口的所有UI组件。"""
|
||
self.root.title("AI 作文批改助手")
|
||
self.root.geometry("550x450")
|
||
main_frame = ttk.Frame(self.root, padding="10")
|
||
main_frame.grid(row=0, column=0, sticky=(tk.W, tk.E, tk.N, tk.S))
|
||
self.root.grid_columnconfigure(0, weight=1)
|
||
self.root.grid_rowconfigure(0, weight=1)
|
||
|
||
# 顶部进度条
|
||
top_frame = ttk.Frame(main_frame)
|
||
top_frame.grid(row=0, column=0, columnspan=2, sticky=(tk.W, tk.E), pady=5)
|
||
top_frame.grid_columnconfigure(0, weight=1)
|
||
self.progress_bar = ttk.Progressbar(top_frame, orient="horizontal", mode="determinate")
|
||
self.progress_bar.grid(row=0, column=0, sticky=(tk.W, tk.E))
|
||
|
||
# 左侧控制按钮
|
||
left_frame = ttk.Frame(main_frame)
|
||
left_frame.grid(row=1, column=0, sticky=(tk.N, tk.W), padx=(0, 10))
|
||
ttk.Button(left_frame, text="选择图片", command=self._open_file_dialog).pack(fill=tk.X, pady=5)
|
||
ttk.Button(left_frame, text="开始批改", command=self._start_processing).pack(fill=tk.X, pady=5)
|
||
ttk.Button(left_frame, text="设置", command=self._open_settings_dialog).pack(fill=tk.X, pady=5)
|
||
ttk.Button(left_frame, text="关于", command=self._open_about_dialog).pack(fill=tk.X, pady=5)
|
||
|
||
# 右侧输入和日志区域
|
||
right_frame = ttk.Frame(main_frame)
|
||
right_frame.grid(row=1, column=1, sticky=(tk.W, tk.E, tk.N, tk.S))
|
||
main_frame.grid_columnconfigure(1, weight=1)
|
||
main_frame.grid_rowconfigure(1, weight=1)
|
||
|
||
# 配置右侧框架的网格权重，使组件能自适应缩放
|
||
right_frame.grid_rowconfigure(0, weight=7) # 作文题目框占7份
|
||
right_frame.grid_rowconfigure(1, weight=3) # 日志框占3份
|
||
right_frame.grid_columnconfigure(0, weight=1)
|
||
|
||
# 作文题目输入框
|
||
self.topic_input = tk.Text(right_frame, wrap="word")
|
||
self.topic_input.grid(row=0, column=0, sticky="nsew", pady=(0, 5))
|
||
|
||
self.topic_input.insert("1.0", "（在此输入作文题目）")
|
||
self.topic_input.config(fg="grey")
|
||
|
||
# 实现输入框的占位符（placeholder）效果
|
||
def on_focus_in(event):
|
||
if self.topic_input.get("1.0", "end-1c").strip() == "（在此输入作文题目）":
|
||
self.topic_input.delete("1.0", tk.END)
|
||
self.topic_input.config(fg="black")
|
||
|
||
def on_focus_out(event):
|
||
if not self.topic_input.get("1.0", "end-1c").strip():
|
||
self.topic_input.insert("1.0", "（在此输入作文题目）")
|
||
self.topic_input.config(fg="grey")
|
||
|
||
self.topic_input.bind("<FocusIn>", on_focus_in)
|
||
self.topic_input.bind("<FocusOut>", on_focus_out)
|
||
|
||
# 日志输出框（带滚动条）
|
||
listbox_frame = ttk.Frame(right_frame)
|
||
listbox_frame.grid(row=1, column=0, sticky="nsew")
|
||
listbox_frame.grid_rowconfigure(0, weight=1)
|
||
listbox_frame.grid_columnconfigure(0, weight=1)
|
||
|
||
self.listbox = tk.Listbox(listbox_frame)
|
||
self.listbox.grid(row=0, column=0, sticky="nsew")
|
||
|
||
scrollbar = ttk.Scrollbar(listbox_frame, orient="vertical", command=self.listbox.yview)
|
||
scrollbar.grid(row=0, column=1, sticky="ns")
|
||
|
||
self.listbox.config(yscrollcommand=scrollbar.set)
|
||
|
||
def _log(self, message: str):
|
||
"""将消息添加到日志列表框并滚动到底部。"""
|
||
self.listbox.insert(tk.END, message)
|
||
self.listbox.see(tk.END)
|
||
|
||
def _initialize_config(self):
|
||
"""应用启动时检查配置完整性，如果配置不完整则强制用户设置。"""
|
||
is_ok, _ = self.config_manager.check_settings()
|
||
if not is_ok:
|
||
self._show_config_dialog_until_valid()
|
||
|
||
def _show_config_dialog_until_valid(self):
|
||
"""循环显示设置对话框，直到所有必需配置项都已填写。"""
|
||
while True:
|
||
self._open_settings_dialog()
|
||
is_ok, missing_item = self.config_manager.check_settings()
|
||
if is_ok:
|
||
return
|
||
if messagebox.askretrycancel("配置未完成", f"请配置: {missing_item}") == "cancel":
|
||
self.root.quit()
|
||
return
|
||
|
||
def _open_settings_dialog(self):
|
||
"""打开设置对话框，并根据返回结果更新和保存配置。"""
|
||
dialog = SettingsDialog(self.root, self.config_manager)
|
||
if dialog.result:
|
||
# 清理旧的、统一的AI配置和OCR配置，以兼容新版分离的配置
|
||
self.config_manager.config.pop("AiUrl", None)
|
||
self.config_manager.config.pop("AiApiKey", None)
|
||
self.config_manager.config.pop("OcrApiKey", None)
|
||
self.config_manager.config.pop("OcrSecretKey", None)
|
||
for key, value in dialog.result.items():
|
||
if key == "LlmPromptTemplate":
|
||
if value is None:
|
||
# 如果值为None，表示用户希望恢复默认模板，因此从配置中移除该键
|
||
self.config_manager.config.pop(key, None)
|
||
else:
|
||
self.config_manager.set(key, value)
|
||
else:
|
||
self.config_manager.set(key, value)
|
||
self.config_manager.save()
|
||
|
||
def _open_about_dialog(self):
|
||
"""创建并显示“关于”对话框。"""
|
||
AboutDialog(self.root, self.config_manager)
|
||
|
||
def _open_file_dialog(self):
|
||
"""打开文件选择对话框，让用户选择一个或多个图片文件。"""
|
||
paths = filedialog.askopenfilenames(title="选择作文图片", filetypes=[("图片文件", "*.jpg *.jpeg *.png *.bmp")])
|
||
if paths:
|
||
self.file_paths = paths
|
||
self.is_file_selected = True
|
||
self._log(f"已选择 {len(paths)} 个文件")
|
||
else:
|
||
self._log("取消选择")
|
||
|
||
def _start_processing(self):
|
||
"""启动作文批改流程。"""
|
||
if not self.is_file_selected:
|
||
messagebox.showerror("操作错误", "请先选择文件")
|
||
return
|
||
|
||
topic = self.topic_input.get("1.0", tk.END).strip()
|
||
if not topic or topic == "（在此输入作文题目）":
|
||
messagebox.showerror("操作错误", "请输入作文题目")
|
||
return
|
||
|
||
# 重置进度条和计数器
|
||
self.progress_bar['value'] = 0
|
||
self.progress_bar['maximum'] = len(self.file_paths)
|
||
self.processed_count = 0
|
||
|
||
# 在后台线程中启动并发处理
|
||
thread = threading.Thread(target=self._concurrent_worker_manager, args=(self.file_paths, topic), daemon=True)
|
||
thread.start()
|
||
|
||
def _process_ui_queue(self):
|
||
"""定期检查UI更新队列，并执行相应的UI操作（如记日志、更新进度条）。"""
|
||
try:
|
||
while True:
|
||
task, data = self.ui_queue.get_nowait()
|
||
if task == "log":
|
||
self._log(data)
|
||
elif task == "progress":
|
||
with self.lock:
|
||
self.processed_count += 1
|
||
self.progress_bar['value'] = self.processed_count
|
||
elif task == "finish":
|
||
messagebox.showinfo("完成", "所有文件处理完成")
|
||
self.progress_bar['value'] = 0
|
||
self.is_file_selected = False
|
||
except queue.Empty:
|
||
pass
|
||
finally:
|
||
# 持续轮询队列
|
||
self.root.after(100, self._process_ui_queue)
|
||
|
||
def _concurrent_worker_manager(self, file_paths: List[str], topic: str):
|
||
"""使用线程池并发处理所有选定的文件。"""
|
||
try:
|
||
# 强制将从配置中读取的值转换为整数，提供默认值以防万一
|
||
max_workers = int(self.config_manager.get("MaxWorkers", 4))
|
||
except (ValueError, TypeError):
|
||
max_workers = 4
|
||
|
||
with concurrent.futures.ThreadPoolExecutor(max_workers=max_workers) as executor:
|
||
for file_path in file_paths:
|
||
executor.submit(self._process_single_file, file_path, topic)
|
||
|
||
# 所有任务完成后，向UI队列发送完成信号
|
||
self.ui_queue.put(("finish", None))
|
||
|
||
def _process_single_file(self, file_path: str, topic: str):
|
||
"""处理单个图片文件的完整流程：调用API、保存报告、更新UI队列。"""
|
||
base_name = os.path.basename(file_path)
|
||
self.ui_queue.put(("log", f"开始处理: {base_name}"))
|
||
try:
|
||
final_report, vlm_usage, llm_usage, html_path = self.api_service.process_essay_image(file_path, topic)
|
||
|
||
# 检查是否保存Markdown文件
|
||
save_markdown = self.config_manager.get("SaveMarkdown", True)
|
||
report_filename_md = os.path.splitext(file_path)[0] + "_report.md"
|
||
|
||
# 保存Markdown源文件（如果配置开启）
|
||
if save_markdown:
|
||
with open(report_filename_md, 'w', encoding='utf-8') as f:
|
||
f.write(final_report)
|
||
|
||
vlm_in = vlm_usage.get("prompt_tokens", 0)
|
||
vlm_out = vlm_usage.get("completion_tokens", 0)
|
||
llm_in = llm_usage.get("prompt_tokens", 0)
|
||
llm_out = llm_usage.get("completion_tokens", 0)
|
||
|
||
usage_log = f"Token用量: VLM(in:{vlm_in}, out:{vlm_out}), LLM(in:{llm_in}, out:{llm_out})"
|
||
|
||
# 记录所有生成的文件
|
||
output_files = []
|
||
if save_markdown:
|
||
output_files.append(os.path.basename(report_filename_md))
|
||
|
||
# 检查是否只勾选了HTML，如果是则删除Markdown文件
|
||
render_html = self.config_manager.get("RenderMarkdown", True)
|
||
if html_path and os.path.exists(html_path):
|
||
output_files.append(os.path.basename(html_path))
|
||
self.ui_queue.put(("log", f"已生成HTML报告: {os.path.basename(html_path)}"))
|
||
|
||
# 如果只勾选HTML，不勾选Markdown，则删除Markdown文件
|
||
if not save_markdown and render_html and os.path.exists(report_filename_md):
|
||
os.remove(report_filename_md)
|
||
self.ui_queue.put(("log", f"已删除Markdown文件（仅保留HTML）"))
|
||
|
||
self.ui_queue.put(("log", f"完成批改: {base_name} -> {', '.join(output_files)}"))
|
||
self.ui_queue.put(("log", usage_log))
|
||
|
||
# 加锁以保证线程安全地更新和保存配置
|
||
with self.lock:
|
||
self.config_manager.update_token_usage(vlm_in, vlm_out, llm_in, llm_out)
|
||
self.config_manager.save()
|
||
|
||
except Exception as e:
|
||
self.ui_queue.put(("log", f"文件: {base_name} 失败: {e}"))
|
||
|
||
# 无论成功或失败，都更新进度
|
||
self.ui_queue.put(("progress", 1))
|