Add Multi-language Support with Chinese UI

locales/zh.json

@@ -0,0 +1,41 @@
+{
+    "Source x Target Mapper": "Source x Target Mapper",
+    "select an source image": "选择一个源图像",
+    "Preview": "预览",
+    "select an target image or video": "选择一个目标图像或视频",
+    "save image output file": "保存图像输出文件",
+    "save video output file": "保存视频输出文件",
+    "select an target image": "选择一个目标图像",
+    "source": "源",
+    "Select a target": "选择一个目标",
+    "Select a face": "选择一张脸",
+    "Keep audio": "保留音频",
+    "Face Enhancer": "面纹增强器",
+    "Many faces": "多脸",
+    "Show FPS": "显示帧率",
+    "Keep fps": "保持帧率",
+    "Keep frames": "保持帧数",
+    "Fix Blueish Cam": "修复蓝色趋势的摄像头",
+    "Mouth Mask": "口罩",
+    "Show Mouth Mask Box": "显示口罩盒",
+    "Start": "开始",
+    "Live": "直播",
+    "Destroy": "结束",
+    "Map faces": "识别人脸",
+    "Processing...": "处理中...",
+    "Processing succeed!": "处理成功!",
+    "Processing ignored!": "处理被忽略!",
+    "Failed to start camera": "启动相机失败",
+    "Please complete pop-up or close it.": "请先完成弹出窗口或者关闭它",
+    "Getting unique faces": "获取独特面部",
+    "Please select a source image first": "请先选择一个源图像",
+    "No faces found in target": "目标图像中没有人脸",
+    "Add": "添加",
+    "Submit": "确认",
+    "Select source image": "请选取源图像",
+    "Select target image": "请选取目标图像",
+    "Please provide mapping!": "请提供映射",
+    "At least 1 source with target is required!": "至少需要一个来源图像与目标图像相关！",
+    "Face could not be detected in last upload!": "最近上传的图像中没有检测到人脸!",
+    "Select Camera:": "选择摄像头"
+}

modules/core.py

@@ -46,6 +46,7 @@ def parse_args() -> None:
     program.add_argument('--video-quality', help='adjust output video quality', dest='video_quality', type=int, default=18, choices=range(52), metavar='[0-51]')
     program.add_argument('--live-mirror', help='The live camera display as you see it in the front-facing camera frame', dest='live_mirror', action='store_true', default=False)
     program.add_argument('--live-resizable', help='The live camera frame is resizable', dest='live_resizable', action='store_true', default=False)
+    program.add_argument('-l', '--lang', help='UI language', default='en')
     program.add_argument('--max-memory', help='maximum amount of RAM in GB', dest='max_memory', type=int, default=suggest_max_memory())
     program.add_argument('--execution-provider', help='execution provider', dest='execution_provider', default=['cpu'], choices=suggest_execution_providers(), nargs='+')
     program.add_argument('--execution-threads', help='number of execution threads', dest='execution_threads', type=int, default=suggest_execution_threads())
@@ -105,6 +106,7 @@ def parse_args() -> None:
     if args.gpu_threads_deprecated:
         print('\033[33mArgument --gpu-threads is deprecated. Use --execution-threads instead.\033[0m')
         modules.globals.execution_threads = args.gpu_threads_deprecated
+    return args
 
 
 def encode_execution_providers(execution_providers: List[str]) -> List[str]:
@@ -243,7 +245,7 @@ def destroy(to_quit=True) -> None:
 
 
 def run() -> None:
-    parse_args()
+    args = parse_args()
     if not pre_check():
         return
     for frame_processor in get_frame_processors_modules(modules.globals.frame_processors):
@@ -253,5 +255,5 @@ def run() -> None:
     if modules.globals.headless:
         start()
     else:
-        window = ui.init(start, destroy)
+        window = ui.init(start, destroy, args.lang)
         window.mainloop()

modules/gettext.py

