v1.3

main.py

@@ -1,80 +1,135 @@
 import ctypes
+import glob
 import json
 import os
 import re
 import sys
 import time
-import sv_ttk
-import glob
-import dialogs
 from random import choices
-from tkinter import filedialog, StringVar, BooleanVar, Tk
+from tkinter import filedialog, StringVar, BooleanVar, Tk, DISABLED, END, NORMAL, WORD, Text
 from tkinter.ttk import *
+import sv_ttk
+import dialogs
+import threading
 
 # 匹配的前缀
 prefixes = [
-    "displayName", "tooltip", "tell", "Text.yellow", "Text.of",
-    "Text.white", "Text.green", "Text.gray", "Text.gold", "Text.darkPurple",
-    "Text.blue", "Text.darkGreen", "Text.aqua", "Text.red"
+    "displayName", "tooltip", "tell", "yellow", "Text.of",
+    "white", "green", "gray", "gold", "darkPurple",
+    "blue", "darkGreen", "aqua", "red", "string", "lightPurple"
 ]
 
 # 定义存储键值对的字典
 translation_dict = {}
 
-
 def browseJson():
     json_var.set(
         filedialog.askopenfilename(filetypes=(('选择 json 语言文件', "*.json"),),
                                    initialdir=os.path.abspath(os.path.dirname(__file__))))
-
+def browseEnglishJson():
+    translate_var.set(
+        filedialog.askopenfilename(filetypes=(('选择 json 英文语言文件', "*.json"),),
+                                   initialdir=os.path.abspath(os.path.dirname(__file__))))
 
 def browseKubeJS():
     kubejs_var.set(
         filedialog.askdirectory(title='选择处理 KubeJS 的文件夹',
                                 initialdir=os.path.abspath(os.path.dirname(__file__))))
 
-
-def on_mode_select(event):
-    if mode_entry.get() == '回填':
-        return mode_entry.set('回填')
-
+def start_ai_translate():
+    threading.Thread(target=ai_translate).start()
+
+def ai_translate():
+    if translate_var.get() == "":
+        dialogs.show_message("错误", "请先选择json文件")
+    else:
+        sys.stdout = output_text
+        print('正在加载模型，请稍等，预计8秒左右\n')
+        root.update_idletasks()
+        from transformers import MarianMTModel, MarianTokenizer
+
+        start_time = time.time()
+        tokenizer = MarianTokenizer.from_pretrained('model')
+        model = MarianMTModel.from_pretrained('model')
+
+        print('加载模型完成，正在加载英文语言文件，请稍等')
+        root.update_idletasks()
+        with open(translate_var.get(), 'r', encoding='utf-8') as f:
+            data = json.load(f)
+        print('加载完成，正在AI汉化中，请稍等（耗时与CPU性能有关）')
+        root.update_idletasks()
+        count = 0
+        for key in data.keys():
+            text = data[key]
+            input_ids = tokenizer.encode(text, return_tensors='pt')
+            outputs = model.generate(input_ids)
+            translated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+            translated_text = translated_text.replace(',', '，').replace('!', '！').replace('?', '？').replace(': ', '：').replace('...', '…')
+            data[key] = translated_text
+            count += 1
+            if count % 10 == 0:
+                # 每翻译10条文本就保存一次
+                with open('zh_cn.json', 'w', encoding='utf-8') as f:
+                    json.dump(data, f, ensure_ascii=False, indent=4)
+                print(f'已汉化完成{count}条翻译，并自动保存zh_cn.json')
+                root.update_idletasks()  # Update the GUI
+
+        end_time = time.time()
+        total_time = end_time - start_time
+
+        with open('zh_cn.json', 'w', encoding='utf-8') as f:
+            json.dump(data, f, ensure_ascii=False, indent=4)
+            f.close()
+
+        print(f'AI一键汉化完成！共耗时{total_time}秒\n汉化文件zh_cn.json已保存至当前目录下')
+        root.update_idletasks()
+        dialogs.show_message('KubeJS翻译工具', f'AI一键汉化完成！共耗时{total_time}秒\n汉化文件zh_cn.json已保存至当前目录下')
 
 def gui():
