revert IIFE

This commit is contained in:
Haoming 2024-12-18 01:02:40 +08:00
parent fbc51fa210
commit 8bf30e3c42

View File

@ -3,14 +3,13 @@
// Counts open and closed brackets (round, square, curly) in the prompt and negative prompt text boxes in the txt2img and img2img tabs. // Counts open and closed brackets (round, square, curly) in the prompt and negative prompt text boxes in the txt2img and img2img tabs.
// If there's a mismatch, the keyword counter turns red, and if you hover on it, a tooltip tells you what's wrong. // If there's a mismatch, the keyword counter turns red, and if you hover on it, a tooltip tells you what's wrong.
(function() { function checkBrackets(textArea, counterElem) {
const pairs = [ const pairs = [
['(', ')', 'round brackets'], ['(', ')', 'round brackets'],
['[', ']', 'square brackets'], ['[', ']', 'square brackets'],
['{', '}', 'curly brackets'] ['{', '}', 'curly brackets']
]; ];
function checkBrackets(textArea, counterElem) {
const counts = {}; const counts = {};
const errors = new Set(); const errors = new Set();
let i = 0; let i = 0;
@ -57,21 +56,20 @@
counterElem.title = [...errors].join('\n'); counterElem.title = [...errors].join('\n');
counterElem.classList.toggle('error', errors.size !== 0); counterElem.classList.toggle('error', errors.size !== 0);
} }
function setupBracketChecking(id_prompt, id_counter) { function setupBracketChecking(id_prompt, id_counter) {
const textarea = gradioApp().querySelector(`#${id_prompt} > label > textarea`); const textarea = gradioApp().querySelector(`#${id_prompt} > label > textarea`);
const counter = gradioApp().getElementById(id_counter); const counter = gradioApp().getElementById(id_counter);
if (textarea && counter) { if (textarea && counter) {
onEdit(`${id_prompt}_BracketChecking`, textarea, 400, () => checkBrackets(textarea, counter)); onEdit(`${id_prompt}_BracketChecking`, textarea, 400, () => checkBrackets(textarea, counter));
} }
} }
onUiLoaded(function() { onUiLoaded(function() {
setupBracketChecking('txt2img_prompt', 'txt2img_token_counter'); setupBracketChecking('txt2img_prompt', 'txt2img_token_counter');
setupBracketChecking('txt2img_neg_prompt', 'txt2img_negative_token_counter'); setupBracketChecking('txt2img_neg_prompt', 'txt2img_negative_token_counter');
setupBracketChecking('img2img_prompt', 'img2img_token_counter'); setupBracketChecking('img2img_prompt', 'img2img_token_counter');
setupBracketChecking('img2img_neg_prompt', 'img2img_negative_token_counter'); setupBracketChecking('img2img_neg_prompt', 'img2img_negative_token_counter');
}); });
})();