File size: 2,932 Bytes
bf13828 d2b7e94 bf13828 d2b7e94 bf13828 ebc4336 bf13828 ebc4336 bf13828 ebc4336 53eba30 ebc4336 bf13828 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 |
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")
# current_translation = {k: 'XXX' for k in current_translation.keys()} # use this to see if all texts are covered
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
|