-    saveButton.grid_forget()
     r = 1
     json_lb.grid(row=r, column=0, pady=5, padx=20, sticky='nsew')
     json_entry.grid(row=r, column=1, pady=5, sticky='nsew')
-    jsonButton.grid(row=r, column=2, pady=5, padx=30, sticky='nsew')
+    json_button.grid(row=r, column=2, pady=5, padx=30, sticky='nsew')
     r += 1
     kubejs_lb.grid(row=r, column=0, pady=5, padx=20, sticky='nsew')
     kubejs_entry.grid(row=r, column=1, pady=5, sticky='nsew')
-    kubeJSButton.grid(row=r, column=2, pady=5, padx=30, sticky='nsew')
+    kubejs_button.grid(row=r, column=2, pady=5, padx=30, sticky='nsew')
     r += 2
     mode_lb.grid(row=r, column=0, pady=5, padx=20, sticky='nsew')
     mode_entry.current(0)
     mode_entry.grid(row=r, column=1, pady=5, padx=30, sticky='nsew')
     sv_ttk_theme.grid(row=r, column=2, pady=5, padx=30, sticky='nsew')
     r += 1
-    replaceButton.grid(row=r, column=1, pady=5, padx=30, sticky='nsew')
-    saveButton.grid(row=r, column=2, pady=5, padx=30, sticky='nsew')
+    replace_button.grid(row=r, column=1, pady=5, padx=30, sticky='nsew')
+    save_button.grid(row=r, column=2, pady=5, padx=30, sticky='nsew')
 
     # 让列和行自适应
     for i in range(3):  # 列数
-        root.columnconfigure(i, weight=1)
+        tab1.columnconfigure(i, weight=1)
+        tab2.columnconfigure(i, weight=1)
 
     for i in range(r + 1):  # 行数
-        root.rowconfigure(i, weight=1)
+        tab1.rowconfigure(i, weight=1)
+        tab2.rowconfigure(i, weight=1)
 
+    r = 1
+    translate_lb.grid(row=r, column=0, pady=5, padx=20, sticky='nsew')
+    translate_entry.grid(row=r, column=1, pady=5, sticky='nsew')
+    translate_button.grid(row=r, column=2, pady=5, padx=30, sticky='nsew')
+    r += 1
+    translate_title.grid(row=r, column=0, pady=5, padx=20, sticky='nsew')
+    output_text.grid(row=r, column=1, pady=5, padx=20, sticky='nsew')
+    scrollbar.grid(row=r, column=2, padx=20, sticky="ns")
+    r += 1
+    save_translate_button.grid(row=r, column=2, pady=5, padx=20, sticky='nsew')
 
-def replace_keys_in_js(json_file, folder_path):
-    with open(json_file, 'r', encoding='utf-8') as json_file:
-        translation_dict = json.load(json_file)
-    total_files = sum(1 for root, dirs, files in os.walk(folder_path) for filename in files if filename.endswith('.js'))
+    notebook.grid(row=0, column=0, sticky='nsew')
 
+def create_progress_bar(title, total_files):
     progress = Tk()
-    progress.title('替换进度')
+    progress.title(title)
 
-    progress_title = Label(progress, text="替换进度")
+    progress_title = Label(progress, text=title)
     progress_title.grid(row=0, column=0, pady=20, padx=20)
     progress_lb = Label(progress, text="0/{}".format(total_files))
     progress_lb.grid(row=1, column=0, pady=20, padx=20)
