StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Simplify finding word boundaries

Browse Source 
This also makes it work with prompts without spaces between words
This commit is contained in:
missionfloyd 2023-04-20 01:34:13 -06:00
parent fbd34a6847
commit ee172c0fc1
1 changed files with 9 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
javascript/edit-attention.js
View File

@ -48,24 +48,16 @@ function keyupEditAttention(event){
if (selectionStart !== selectionEnd) return false;
const delimiters = ".,\/#!$%\^&\*;:{}=\-_`~() ";
// Select the current word, find the start and end of the word (first space before and after)
const wordStart = text.substring(0, selectionStart).lastIndexOf(" ") + 1;
const wordEnd = text.substring(selectionEnd).indexOf(" ");
// If there is no space after the word, select to the end of the string
if (wordEnd === -1) {
selectionEnd = text.length;
} else {
selectionEnd += wordEnd;
// seek backward until to find beggining
while (!delimiters.includes(text[selectionStart - 1]) && selectionStart > 0) {
selectionStart--;
}
selectionStart = wordStart;
// Remove all punctuation at the end and beginning of the word
while (delimiters.includes(text[selectionStart])) {
selectionStart++;
}
while (delimiters.includes(text[selectionEnd - 1])) {
selectionEnd--;
// seek forward to find end
while (!delimiters.includes(text[selectionEnd]) && selectionEnd < text.length) {
selectionEnd++;
}
target.setSelectionRange(selectionStart, selectionEnd);
return true;
}