76 lines
2.3 KiB
JavaScript
76 lines
2.3 KiB
JavaScript
let promptTokenCountDebounceTime = 800;
|
|
let promptTokenCountTimeouts = {};
|
|
var promptTokenCountUpdateFunctions = {};
|
|
|
|
function update_txt2img_tokens(...args) {
|
|
// Called from Gradio
|
|
update_token_counter("txt2img_token_button");
|
|
if (args.length == 2) {
|
|
return args[0];
|
|
}
|
|
return args;
|
|
}
|
|
|
|
function update_img2img_tokens(...args) {
|
|
// Called from Gradio
|
|
update_token_counter("img2img_token_button");
|
|
if (args.length == 2) {
|
|
return args[0];
|
|
}
|
|
return args;
|
|
}
|
|
|
|
function update_token_counter(button_id) {
|
|
if (promptTokenCountTimeouts[button_id]) {
|
|
clearTimeout(promptTokenCountTimeouts[button_id]);
|
|
}
|
|
promptTokenCountTimeouts[button_id] = setTimeout(
|
|
() => gradioApp().getElementById(button_id)?.click(),
|
|
promptTokenCountDebounceTime,
|
|
);
|
|
}
|
|
|
|
|
|
function recalculatePromptTokens(name) {
|
|
promptTokenCountUpdateFunctions[name]?.();
|
|
}
|
|
|
|
function recalculate_prompts_txt2img() {
|
|
// Called from Gradio
|
|
recalculatePromptTokens('txt2img_prompt');
|
|
recalculatePromptTokens('txt2img_neg_prompt');
|
|
return Array.from(arguments);
|
|
}
|
|
|
|
function recalculate_prompts_img2img() {
|
|
// Called from Gradio
|
|
recalculatePromptTokens('img2img_prompt');
|
|
recalculatePromptTokens('img2img_neg_prompt');
|
|
return Array.from(arguments);
|
|
}
|
|
|
|
function setupTokenCounting(id, id_counter, id_button) {
|
|
var prompt = gradioApp().getElementById(id);
|
|
var counter = gradioApp().getElementById(id_counter);
|
|
var textarea = gradioApp().querySelector(`#${id} > label > textarea`);
|
|
|
|
if (counter.parentElement == prompt.parentElement) {
|
|
return;
|
|
}
|
|
|
|
prompt.parentElement.insertBefore(counter, prompt);
|
|
prompt.parentElement.style.position = "relative";
|
|
|
|
promptTokenCountUpdateFunctions[id] = function() {
|
|
update_token_counter(id_button);
|
|
};
|
|
textarea.addEventListener("input", promptTokenCountUpdateFunctions[id]);
|
|
}
|
|
|
|
function setupTokenCounters() {
|
|
setupTokenCounting('txt2img_prompt', 'txt2img_token_counter', 'txt2img_token_button');
|
|
setupTokenCounting('txt2img_neg_prompt', 'txt2img_negative_token_counter', 'txt2img_negative_token_button');
|
|
setupTokenCounting('img2img_prompt', 'img2img_token_counter', 'img2img_token_button');
|
|
setupTokenCounting('img2img_neg_prompt', 'img2img_negative_token_counter', 'img2img_negative_token_button');
|
|
}
|