localization support
AUTOMATIC
3 years ago
parent
695377a8b9
commit
cf47d13c1e
@ -0,0 +1,146 @@
|
|||||||
|
|
||||||
|
// localization = {} -- the dict with translations is created by the backend
|
||||||
|
|
||||||
|
ignore_ids_for_localization={
|
||||||
|
setting_sd_hypernetwork: 'OPTION',
|
||||||
|
setting_sd_model_checkpoint: 'OPTION',
|
||||||
|
setting_realesrgan_enabled_models: 'OPTION',
|
||||||
|
modelmerger_primary_model_name: 'OPTION',
|
||||||
|
modelmerger_secondary_model_name: 'OPTION',
|
||||||
|
modelmerger_tertiary_model_name: 'OPTION',
|
||||||
|
train_embedding: 'OPTION',
|
||||||
|
train_hypernetwork: 'OPTION',
|
||||||
|
txt2img_style_index: 'OPTION',
|
||||||
|
txt2img_style2_index: 'OPTION',
|
||||||
|
img2img_style_index: 'OPTION',
|
||||||
|
img2img_style2_index: 'OPTION',
|
||||||
|
setting_random_artist_categories: 'SPAN',
|
||||||
|
setting_face_restoration_model: 'SPAN',
|
||||||
|
setting_realesrgan_enabled_models: 'SPAN',
|
||||||
|
extras_upscaler_1: 'SPAN',
|
||||||
|
extras_upscaler_2: 'SPAN',
|
||||||
|
}
|
||||||
|
|
||||||
|
re_num = /^[\.\d]+$/
|
||||||
|
re_emoji = /[\p{Extended_Pictographic}\u{1F3FB}-\u{1F3FF}\u{1F9B0}-\u{1F9B3}]/u
|
||||||
|
|
||||||
|
original_lines = {}
|
||||||
|
translated_lines = {}
|
||||||
|
|
||||||
|
function textNodesUnder(el){
|
||||||
|
var n, a=[], walk=document.createTreeWalker(el,NodeFilter.SHOW_TEXT,null,false);
|
||||||
|
while(n=walk.nextNode()) a.push(n);
|
||||||
|
return a;
|
||||||
|
}
|
||||||
|
|
||||||
|
function canBeTranslated(node, text){
|
||||||
|
if(! text) return false;
|
||||||
|
if(! node.parentElement) return false;
|
||||||
|
|
||||||
|
parentType = node.parentElement.nodeName
|
||||||
|
if(parentType=='SCRIPT' || parentType=='STYLE' || parentType=='TEXTAREA') return false;
|
||||||
|
|
||||||
|
if (parentType=='OPTION' || parentType=='SPAN'){
|
||||||
|
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
|
||||||
|
}
|
||||||
|
|
||||||
|
tl = localization[text]
|
||||||
|
if(tl !== undefined){
|
||||||
|
translated_lines[tl] = 1
|
||||||
|
}
|
||||||
|
|
||||||
|
return tl
|
||||||
|
}
|
||||||
|
|
||||||
|
function processTextNode(node){
|
||||||
|
text = node.textContent.trim()
|
||||||
|
|
||||||
|
if(! canBeTranslated(node, text)) return
|
||||||
|
|
||||||
|
tl = getTranslation(text)
|
||||||
|
if(tl !== undefined){
|
||||||
|
node.textContent = tl
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function processNode(node){
|
||||||
|
if(node.nodeType == 3){
|
||||||
|
processTextNode(node)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if(node.title){
|
||||||
|
tl = getTranslation(node.title)
|
||||||
|
if(tl !== undefined){
|
||||||
|
node.title = tl
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if(node.placeholder){
|
||||||
|
tl = getTranslation(node.placeholder)
|
||||||
|
if(tl !== undefined){
|
||||||
|
node.placeholder = tl
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
textNodesUnder(node).forEach(function(node){
|
||||||
|
processTextNode(node)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
function dumpTranslations(){
|
||||||
|
dumped = {}
|
||||||
|
|
||||||
|
Object.keys(original_lines).forEach(function(text){
|
||||||
|
if(dumped[text] !== undefined) return
|
||||||
|
|
||||||
|
dumped[text] = localization[text] || text
|
||||||
|
})
|
||||||
|
|
||||||
|
return dumped
|
||||||
|
}
|
||||||
|
|
||||||
|
onUiUpdate(function(m){
|
||||||
|
m.forEach(function(mutation){
|
||||||
|
mutation.addedNodes.forEach(function(node){
|
||||||
|
processNode(node)
|
||||||
|
})
|
||||||
|
});
|
||||||
|
})
|
||||||
|
|
||||||
|
|
||||||
|
document.addEventListener("DOMContentLoaded", function() {
|
||||||
|
processNode(gradioApp())
|
||||||
|
})
|
||||||
|
|
||||||
|
function download_localization() {
|
||||||
|
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);
|
||||||
|
}
|
||||||
@ -0,0 +1,31 @@
|
|||||||
|
import json
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
import traceback
|
||||||
|
|
||||||
|
localizations = {}
|
||||||
|
|
||||||
|
|
||||||
|
def list_localizations(dirname):
|
||||||
|
localizations.clear()
|
||||||
|
|
||||||
|
for file in os.listdir(dirname):
|
||||||
|
fn, ext = os.path.splitext(file)
|
||||||
|
if ext.lower() != ".json":
|
||||||
|
continue
|
||||||
|
|
||||||
|
localizations[fn] = os.path.join(dirname, file)
|
||||||
|
|
||||||
|
|
||||||
|
def localization_js(current_localization_name):
|
||||||
|
fn = localizations.get(current_localization_name, None)
|
||||||
|
data = {}
|
||||||
|
if fn is not None:
|
||||||
|
try:
|
||||||
|
with open(fn, "r", encoding="utf8") as file:
|
||||||
|
data = json.load(file)
|
||||||
|
except Exception:
|
||||||
|
print(f"Error loading localization from {fn}:", file=sys.stderr)
|
||||||
|
print(traceback.format_exc(), file=sys.stderr)
|
||||||
|
|
||||||
|
return f"var localization = {json.dumps(data)}\n"
|
||||||
Loading…
Reference in New Issue