31 lines
1.6 KiB
Python
31 lines
1.6 KiB
Python
# This helper script loads the list of localization files and
|
|
# exposes the current localization file name and path to the javascript side
|
|
|
|
import gradio as gr
|
|
from pathlib import Path
|
|
from modules import localization, script_callbacks, shared
|
|
import json
|
|
|
|
# Webui root path
|
|
ROOT_DIR = Path().absolute()
|
|
|
|
# The localization files
|
|
I18N_DIRS = { k: str(Path(v).relative_to(ROOT_DIR).as_posix()) for k, v in localization.localizations.items() }
|
|
|
|
# Register extension options
|
|
def on_ui_settings():
|
|
BL_SECTION = ("bl", "Bilingual Localization")
|
|
# enable in settings
|
|
shared.opts.add_option("bilingual_localization_enabled", shared.OptionInfo(True, "Enable Bilingual Localization", section=BL_SECTION))
|
|
|
|
# current localization file
|
|
shared.opts.add_option("bilingual_localization_file", shared.OptionInfo("None", "Localization file (Please leave `User interface` - `Localization` as None)", gr.Dropdown, lambda: {"choices": ["None"] + list(localization.localizations.keys())}, refresh=lambda: localization.list_localizations(shared.cmd_opts.localizations_dir), section=BL_SECTION))
|
|
|
|
# translation order
|
|
shared.opts.add_option("bilingual_localization_order", shared.OptionInfo("Translation First", "Translation display order", gr.Radio, {"choices": ["Translation First", "Original First"]}, section=BL_SECTION))
|
|
|
|
# all localization files path in hidden option
|
|
shared.opts.add_option("bilingual_localization_dirs", shared.OptionInfo(json.dumps(I18N_DIRS), "Localization dirs", section=BL_SECTION, component_args={"visible": False}))
|
|
|
|
script_callbacks.on_ui_settings(on_ui_settings)
|