# filename: srt_interactive_refiner_v2.0.py import tkinter as tk from tkinter import filedialog, ttk, messagebox, scrolledtext import sv_ttk import re from datetime import timedelta import os import requests import json import threading import queue import copy # --- 配置区 --- OLLAMA_HOST = "http://127.0.0.1:11434" # --- SRT核心类 和 解析函数 --- class SrtEntry: def __init__(self, index, start_td, end_td, text, original_index=None): self.index = index self.start_td = start_td self.end_td = end_td self.text = text.strip() self.original_index = original_index if original_index is not None else index @property def start_str(self): return self._td_to_str(self.start_td) @property def end_str(self): return self._td_to_str(self.end_td) @staticmethod def _td_to_str(td): total_seconds = int(td.total_seconds()) ms = int((td.total_seconds() - total_seconds) * 1000) h, m, s = total_seconds // 3600, (total_seconds % 3600) // 60, total_seconds % 60 return f"{h:02d}:{m:02d}:{s:02d},{ms:03d}" def to_srt_block(self): return f"{self.index}\n{self.start_str} --> {self.end_str}\n{self.text}\n\n" def parse_srt(content): entries = [] pattern = re.compile(r'(\d+)\n(\d{2}:\d{2}:\d{2},\d{3})\s*-->\s*(\d{2}:\d{2}:\d{2},\d{3})\n([\s\S]*?)(?=\n\n|\Z)', re.MULTILINE) def to_td(time_str): h, m, s, ms = map(int, re.split('[:,]', time_str)) return timedelta(hours=h, minutes=m, seconds=s, milliseconds=ms) for match in pattern.finditer(content): index = int(match.group(1)) entries.append(SrtEntry(index, to_td(match.group(2)), to_td(match.group(3)), match.group(4))) return entries # --- GUI 应用 --- class App: def __init__(self, root): self.root = root self.root.title("交互式字幕编辑器 V2.0 (最终稳定版)") self.root.geometry("1400x800") self.srt_path = "" self.original_entries = [] self.working_entries = [] self.current_selected_work_iid = None self.current_selected_orig_iid = None self.gui_queue = queue.Queue() self.is_ollama_available = False self.vars = { "ollama_model": tk.StringVar(), "status": tk.StringVar(value="准备就绪") } self.build_ui() sv_ttk.set_theme("dark") self.root.after(100, self.process_queue) self.root.after(100, self.load_ollama_models) def build_ui(self): main_pane = ttk.PanedWindow(self.root, orient=tk.HORIZONTAL) main_pane.pack(fill=tk.BOTH, expand=True, padx=10, pady=10) left_pane = ttk.Frame(main_pane) main_pane.add(left_pane, weight=2) control_frame = ttk.Frame(left_pane) control_frame.pack(fill=tk.X, pady=(0, 10)) self.load_btn = ttk.Button(control_frame, text="加载SRT", command=self.load_srt) self.load_btn.pack(side=tk.LEFT, padx=(0, 5)) self.save_btn = ttk.Button(control_frame, text="另存为...", command=self.save_srt, state="disabled") self.save_btn.pack(side=tk.LEFT, padx=(0, 5)) self.refine_all_btn = ttk.Button(control_frame, text="🚀 一键润色", command=self.refine_all_lines, state="disabled", style="Accent.TButton") self.refine_all_btn.pack(side=tk.LEFT, padx=(10,5)) ttk.Label(control_frame, text="Ollama:").pack(side=tk.LEFT, padx=(10, 5)) self.model_combo = ttk.Combobox(control_frame, textvariable=self.vars['ollama_model'], state="readonly", width=25) self.model_combo.pack(side=tk.LEFT, fill=tk.X, expand=True) orig_frame = ttk.Labelframe(left_pane, text="原始字幕 (对照区)") orig_frame.pack(fill=tk.BOTH, expand=True) self.tree_orig = self.create_treeview(orig_frame) right_pane = ttk.Frame(main_pane) main_pane.add(right_pane, weight=3) work_frame = ttk.Labelframe(right_pane, text="工作区 (可编辑)") work_frame.pack(fill=tk.BOTH, expand=True) self.tree_work = self.create_treeview(work_frame) self.tree_work.bind("<>", self.on_tree_select) self.tree_work.tag_configure("modified", background="#3a6b3a") editor_frame = ttk.Labelframe(right_pane, text="单句编辑器") editor_frame.pack(fill=tk.X, pady=(10, 0)) self.editor_text = scrolledtext.ScrolledText(editor_frame, height=4, wrap=tk.WORD, state="disabled") self.editor_text.pack(fill=tk.X, padx=5, pady=5) button_bar = ttk.Frame(editor_frame) button_bar.pack(fill=tk.X, padx=5, pady=(0, 5)) button_bar.columnconfigure((0, 1, 2, 3), weight=1) self.refine_btn = ttk.Button(button_bar, text="润色", command=self.refine_current_line, state="disabled") self.refine_btn.grid(row=0, column=0, sticky="ew", padx=(0, 2)) self.revert_btn = ttk.Button(button_bar, text="还原文本", command=self.revert_current_line_text, state="disabled") self.revert_btn.grid(row=0, column=1, sticky="ew", padx=2) self.apply_btn = ttk.Button(button_bar, text="✅ 应用更改", command=self.apply_editor_changes, state="disabled", style="Accent.TButton") self.apply_btn.grid(row=1, column=0, columnspan=2, sticky="ew", padx=(0,2), pady=(5,0)) self.delete_btn = ttk.Button(button_bar, text="❌ 删除此行", command=self.delete_current_line, state="disabled") self.delete_btn.grid(row=1, column=2, columnspan=2, sticky="ew", padx=2, pady=(5,0)) status_bar = ttk.Frame(self.root) status_bar.pack(side=tk.BOTTOM, fill=tk.X, padx=10, pady=(0, 5)) self.progress_bar = ttk.Progressbar(status_bar, orient=tk.HORIZONTAL, mode='determinate') self.progress_bar.pack(side=tk.LEFT, fill=tk.X, expand=True, padx=(0, 10)) ttk.Label(status_bar, textvariable=self.vars['status']).pack(side=tk.LEFT) def create_treeview(self, parent): cols = ("#0", "开始", "结束", "文本") tree = ttk.Treeview(parent, columns=cols[1:], show="headings") for col in cols: tree.heading(col, text=col, anchor="w") tree.column("#0", width=40, anchor="center"); tree.column("开始", width=90, anchor="w") tree.column("结束", width=90, anchor="w"); tree.column("文本", width=400, anchor="w") vsb = ttk.Scrollbar(parent, orient="vertical", command=tree.yview); hsb = ttk.Scrollbar(parent, orient="horizontal", command=tree.xview) tree.configure(yscrollcommand=vsb.set, xscrollcommand=hsb.set) vsb.pack(side=tk.RIGHT, fill=tk.Y); hsb.pack(side=tk.BOTTOM, fill=tk.X); tree.pack(side=tk.LEFT, fill=tk.BOTH, expand=True) return tree def load_srt(self): path = filedialog.askopenfilename(filetypes=[("SRT Subtitles", "*.srt")]) if not path: return self.srt_path = path try: with open(path, 'r', encoding='utf-8-sig') as f: content = f.read() self.original_entries = parse_srt(content) self.working_entries = copy.deepcopy(self.original_entries) self.populate_tree(self.tree_orig, self.original_entries) self.repopulate_work_tree() self.save_btn.config(state="normal") self.refine_all_btn.config(state="normal" if self.is_ollama_available else "disabled") self.current_selected_work_iid = None self.editor_text.config(state="normal"); self.editor_text.delete(1.0, tk.END); self.editor_text.config(state="disabled") self.set_editor_buttons_state(False) self.vars['status'].set(f"已加载 {len(self.original_entries)} 条字幕。") except Exception as e: messagebox.showerror("加载失败", f"无法加载或解析文件: {e}") def populate_tree(self, tree, entries): tree.delete(*tree.get_children()) for entry in entries: values = (entry.start_str, entry.end_str, entry.text.replace('\n', ' ')) tree.insert("", "end", text=str(entry.original_index), values=values, iid=str(entry.original_index)) def repopulate_work_tree(self): last_selected = self.current_selected_work_iid self.tree_work.delete(*self.tree_work.get_children()) for i, entry in enumerate(self.working_entries): entry.index = i + 1 values = (entry.start_str, entry.end_str, entry.text.replace('\n', ' ')) is_modified = entry.text != self.original_entries[entry.original_index - 1].text tags = ("modified",) if is_modified else () self.tree_work.insert("", "end", text=str(entry.index), values=values, iid=str(entry.index), tags=tags) if last_selected and self.tree_work.exists(str(last_selected)): self.tree_work.selection_set(str(last_selected)) self.tree_work.see(str(last_selected)) def on_tree_select(self, event): selection = self.tree_work.selection() if not selection: return work_iid = int(selection[0]) if work_iid > len(self.working_entries): return self.current_selected_work_iid = work_iid orig_iid = self.working_entries[work_iid - 1].original_index self.current_selected_orig_iid = orig_iid entry_text = self.working_entries[work_iid - 1].text self.editor_text.config(state="normal") self.editor_text.delete(1.0, tk.END) self.editor_text.insert(tk.END, entry_text) self.set_editor_buttons_state(True) if self.tree_orig.exists(str(orig_iid)): self.tree_orig.selection_set(str(orig_iid)) self.tree_orig.see(str(orig_iid)) def set_editor_buttons_state(self, is_enabled): state = "normal" if is_enabled else "disabled" is_ready = self.is_ollama_available and is_enabled self.refine_btn.config(state="normal" if is_ready else "disabled") self.revert_btn.config(state=state) self.apply_btn.config(state=state) self.delete_btn.config(state=state) def save_srt(self): if not self.working_entries: return self.apply_editor_changes() original_basename = os.path.splitext(os.path.basename(self.srt_path))[0] save_path = filedialog.asksaveasfilename(defaultextension=".srt", initialfile=f"{original_basename}_edited.srt", filetypes=[("SRT Subtitles", "*.srt")]) if not save_path: return try: for i, entry in enumerate(self.working_entries): entry.index = i + 1 with open(save_path, 'w', encoding='utf-8') as f: for entry in self.working_entries: f.write(entry.to_srt_block()) messagebox.showinfo("保存成功", f"文件已保存至:\n{save_path}") except Exception as e: messagebox.showerror("保存失败", f"无法保存文件: {e}") def apply_editor_changes(self): if self.current_selected_work_iid is not None: entry_index_in_list = self.current_selected_work_iid - 1 if entry_index_in_list < len(self.working_entries): new_text = self.editor_text.get(1.0, tk.END).strip() if new_text != self.working_entries[entry_index_in_list].text: self.working_entries[entry_index_in_list].text = new_text self.repopulate_work_tree() self.vars['status'].set(f"第 {self.current_selected_work_iid} 行已更新。") def delete_current_line(self): if self.current_selected_work_iid is None: return entry_index_in_list = self.current_selected_work_iid - 1 entry_to_delete = self.working_entries[entry_index_in_list] if messagebox.askyesno("确认删除", f"确定要删除第 {entry_to_delete.index} 行吗？\n'{entry_to_delete.text[:50]}...'"): self.working_entries.pop(entry_index_in_list) self.repopulate_work_tree() self.current_selected_work_iid = None self.editor_text.config(state="normal"); self.editor_text.delete(1.0, tk.END); self.editor_text.config(state="disabled") self.set_editor_buttons_state(False) self.vars['status'].set(f"原始行 {entry_to_delete.original_index} 已删除。") def refine_current_line(self): if self.current_selected_work_iid is None: return self.apply_editor_changes() original_text = self.original_entries[self.current_selected_orig_iid - 1].text self.set_buttons_state(False) threading.Thread(target=self._call_llm_for_refine, args=(original_text, self.vars['ollama_model'].get(), self.current_selected_work_iid), daemon=True).start() def refine_all_lines(self): if not self.working_entries: return if messagebox.askyesno("确认", f"即将对全部字幕进行润色，这会覆盖您当前的修改。是否继续？"): self.set_buttons_state(False) self.progress_bar['value'] = 0; self.progress_bar['maximum'] = len(self.original_entries) threading.Thread(target=self._refine_all_worker, args=(self.vars['ollama_model'].get(),), daemon=True).start() def _refine_all_worker(self, model_name): for i, entry in enumerate(self.original_entries): self.vars['status'].set(f"正在处理 {i+1}/{len(self.original_entries)}...") self.progress_bar['value'] = i + 1 refined_text = self._call_llm_for_refine_sync(entry.text, model_name) if refined_text: self.gui_queue.put({"type": "batch_line_refined", "data": {"orig_iid": entry.index, "text": refined_text}}) else: self.gui_queue.put({"type": "batch_line_refined", "data": {"orig_iid": entry.index, "text": entry.text, "no_change": True}}) self.gui_queue.put({"type": "batch_finish"}) def revert_current_line_text(self): if self.current_selected_orig_iid is None: return original_text = self.original_entries[self.current_selected_orig_iid - 1].text self.editor_text.delete(1.0, tk.END); self.editor_text.insert(tk.END, original_text) def _call_llm_for_refine(self, text, model, work_iid): refined_text = self._call_llm_for_refine_sync(text, model) if refined_text: self.gui_queue.put({"type": "line_refined", "data": {"iid": work_iid, "text": refined_text}}) else: self.gui_queue.put({"type": "refine_failed", "data": work_iid}) def _call_llm_for_refine_sync(self, text, model): prompt = f"""你是一个专业的视频字幕精炼师。任务是优化“待处理字幕”，使其更适合专业配音。 规则: 1. 改为流畅、专业的书面语，但必须保留所有的核心操作指令和细节。 2. 优先去除明显的口语化词汇、重复和不必要的填充词。 3. 在不影响信息完整性的前提下，可以适当缩短句子。 4. 【重要】只输出精炼后的字幕文本，不要包含任何标签、解释或引号。 --- [待处理字幕] {text} --- [精炼后的文本]：""" payload = {"model": model, "prompt": prompt, "stream": False, "options": {'temperature': 0.3}} try: response = requests.post(f"{OLLAMA_HOST}/api/generate", json=payload, timeout=45) response.raise_for_status() response_data = response.json() refined_text = response_data.get('response', '').strip().replace("\n", " ") return re.sub(r'^["\'“‘]|["\'”’]$', '', refined_text) except Exception as e: self.gui_queue.put({"type": "error", "data": f"API调用失败: {e}"}) return None def process_queue(self): try: while True: msg = self.gui_queue.get_nowait() msg_type, data = msg.get("type"), msg.get("data") if msg_type == "line_refined": if data["iid"] == self.current_selected_work_iid: self.editor_text.delete(1.0, tk.END) self.editor_text.insert(tk.END, data["text"]) self.set_buttons_state(True) elif msg_type == "batch_line_refined": orig_iid = data["orig_iid"] for work_entry in self.working_entries: if work_entry.original_index == orig_iid: work_entry.text = data["text"] values = (work_entry.start_str, work_entry.end_str, work_entry.text.replace('\n', ' ')) self.tree_work.item(str(work_entry.index), values=values, tags=("modified",)) break elif msg_type == "batch_finish": self.repopulate_work_tree() self.vars['status'].set("批量润色完成！") self.set_buttons_state(True) messagebox.showinfo("完成", "所有字幕已批量润色。请检查并进行微调。") elif msg_type == "refine_failed": if data == self.current_selected_work_iid: self.set_buttons_state(True) elif msg_type == "error": messagebox.showerror("Ollama 错误", data) elif msg_type == "models_loaded": if data: self.model_combo['values'] = data; self.model_combo.set(data[0]) self.is_ollama_available = True if self.working_entries: self.refine_all_btn.config(state="normal") else: self.gui_queue.put({"type": "error", "data": "Ollama连接成功, 但未检测到任何模型。请确保您已下载模型。"}) except queue.Empty: pass finally: self.root.after(100, self.process_queue) def set_buttons_state(self, is_enabled): state = "normal" if is_enabled else "disabled" self.load_btn.config(state=state); self.save_btn.config(state=state); self.refine_all_btn.config(state=state) self.set_editor_buttons_state(is_enabled and self.current_selected_work_iid is not None) def load_ollama_models(self): threading.Thread(target=self._load_models_worker, daemon=True).start() def _load_models_worker(self): # *** 修复: 补全此函数并增强错误处理 *** try: self.vars['status'].set("正在连接Ollama...") response = requests.get(f"{OLLAMA_HOST}/api/tags", timeout=5) response.raise_for_status() # 如果状态码不是 200-299，则抛出异常 # 检查返回的是否是有效的JSON try: models_data = response.json() except json.JSONDecodeError: self.gui_queue.put({"type": "error", "data": "Ollama返回了无效的数据格式，无法解析模型列表。"}) return models = models_data.get('models') if models is not None: model_names = [m['name'] for m in models] self.gui_queue.put({"type": "models_loaded", "data": model_names}) self.vars['status'].set("Ollama连接成功！") else: # models 键不存在 self.gui_queue.put({"type": "models_loaded", "data": []}) except requests.exceptions.Timeout: self.gui_queue.put({"type": "error", "data": f"连接Ollama超时 ({OLLAMA_HOST})。\n请检查服务是否运行且地址正确。"}) except requests.exceptions.ConnectionError: self.gui_queue.put({"type": "error", "data": f"无法连接到Ollama ({OLLAMA_HOST})。\n请确保Ollama服务正在运行。"}) except requests.exceptions.RequestException as e: self.gui_queue.put({"type": "error", "data": f"连接Ollama时发生网络错误: {e}"}) finally: if not self.is_ollama_available: self.vars['status'].set("Ollama连接失败，润色功能不可用。") if __name__ == "__main__": root = tk.Tk() app = App(root) root.mainloop()