@@ -0,0 +1,29 @@
+import json
+import os
+from pathlib import Path
+
+class LanguageManager:
+    def __init__(self, default_language="en"):
+        self.current_language = default_language
+        self.translations = {}
+        self.load_language(default_language)
+    
+    def load_language(self, language_code):
+        """load language file"""
+        try:
+            file_path = Path(__file__).parent.parent / f"locales/{language_code}.json"
+            print(file_path)
+            with open(file_path, "r", encoding="utf-8") as file:
+                self.translations = json.load(file)
+            self.current_language = language_code
+            return True
+        except FileNotFoundError:
+            print(f"Language file not found: {language_code}")
+            return False
+    
+    def _(self, key, default=None):
+        """get translate text"""
+        return self.translations.get(key, default if default else key)
+    
+    def change_language(self, language_code):
+        return self.load_language(language_code)

modules/ui.py

@@ -26,11 +26,13 @@ from modules.utilities import (
     has_image_extension,
 )
 from modules.video_capture import VideoCapturer
+from modules.gettext import LanguageManager
 import platform
 
 if platform.system() == "Windows":
     from pygrabber.dshow_graph import FilterGraph
 
+
 ROOT = None
 POPUP = None
 POPUP_LIVE = None
@@ -63,6 +65,7 @@ RECENT_DIRECTORY_SOURCE = None
 RECENT_DIRECTORY_TARGET = None
 RECENT_DIRECTORY_OUTPUT = None
 
+_ = None
 preview_label = None
 preview_slider = None
 source_label = None
@@ -77,9 +80,11 @@ target_label_dict_live = {}
 img_ft, vid_ft = modules.globals.file_types
 
 
