Spaces:
Runtime error
Runtime error
// Stable Diffusion WebUI - Bracket checker | |
// By Hingashi no Florin/Bwin4L & @akx | |
// 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. | |
let promptCheckerInitialized = false; | |
function checkBrackets(textArea, counterElt) { | |
const counts = {}; | |
const errors = []; | |
function checkPair(open, close, kind) { | |
if (counts[open] !== counts[close]) errors.push(`${open}...${close} - Detected ${counts[open] || 0} opening and ${counts[close] || 0} closing ${kind}.`); | |
} | |
(textArea.value.match(/[(){}[\]]/g) || []).forEach((bracket) => { counts[bracket] = (counts[bracket] || 0) + 1; }); | |
checkPair('(', ')', 'round brackets'); | |
checkPair('[', ']', 'square brackets'); | |
checkPair('{', '}', 'curly brackets'); | |
counterElt.title = errors.join('\n'); | |
counterElt.classList.toggle('error', errors.length !== 0); | |
} | |
function setupBracketChecking(idPrompt, idCounter) { | |
const textarea = gradioApp().querySelector(`#${idPrompt} > label > textarea`); | |
const counter = gradioApp().getElementById(idCounter); | |
if (!textarea || !counter) return; | |
if (!promptCheckerInitialized) log('initPromptChecker'); | |
promptCheckerInitialized = true; | |
textarea.addEventListener('input', () => checkBrackets(textarea, counter)); | |
} | |
onAfterUiUpdate(() => { | |
if (promptCheckerInitialized) return; | |
setupBracketChecking('txt2img_prompt', 'txt2img_token_counter'); | |
setupBracketChecking('txt2img_neg_prompt', 'txt2img_negative_token_counter'); | |
setupBracketChecking('img2img_prompt', 'img2img_token_counter'); | |
setupBracketChecking('img2img_neg_prompt', 'img2img_negative_token_counter'); | |
setupBracketChecking('control_prompt', 'control_token_counter'); | |
setupBracketChecking('control_neg_prompt', 'control_negative_token_counter'); | |
}); | |