|
import json |
|
import logging |
|
import os |
|
|
|
import gradio as gr |
|
|
|
logger = logging.getLogger(__name__) |
|
|
|
current_translation = {} |
|
localization_root = os.path.join( |
|
os.path.dirname(os.path.dirname(os.path.dirname(__file__))), "language" |
|
) |
|
|
|
|
|
def localization_js(filename): |
|
global current_translation |
|
|
|
if isinstance(filename, str): |
|
full_name = os.path.abspath(os.path.join(localization_root, filename + ".json")) |
|
if os.path.exists(full_name): |
|
try: |
|
with open(full_name, encoding="utf-8") as f: |
|
current_translation = json.load(f) |
|
assert isinstance(current_translation, dict) |
|
for k, v in current_translation.items(): |
|
assert isinstance(k, str), f"Key is not a string, got {k}" |
|
assert isinstance(v, str) or isinstance( |
|
v, list |
|
), f"Value for key {k} is not a string or list" |
|
|
|
logger.info(f"Loaded localization file {full_name}") |
|
except Exception as e: |
|
logger.warning(str(e)) |
|
logger.warning(f"Failed to load localization file {full_name}") |
|
else: |
|
logger.warning(f"Localization file {full_name} does not exist") |
|
else: |
|
logger.warning(f"Localization file {filename} is not a string") |
|
|
|
|
|
|
|
return f"window.localization = {json.dumps(current_translation)}" |
|
|
|
|
|
def dump_english_config(components): |
|
all_texts = [] |
|
for c in components: |
|
if isinstance(c, gr.Markdown) and "no-translate" in c.elem_classes: |
|
continue |
|
if isinstance(c, gr.Dropdown): |
|
continue |
|
if isinstance(c, gr.HTML): |
|
continue |
|
if isinstance(c, gr.Textbox): |
|
continue |
|
|
|
label = getattr(c, "label", None) |
|
value = getattr(c, "value", None) |
|
choices = getattr(c, "choices", None) |
|
info = getattr(c, "info", None) |
|
|
|
if isinstance(label, str): |
|
all_texts.append(label) |
|
if isinstance(value, str): |
|
all_texts.append(value) |
|
if isinstance(info, str): |
|
all_texts.append(info) |
|
if isinstance(choices, list): |
|
for x in choices: |
|
if isinstance(x, str): |
|
all_texts.append(x) |
|
if isinstance(x, tuple): |
|
for y in x: |
|
if isinstance(y, str): |
|
all_texts.append(y) |
|
|
|
config_dict = {k: k for k in all_texts if k != "" and "progress-container" not in k} |
|
full_name = os.path.abspath(os.path.join(localization_root, "en.json")) |
|
|
|
with open(full_name, "w", encoding="utf-8") as json_file: |
|
json.dump(config_dict, json_file, indent=4, ensure_ascii=False) |
|
|
|
return |
|
|
