diff --git a/gui.py b/gui.py index e4ea68bf..6332b3ca 100644 --- a/gui.py +++ b/gui.py @@ -30,6 +30,13 @@ def run_tensorboard(): subprocess.Popen([sys.executable, "-m", "tensorboard.main", "--logdir", "logs", "--host", args.tensorboard_host, "--port", str(args.tensorboard_port)]) +def get_locale_compat(): + lang = os.environ.get("LC_ALL") or os.environ.get("LC_CTYPE") or os.environ.get("LANG") + if lang: + lang = locale.normalize(lang).split('.')[0] + else: + lang = locale.getlocale()[0] or 'C' + return lang @catch_exception def run_tag_editor(): @@ -44,12 +51,11 @@ def run_tag_editor(): if args.localization: cmd.extend(["--localization", args.localization]) else: - l = locale.getdefaultlocale()[0] + l = get_locale_compat() if l and l.startswith("zh"): cmd.extend(["--localization", "zh-Hans"]) subprocess.Popen(cmd) - def launch(): log.info("Starting SD-Trainer Mikazuki GUI...") log.info(f"Base directory: {base_dir_path()}, Working directory: {os.getcwd()}")