Spaces:
Running
Running
| import json | |
| import os | |
| import gradio as gr | |
| import logging | |
| 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 | |