2023-05-17 15:46:58 +03:00
|
|
|
|
|
|
|
// localization = {} -- the dict with translations is created by the backend
|
|
|
|
|
2023-05-18 09:59:10 +03:00
|
|
|
var ignore_ids_for_localization = {
|
2023-05-17 15:46:58 +03:00
|
|
|
setting_sd_hypernetwork: 'OPTION',
|
|
|
|
setting_sd_model_checkpoint: 'OPTION',
|
|
|
|
modelmerger_primary_model_name: 'OPTION',
|
|
|
|
modelmerger_secondary_model_name: 'OPTION',
|
|
|
|
modelmerger_tertiary_model_name: 'OPTION',
|
|
|
|
train_embedding: 'OPTION',
|
|
|
|
train_hypernetwork: 'OPTION',
|
|
|
|
txt2img_styles: 'OPTION',
|
|
|
|
img2img_styles: 'OPTION',
|
|
|
|
setting_random_artist_categories: 'SPAN',
|
|
|
|
setting_face_restoration_model: 'SPAN',
|
|
|
|
setting_realesrgan_enabled_models: 'SPAN',
|
|
|
|
extras_upscaler_1: 'SPAN',
|
|
|
|
extras_upscaler_2: 'SPAN',
|
|
|
|
};
|
|
|
|
|
2023-05-18 09:59:10 +03:00
|
|
|
var re_num = /^[.\d]+$/;
|
|
|
|
var re_emoji = /[\p{Extended_Pictographic}\u{1F3FB}-\u{1F3FF}\u{1F9B0}-\u{1F9B3}]/u;
|
2023-05-17 15:46:58 +03:00
|
|
|
|
2023-05-18 09:59:10 +03:00
|
|
|
var original_lines = {};
|
|
|
|
var translated_lines = {};
|
2023-05-17 15:46:58 +03:00
|
|
|
|
|
|
|
function hasLocalization() {
|
|
|
|
return window.localization && Object.keys(window.localization).length > 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
function textNodesUnder(el) {
|
|
|
|
var n, a = [], walk = document.createTreeWalker(el, NodeFilter.SHOW_TEXT, null, false);
|
2023-05-18 09:59:10 +03:00
|
|
|
while ((n = walk.nextNode())) a.push(n);
|
2023-05-17 15:46:58 +03:00
|
|
|
return a;
|
|
|
|
}
|
|
|
|
|
|
|
|
function canBeTranslated(node, text) {
|
|
|
|
if (!text) return false;
|
|
|
|
if (!node.parentElement) return false;
|
|
|
|
|
|
|
|
var parentType = node.parentElement.nodeName;
|
|
|
|
if (parentType == 'SCRIPT' || parentType == 'STYLE' || parentType == 'TEXTAREA') return false;
|
|
|
|
|
|
|
|
if (parentType == 'OPTION' || parentType == 'SPAN') {
|
|
|
|
var pnode = node;
|
|
|
|
for (var level = 0; level < 4; level++) {
|
|
|
|
pnode = pnode.parentElement;
|
|
|
|
if (!pnode) break;
|
|
|
|
|
|
|
|
if (ignore_ids_for_localization[pnode.id] == parentType) return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (re_num.test(text)) return false;
|
|
|
|
if (re_emoji.test(text)) return false;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
function getTranslation(text) {
|
|
|
|
if (!text) return undefined;
|
|
|
|
|
|
|
|
if (translated_lines[text] === undefined) {
|
|
|
|
original_lines[text] = 1;
|
|
|
|
}
|
|
|
|
|
2023-05-18 09:59:10 +03:00
|
|
|
var tl = localization[text];
|
2023-05-17 15:46:58 +03:00
|
|
|
if (tl !== undefined) {
|
|
|
|
translated_lines[tl] = 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
return tl;
|
|
|
|
}
|
|
|
|
|
|
|
|
function processTextNode(node) {
|
|
|
|
var text = node.textContent.trim();
|
|
|
|
|
|
|
|
if (!canBeTranslated(node, text)) return;
|
|
|
|
|
2023-05-18 09:59:10 +03:00
|
|
|
var tl = getTranslation(text);
|
2023-05-17 15:46:58 +03:00
|
|
|
if (tl !== undefined) {
|
|
|
|
node.textContent = tl;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
function processNode(node) {
|
|
|
|
if (node.nodeType == 3) {
|
|
|
|
processTextNode(node);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (node.title) {
|
2023-05-18 09:59:10 +03:00
|
|
|
let tl = getTranslation(node.title);
|
2023-05-17 15:46:58 +03:00
|
|
|
if (tl !== undefined) {
|
|
|
|
node.title = tl;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (node.placeholder) {
|
2023-05-18 09:59:10 +03:00
|
|
|
let tl = getTranslation(node.placeholder);
|
2023-05-17 15:46:58 +03:00
|
|
|
if (tl !== undefined) {
|
|
|
|
node.placeholder = tl;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
textNodesUnder(node).forEach(function(node) {
|
|
|
|
processTextNode(node);
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
function dumpTranslations() {
|
|
|
|
if (!hasLocalization()) {
|
|
|
|
// If we don't have any localization,
|
|
|
|
// we will not have traversed the app to find
|
|
|
|
// original_lines, so do that now.
|
|
|
|
processNode(gradioApp());
|
|
|
|
}
|
|
|
|
var dumped = {};
|
|
|
|
if (localization.rtl) {
|
|
|
|
dumped.rtl = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
for (const text in original_lines) {
|
|
|
|
if (dumped[text] !== undefined) continue;
|
|
|
|
dumped[text] = localization[text] || text;
|
|
|
|
}
|
|
|
|
|
|
|
|
return dumped;
|
|
|
|
}
|
|
|
|
|
|
|
|
function download_localization() {
|
|
|
|
var text = JSON.stringify(dumpTranslations(), null, 4);
|
|
|
|
|
|
|
|
var element = document.createElement('a');
|
|
|
|
element.setAttribute('href', 'data:text/plain;charset=utf-8,' + encodeURIComponent(text));
|
|
|
|
element.setAttribute('download', "localization.json");
|
|
|
|
element.style.display = 'none';
|
|
|
|
document.body.appendChild(element);
|
|
|
|
|
|
|
|
element.click();
|
|
|
|
|
|
|
|
document.body.removeChild(element);
|
|
|
|
}
|
|
|
|
|
|
|
|
document.addEventListener("DOMContentLoaded", function() {
|
|
|
|
if (!hasLocalization()) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
onUiUpdate(function(m) {
|
|
|
|
m.forEach(function(mutation) {
|
|
|
|
mutation.addedNodes.forEach(function(node) {
|
|
|
|
processNode(node);
|
|
|
|
});
|
|
|
|
});
|
|
|
|
});
|
|
|
|
|
|
|
|
processNode(gradioApp());
|
|
|
|
|
2023-05-18 09:59:10 +03:00
|
|
|
if (localization.rtl) { // if the language is from right to left,
|
2023-05-17 15:46:58 +03:00
|
|
|
(new MutationObserver((mutations, observer) => { // wait for the style to load
|
|
|
|
mutations.forEach(mutation => {
|
|
|
|
mutation.addedNodes.forEach(node => {
|
|
|
|
if (node.tagName === 'STYLE') {
|
|
|
|
observer.disconnect();
|
|
|
|
|
2023-05-18 09:59:10 +03:00
|
|
|
for (const x of node.sheet.rules) { // find all rtl media rules
|
2023-05-17 15:46:58 +03:00
|
|
|
if (Array.from(x.media || []).includes('rtl')) {
|
2023-05-18 09:59:10 +03:00
|
|
|
x.media.appendMedium('all'); // enable them
|
2023-05-17 15:46:58 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
});
|
|
|
|
});
|
2023-05-18 09:59:10 +03:00
|
|
|
})).observe(gradioApp(), {childList: true});
|
2023-05-17 15:46:58 +03:00
|
|
|
}
|
|
|
|
});
|