@@ -85,8 +140,25 @@ def replace_keys_in_js(json_file, folder_path):
     x = (progress.winfo_screenwidth() // 2) - (progress.winfo_width() // 2) + 250
     y = (progress.winfo_screenheight() // 2) - (progress.winfo_height() // 2)
     progress.geometry('{}x{}+{}+{}'.format(700, 200, x, y))
+
+    return progress, progress_lb, progress_bar
+
+def update_progress(progress_lb, progress_bar, current_progress, total_files):
+    progress_lb.config(text="{}/{}".format(current_progress, total_files))
+    progress_lb.update()
+    if current_progress < total_files:
+        progress_bar.step(1)
+        progress_bar.update()
+
+def replace_keys_in_js(json_file, folder_path):
+    with open(json_file, 'r', encoding='utf-8') as json_file:
+        translation_dict = json.load(json_file)
+
+    total_files = sum(
+        1 for root, dirs, files in os.walk(folder_path) for filename in files if filename.endswith('.js'))
+    progress, progress_lb, progress_bar = create_progress_bar('替换进度', total_files)
+
     current_progress = 0
-    # 处理替换
     for file_path in glob.glob(os.path.join(folder_path, '**/*.js'), recursive=True):
         with open(file_path, 'r', encoding='utf-8') as file:
             js_content = file.read()
@@ -96,11 +168,7 @@ def replace_keys_in_js(json_file, folder_path):
             file.write(js_content)
 
         current_progress += 1
-        progress_lb.config(text="{}/{}".format(current_progress, total_files))
-        progress_lb.update()
-        if current_progress < total_files:
-            progress_bar.step(1)
-            progress_bar.update()
+        update_progress(progress_lb, progress_bar, current_progress, total_files)
 
     success_label = Label(progress, text="KubeJS翻译工具回填翻译成功！", font=("微软雅黑", 22, "bold"))
     success_label.grid(row=2, column=0, columnspan=2, padx=10)
@@ -111,7 +179,7 @@ def replace_keys_in_js(json_file, folder_path):
 
 def runFromGui():
     if len(kubejs_var.get()) == 0 and mode_entry.get() == '提取':
-        dialogs.show_message('错误', '你需要选一个文件夹。')
+        dialogs.show_message('错误', '请先选择文件夹。')
     elif mode_entry.get() == '提取' and not replace2_var.get():
         # 遍历文件夹里的所有.js文件
         for file_path in glob.glob(os.path.join(kubejs_var.get(), '**/*.js'), recursive=True):
@@ -141,21 +209,8 @@ def runFromGui():
     elif mode_entry.get() == "提取" and replace2_var.get():
         total_files = sum(
             1 for root, dirs, files in os.walk(kubejs_var.get()) for filename in files if filename.endswith('.js'))
+        progress, progress_lb, progress_bar = create_progress_bar('替换进度', total_files)
 
-        progress = Tk()
-        progress.title('替换进度')
-
-        progress_title = Label(progress, text="替换进度")
-        progress_title.grid(row=0, column=0, pady=20, padx=20)
-        progress_lb = Label(progress, text="0/{}".format(total_files))
-        progress_lb.grid(row=1, column=0, pady=20, padx=20)
-        progress_bar = Progressbar(progress, length=300, mode='determinate', maximum=total_files)
-        progress_bar.grid(row=1, column=1, pady=20, padx=20)
-
-        progress.update_idletasks()
-        x = (progress.winfo_screenwidth() // 2) - (progress.winfo_width() // 2) + 250
-        y = (progress.winfo_screenheight() // 2) - (progress.winfo_height() // 2)
-        progress.geometry('{}x{}+{}+{}'.format(700, 200, x, y))
         current_progress = 0
         # 遍历文件夹里的所有.js文件
         for file_path in glob.glob(os.path.join(kubejs_var.get(), '**/*.js'), recursive=True):
@@ -174,11 +229,7 @@ def runFromGui():
                 translation_dict[random_key] = match
 
             current_progress += 1
-            progress_lb.config(text="{}/{}".format(current_progress, total_files))
-            progress_lb.update()
-            if current_progress < total_files:
-                progress_bar.step(1)
-                progress_bar.update()
+            update_progress(progress_lb, progress_bar, current_progress, total_files)
             # 写回原始文件
             with open(file_path, 'w', encoding='utf-8') as file:
                 file.write(js_content)
@@ -211,41 +262,70 @@ def replace2():
         if not replace_mode2:
             replace2_var.set(False)
 
+class ReadOnlyText(Text):
+    def __init__(self, master=None, **kwargs):
+        super().__init__(master, **kwargs)
+        self.config(state=DISABLED)
+
+    def write(self, text):
+        self.config(state=NORMAL)
+        self.insert(END, text)
+        self.see(END)
+        self.config(state=DISABLED)
+
+    def flush(self):
+        pass
 
 root = Tk()
 if sys.platform == "win32":
     try:
         # >= Windows 8.1
         ScaleFactor = ctypes.windll.shcore.SetProcessDpiAwareness(2)
-    except:
+    except OSError:
         # <= Windows 8.0
         ScaleFactor = ctypes.windll.user32.SetProcessDPIAware(0)
 
-
-root.title("KubeJS翻译工具 1.2")
-root.iconbitmap(bitmap="icon.ico")
-kubejs_var, json_var = StringVar(), StringVar()
-kubejs_entry = Entry(textvariable=kubejs_var)
-json_entry = Entry(textvariable=json_var)
-kubejs_lb = Label(text="选择处理 KubeJS 的文件夹", font=10)
-json_lb = Label(text="选择 json 语言文件（仅回填模式）", font=10)
-kubeJSButton = Button(text="选择", command=browseKubeJS)
-jsonButton = Button(text="选择", command=browseJson)
-mode_lb = Label(text="工作模式", font=10)
+root.title("KubeJS翻译工具 1.3")
+# root.iconbitmap(bitmap="icon.ico")
+root.columnconfigure(0, weight=1)
+root.rowconfigure(0, weight=1)
+
+notebook = Notebook(root)
+tab1, tab2 = Frame(notebook), Frame(notebook)
+notebook.add(tab1, text="工具")
+notebook.add(tab2, text="大模型 AI 汉化")
+kubejs_var, json_var, translate_var = StringVar(), StringVar(), StringVar()
+kubejs_entry = Entry(tab1, textvariable=kubejs_var)
+json_entry = Entry(tab1, textvariable=json_var)
+translate_entry = Entry(tab2, textvariable=translate_var)
+
+kubejs_lb = Label(tab1, text="选择处理 KubeJS 的文件夹", font=10)
+json_lb = Label(tab1, text="选择 json 语言文件（仅回填模式）", font=10)
+kubejs_button = Button(tab1, text="选择", command=browseKubeJS)
+json_button = Button(tab1, text="选择", command=browseJson)
+mode_lb = Label(tab1, text="工作模式", font=10)
 mode_list = ["提取", "回填"]
-mode_entry = Combobox(state="readonly", values=mode_list)
-mode_entry.bind("<<ComboboxSelected>>", on_mode_select)
+mode_entry = Combobox(tab1, state="readonly", values=mode_list)
+mode_entry.bind("<<ComboboxSelected>>")
 sv_ttk.use_light_theme()
-sv_ttk_theme = Checkbutton(text="暗色模式", style="Switch.TCheckbutton", command=sv_ttk.toggle_theme)
-saveButton = Button(text="生成", style="Accent.TButton", command=runFromGui)
+sv_ttk_theme = Checkbutton(tab1, text="暗色模式", style="Switch.TCheckbutton", command=sv_ttk.toggle_theme)
+save_button = Button(tab1, text="生成", style="Accent.TButton", command=runFromGui)
 replace2_var = BooleanVar()
-replaceButton = Checkbutton(root, text="暴力替换", variable=replace2_var, command=replace2)
+replace_button = Checkbutton(tab1, text="暴力替换", variable=replace2_var, command=replace2)
+
+translate_title = Label(tab2, text="使用本地AI汉化很慢！\nCPU占用约为50%左右", font=10)
+translate_lb = Label(tab2, text="选择英文语言文件", font=10)
+translate_button = Button(tab2, text="选择", command=browseEnglishJson)
+save_translate_button = Button(tab2, text="一键汉化", style="Accent.TButton", command=start_ai_translate)
+output_text = ReadOnlyText(tab2, wrap=WORD, width=40, height=10)
+scrollbar = Scrollbar(tab2, command=output_text.yview)
+output_text.config(yscrollcommand=scrollbar.set)
 
 gui()
 root.update_idletasks()
-x = (root.winfo_screenwidth() // 2) - (root.winfo_width() // 2) + 250
-y = (root.winfo_screenheight() // 2) - (root.winfo_height() // 2)
-root.geometry('{}x{}+{}+{}'.format(740, 330, x, y))
+x = (root.winfo_screenwidth() // 2) - (root.winfo_width() // 2) + 300
+y = (root.winfo_screenheight() // 2) - (root.winfo_height() // 2) + 60
+root.geometry('{}x{}+{}+{}'.format(770, 360, x, y))
 
 if __name__ == "__main__":
     root.mainloop()