|
@@ -1,10 +1,9 @@
|
|
-let promptTokenCountDebounceTime = 800;
|
|
|
|
-let promptTokenCountTimeouts = {};
|
|
|
|
-var promptTokenCountUpdateFunctions = {};
|
|
|
|
|
|
+let promptTokenCountUpdateFunctions = {};
|
|
|
|
|
|
function update_txt2img_tokens(...args) {
|
|
function update_txt2img_tokens(...args) {
|
|
// Called from Gradio
|
|
// Called from Gradio
|
|
update_token_counter("txt2img_token_button");
|
|
update_token_counter("txt2img_token_button");
|
|
|
|
+ update_token_counter("txt2img_negative_token_button");
|
|
if (args.length == 2) {
|
|
if (args.length == 2) {
|
|
return args[0];
|
|
return args[0];
|
|
}
|
|
}
|
|
@@ -14,6 +13,7 @@ function update_txt2img_tokens(...args) {
|
|
function update_img2img_tokens(...args) {
|
|
function update_img2img_tokens(...args) {
|
|
// Called from Gradio
|
|
// Called from Gradio
|
|
update_token_counter("img2img_token_button");
|
|
update_token_counter("img2img_token_button");
|
|
|
|
+ update_token_counter("img2img_negative_token_button");
|
|
if (args.length == 2) {
|
|
if (args.length == 2) {
|
|
return args[0];
|
|
return args[0];
|
|
}
|
|
}
|
|
@@ -21,16 +21,7 @@ function update_img2img_tokens(...args) {
|
|
}
|
|
}
|
|
|
|
|
|
function update_token_counter(button_id) {
|
|
function update_token_counter(button_id) {
|
|
- if (opts.disable_token_counters) {
|
|
|
|
- return;
|
|
|
|
- }
|
|
|
|
- if (promptTokenCountTimeouts[button_id]) {
|
|
|
|
- clearTimeout(promptTokenCountTimeouts[button_id]);
|
|
|
|
- }
|
|
|
|
- promptTokenCountTimeouts[button_id] = setTimeout(
|
|
|
|
- () => gradioApp().getElementById(button_id)?.click(),
|
|
|
|
- promptTokenCountDebounceTime,
|
|
|
|
- );
|
|
|
|
|
|
+ promptTokenCountUpdateFunctions[button_id]?.();
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
@@ -69,10 +60,11 @@ function setupTokenCounting(id, id_counter, id_button) {
|
|
prompt.parentElement.insertBefore(counter, prompt);
|
|
prompt.parentElement.insertBefore(counter, prompt);
|
|
prompt.parentElement.style.position = "relative";
|
|
prompt.parentElement.style.position = "relative";
|
|
|
|
|
|
- promptTokenCountUpdateFunctions[id] = function() {
|
|
|
|
- update_token_counter(id_button);
|
|
|
|
- };
|
|
|
|
- textarea.addEventListener("input", promptTokenCountUpdateFunctions[id]);
|
|
|
|
|
|
+ func = onEdit(id, textarea, 800, function() {
|
|
|
|
+ gradioApp().getElementById(id_button)?.click();
|
|
|
|
+ });
|
|
|
|
+ promptTokenCountUpdateFunctions[id] = func;
|
|
|
|
+ promptTokenCountUpdateFunctions[id_button] = func;
|
|
}
|
|
}
|
|
|
|
|
|
function setupTokenCounters() {
|
|
function setupTokenCounters() {
|