-def init(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.CTk:
+def init(start: Callable[[], None], destroy: Callable[[], None], lang: str) -> ctk.CTk:
-    global ROOT, PREVIEW
+    global ROOT, PREVIEW, _
 
+    lang_manager = LanguageManager(lang)
+    _ = lang_manager._
     ROOT = create_root(start, destroy)
     PREVIEW = create_preview(ROOT)
 
@@ -154,7 +159,7 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
     target_label.place(relx=0.6, rely=0.1, relwidth=0.3, relheight=0.25)
 
     select_face_button = ctk.CTkButton(
-        root, text="Select a face", cursor="hand2", command=lambda: select_source_path()
+        root, text=_("Select a face"), cursor="hand2", command=lambda: select_source_path()
     )
     select_face_button.place(relx=0.1, rely=0.4, relwidth=0.3, relheight=0.1)
 
@@ -165,7 +170,7 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
 
     select_target_button = ctk.CTkButton(
         root,
-        text="Select a target",
+        text=_("Select a target"),
         cursor="hand2",
         command=lambda: select_target_path(),
     )
@@ -174,7 +179,7 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
     keep_fps_value = ctk.BooleanVar(value=modules.globals.keep_fps)
     keep_fps_checkbox = ctk.CTkSwitch(
         root,
-        text="Keep fps",
+        text=_("Keep fps"),
         variable=keep_fps_value,
         cursor="hand2",
         command=lambda: (
@@ -187,7 +192,7 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
     keep_frames_value = ctk.BooleanVar(value=modules.globals.keep_frames)
     keep_frames_switch = ctk.CTkSwitch(
         root,
-        text="Keep frames",
+        text=_("Keep frames"),
         variable=keep_frames_value,
         cursor="hand2",
         command=lambda: (
@@ -200,7 +205,7 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
     enhancer_value = ctk.BooleanVar(value=modules.globals.fp_ui["face_enhancer"])
     enhancer_switch = ctk.CTkSwitch(
         root,
-        text="Face Enhancer",
+        text=_("Face Enhancer"),
         variable=enhancer_value,
         cursor="hand2",
         command=lambda: (
@@ -213,7 +218,7 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
     keep_audio_value = ctk.BooleanVar(value=modules.globals.keep_audio)
     keep_audio_switch = ctk.CTkSwitch(
         root,
-        text="Keep audio",
+        text=_("Keep audio"),
         variable=keep_audio_value,
         cursor="hand2",
         command=lambda: (
@@ -226,7 +231,7 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
     many_faces_value = ctk.BooleanVar(value=modules.globals.many_faces)
     many_faces_switch = ctk.CTkSwitch(
         root,
-        text="Many faces",
+        text=_("Many faces"),
         variable=many_faces_value,
         cursor="hand2",
         command=lambda: (
@@ -239,7 +244,7 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
     color_correction_value = ctk.BooleanVar(value=modules.globals.color_correction)
     color_correction_switch = ctk.CTkSwitch(
         root,
-        text="Fix Blueish Cam",
+        text=_("Fix Blueish Cam"),
         variable=color_correction_value,
         cursor="hand2",
         command=lambda: (
@@ -256,7 +261,7 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
     map_faces = ctk.BooleanVar(value=modules.globals.map_faces)
     map_faces_switch = ctk.CTkSwitch(
         root,
-        text="Map faces",
+        text=_("Map faces"),
         variable=map_faces,
         cursor="hand2",
         command=lambda: (
@@ -269,7 +274,7 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
     show_fps_value = ctk.BooleanVar(value=modules.globals.show_fps)
     show_fps_switch = ctk.CTkSwitch(
         root,
-        text="Show FPS",
+        text=_("Show FPS"),
         variable=show_fps_value,
         cursor="hand2",
         command=lambda: (
@@ -282,7 +287,7 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
     mouth_mask_var = ctk.BooleanVar(value=modules.globals.mouth_mask)
     mouth_mask_switch = ctk.CTkSwitch(
         root,
-        text="Mouth Mask",
+        text=_("Mouth Mask"),
         variable=mouth_mask_var,
         cursor="hand2",
         command=lambda: setattr(modules.globals, "mouth_mask", mouth_mask_var.get()),
@@ -292,7 +297,7 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
     show_mouth_mask_box_var = ctk.BooleanVar(value=modules.globals.show_mouth_mask_box)
     show_mouth_mask_box_switch = ctk.CTkSwitch(
         root,
-        text="Show Mouth Mask Box",
+        text=_("Show Mouth Mask Box"),
         variable=show_mouth_mask_box_var,
         cursor="hand2",
         command=lambda: setattr(
@@ -302,22 +307,22 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
     show_mouth_mask_box_switch.place(relx=0.6, rely=0.55)
 
     start_button = ctk.CTkButton(
-        root, text="Start", cursor="hand2", command=lambda: analyze_target(start, root)
+        root, text=_("Start"), cursor="hand2", command=lambda: analyze_target(start, root)
     )
     start_button.place(relx=0.15, rely=0.80, relwidth=0.2, relheight=0.05)
 
     stop_button = ctk.CTkButton(
-        root, text="Destroy", cursor="hand2", command=lambda: destroy()
+        root, text=_("Destroy"), cursor="hand2", command=lambda: destroy()
     )
     stop_button.place(relx=0.4, rely=0.80, relwidth=0.2, relheight=0.05)
 
     preview_button = ctk.CTkButton(
-        root, text="Preview", cursor="hand2", command=lambda: toggle_preview()
+        root, text=_("Preview"), cursor="hand2", command=lambda: toggle_preview()
     )
     preview_button.place(relx=0.65, rely=0.80, relwidth=0.2, relheight=0.05)
 
     # --- Camera Selection ---
-    camera_label = ctk.CTkLabel(root, text="Select Camera:")
+    camera_label = ctk.CTkLabel(root, text=_("Select Camera:"))
     camera_label.place(relx=0.1, rely=0.86, relwidth=0.2, relheight=0.05)
 
     available_cameras = get_available_cameras()
@@ -341,7 +346,7 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
 
     live_button = ctk.CTkButton(
         root,
-        text="Live",
+        text=_("Live"),
         cursor="hand2",
         command=lambda: webcam_preview(
             root,
@@ -475,7 +480,7 @@ def update_popup_source(
     global source_label_dict
 
     source_path = ctk.filedialog.askopenfilename(
-        title="select an source image",
+        title=_("select an source image"),
         initialdir=RECENT_DIRECTORY_SOURCE,
         filetypes=[img_ft],
     )
@@ -526,7 +531,7 @@ def create_preview(parent: ctk.CTkToplevel) -> ctk.CTkToplevel:
 
     preview = ctk.CTkToplevel(parent)
     preview.withdraw()
-    preview.title("Preview")
+    preview.title(_("Preview"))
     preview.configure()
     preview.protocol("WM_DELETE_WINDOW", lambda: toggle_preview())
     preview.resizable(width=True, height=True)
@@ -542,16 +547,16 @@ def create_preview(parent: ctk.CTkToplevel) -> ctk.CTkToplevel:
 
 
 def update_status(text: str) -> None:
-    status_label.configure(text=text)
+    status_label.configure(text=_(text))
     ROOT.update()
 
 
 def update_pop_status(text: str) -> None:
-    popup_status_label.configure(text=text)
+    popup_status_label.configure(text=_(text))
 
 
 def update_pop_live_status(text: str) -> None:
-    popup_status_label_live.configure(text=text)
+    popup_status_label_live.configure(text=_(text))
 
 
 def update_tumbler(var: str, value: bool) -> None:
@@ -570,7 +575,7 @@ def select_source_path() -> None:
 
     PREVIEW.withdraw()
     source_path = ctk.filedialog.askopenfilename(
-        title="select an source image",
+        title=_("select an source image"),
         initialdir=RECENT_DIRECTORY_SOURCE,
         filetypes=[img_ft],
     )
@@ -613,7 +618,7 @@ def select_target_path() -> None:
 
     PREVIEW.withdraw()
     target_path = ctk.filedialog.askopenfilename(
-        title="select an target image or video",
+        title=_("select an target image or video"),
         initialdir=RECENT_DIRECTORY_TARGET,
         filetypes=[img_ft, vid_ft],
     )
@@ -637,7 +642,7 @@ def select_output_path(start: Callable[[], None]) -> None:
 
     if is_image(modules.globals.target_path):
         output_path = ctk.filedialog.asksaveasfilename(
-            title="save image output file",
+            title=_("save image output file"),
             filetypes=[img_ft],
             defaultextension=".png",
             initialfile="output.png",
@@ -645,7 +650,7 @@ def select_output_path(start: Callable[[], None]) -> None:
         )
     elif is_video(modules.globals.target_path):
         output_path = ctk.filedialog.asksaveasfilename(
-            title="save video output file",
+            title=_("save video output file"),
             filetypes=[vid_ft],
             defaultextension=".mp4",
             initialfile="output.mp4",
@@ -960,11 +965,11 @@ def create_source_target_popup_for_webcam(
     popup_status_label_live = ctk.CTkLabel(POPUP_LIVE, text=None, justify="center")
     popup_status_label_live.grid(row=1, column=0, pady=15)
 
-    add_button = ctk.CTkButton(POPUP_LIVE, text="Add", command=lambda: on_add_click())
+    add_button = ctk.CTkButton(POPUP_LIVE, text=_("Add"), command=lambda: on_add_click())
     add_button.place(relx=0.2, rely=0.92, relwidth=0.2, relheight=0.05)
 
     close_button = ctk.CTkButton(
-        POPUP_LIVE, text="Submit", command=lambda: on_submit_click()
+        POPUP_LIVE, text=_("Submit"), command=lambda: on_submit_click()
     )
     close_button.place(relx=0.6, rely=0.92, relwidth=0.2, relheight=0.05)
 
@@ -988,7 +993,7 @@ def refresh_data(map: list):
 
         button = ctk.CTkButton(
             scrollable_frame,
-            text="Select source image",
+            text=_("Select source image"),
             command=lambda id=id: on_sbutton_click(map, id),
             width=DEFAULT_BUTTON_WIDTH,
             height=DEFAULT_BUTTON_HEIGHT,
@@ -1005,7 +1010,7 @@ def refresh_data(map: list):
 
         button = ctk.CTkButton(
             scrollable_frame,
-            text="Select target image",
+            text=_("Select target image"),
             command=lambda id=id: on_tbutton_click(map, id),
             width=DEFAULT_BUTTON_WIDTH,
             height=DEFAULT_BUTTON_HEIGHT,
@@ -1055,7 +1060,7 @@ def update_webcam_source(
     global source_label_dict_live
 
     source_path = ctk.filedialog.askopenfilename(
-        title="select an source image",
+        title=_("select an source image"),
         initialdir=RECENT_DIRECTORY_SOURCE,
         filetypes=[img_ft],
     )
@@ -1107,7 +1112,7 @@ def update_webcam_target(
     global target_label_dict_live
 
     target_path = ctk.filedialog.askopenfilename(
-        title="select an target image",
+        title=_("select an target image"),
         initialdir=RECENT_DIRECTORY_SOURCE,
         filetypes=[img_ft],
     )