nodes/sn0w12-ComfyUI-Sn0w-Scripts
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Loras always highlighted by strength

Browse Source
This commit is contained in:
Lucas Sköld
2024-07-15 01:58:42 +02:00
parent 01ee8d7e59
commit 64e05cf73c
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
web/js/textbox.js
+10 -10
View File
@@ -209,7 +209,7 @@ app.registerExtension({
}
let color = colors[nestingLevel % colors.length];
const uniqueId = generateUniqueId();
let uniqueId = generateUniqueId();
if (inputEl.highlightGradient === true) {
color = `id-${uniqueId}`;
}
@@ -221,8 +221,10 @@ app.registerExtension({
lastIndex = i + 1;
break;
case '<':
highlightedText += text.slice(lastIndex, i) + `<span id="${uniqueId}" style="background-color: ${loraColor};">${escapeHtml(char)}`;
spanStack.push({ id: uniqueId, start: highlightedText.length - 3, originalSpan: `<span id="${uniqueId}" style="background-color: ${loraColor};">`, nestingLevel });
uniqueId = `${uniqueId}-lora`;
color = `id-${uniqueId}`;
highlightedText += text.slice(lastIndex, i) + `<span id="${uniqueId}" style="background-color: ${color};">${escapeHtml(char)}`;
spanStack.push({ id: uniqueId, start: highlightedText.length - 3, originalSpan: `<span id="${uniqueId}" style="background-color: ${color};">`, nestingLevel });
nestingLevel++;
lastIndex = i + 1;
break;
@@ -233,7 +235,7 @@ app.registerExtension({
const { id } = spanStack.pop();
nestingLevel--;
if (inputEl.highlightGradient === true) {
if (inputEl.highlightGradient === true || id.endsWith('lora')) {
// Check for the strength
const strengthText = text.slice(Math.max(0, i - 10), i);
const match = strengthText.match(/(\d+(\.\d+)?)\s*$/);
@@ -270,12 +272,10 @@ app.registerExtension({
}
}
if (inputEl.highlightGradient === true) {
// Apply the updated colors to the highlighted text
uniqueIdMap.forEach((newColor, id) => {
highlightedText = highlightedText.replace(`background-color: id-${id}`, `background-color: ${newColor}`);
});
}
// Apply the updated colors to the highlighted text
uniqueIdMap.forEach((newColor, id) => {
highlightedText = highlightedText.replace(`background-color: id-${id}`, `background-color: ${newColor}`);
});
overlayEl.innerHTML = highlightedText;
}