// 履歴の最大保持数
const MaxHistory = 10;
// プロンプト履歴
let historyBox = (function () {
let _historyBox = [];
return {
push: function (prompt) {
if (prompt == _historyBox[_historyBox.length - 1]) return;
_historyBox.push(prompt);
if (MaxHistory < _historyBox.length) {
_historyBox.shift();
}
},
pop: function () {
let prePrompt = _historyBox.pop();
return prePrompt;
},
};
})();
// ネガティブプロンプト履歴
let nhistoryBox = (function () {
let _historyBox = [];
return {
push: function (prompt) {
if (prompt == _historyBox[_historyBox.length - 1]) return;
_historyBox.push(prompt);
if (MaxHistory < _historyBox.length) {
_historyBox.shift();
}
},
pop: function () {
let prePrompt = _historyBox.pop();
return prePrompt;
},
};
})();
function convert(input) {
const re_attention = /\{|\[|\}|\]|[^\{\}\[\]]+/gmu;
let text = input.replaceAll("(", "\\(").replaceAll(")", "\\)");
let res = [];
let curley_brackets = [];
let square_brackets = [];
const curley_bracket_multiplier = 1.05;
const square_bracket_multiplier = 1 / 1.05;
function multiply_range(start_position, multiplier) {
for (let pos = start_position; pos < res.length; pos++) {
res[pos][1] *= multiplier;
}
}
for (const match of text.matchAll(re_attention)) {
let word = match[0];
if (word == '{') {
curley_brackets.push(res.length);
} else if (word == '[') {
square_brackets.push(res.length);
} else if (word == '}' && curley_brackets.length > 0) {
multiply_range(curley_brackets.pop(), curley_bracket_multiplier);
} else if (word == ']' && square_brackets.length > 0) {
multiply_range(square_brackets.pop(), square_bracket_multiplier);
} else {
res.push([word, 1.0]);
}
}
for (const pos of curley_brackets) {
multiply_range(pos, curley_bracket_multiplier);
}
for (const pos of square_brackets) {
multiply_range(pos, square_bracket_multiplier);
}
if (res.length == 0) {
res = [["", 1.0]];
}
// console.log(res);
// merge runs of identical weights
let i = 0;
while (i+1 < res.length) {
// console.log("test:" + res[i] + " : " + res[i+1])
if (res[i][1] == res[i+1][1]) {
res[i][0] = res[i][0] + res[i+1][0];
// console.log("splicing:" + res[i+1]);
res.splice(i+1, 1);
} else {
i += 1;
}
}
// console.log(res);
let result = "";
for (let i = 0; i < res.length; i++) {
if (res[i][1] == 1.0) {
result += res[i][0];
} else {
result += "(" + res[i][0] + ":" + res[i][1].toString() + ")";
}
}
return result;
}
function onClickConvert() {
const default_prompt = "masterpiece, best quality,\n";
const default_negative = "lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry, artist name";
let result = "";
let prompt = gradioApp().querySelector("#txt2img_prompt > label > textarea");
historyBox.push(prompt.value);
result = convert(prompt.value);
if (result.length != 0) {
if (result.match(/^masterpiece, best quality,/) == null) {
result = default_prompt + result;
}
}
prompt.value = result;
prompt.dispatchEvent(new Event("input", { bubbles: true }));
result = "";
let negativep = gradioApp().querySelector("#txt2img_neg_prompt > label > textarea");
nhistoryBox.push(negativep.value);
result = convert(negativep.value);
if (result.length != 0) {
if (result.match(/^lowres,/) == null) {
result = default_negative + ",\n" + result;
}
} else {
result = default_negative;
}
negativep.value = result;
negativep.dispatchEvent(new Event("input", { bubbles: true }));
}
function onClickGenerate() {
let prompt = gradioApp().querySelector("#txt2img_prompt > label > textarea");
historyBox.push(prompt.value);
let negativep = gradioApp().querySelector("#txt2img_neg_prompt > label > textarea");
nhistoryBox.push(negativep.value);
}
function onClickUndo() {
let prompt = gradioApp().querySelector("#txt2img_prompt > label > textarea");
let prePrompt = historyBox.pop();
if (!prePrompt) {
prompt.value = "";
} else {
prompt.value = prePrompt;
}
prompt.dispatchEvent(new Event("input", { bubbles: true }));
let negativep = gradioApp().querySelector("#txt2img_neg_prompt > label > textarea");
let preNegativep = nhistoryBox.pop();
if (!preNegativep) {
negativep.value = "";
} else {
negativep.value = preNegativep;
}
negativep.dispatchEvent(new Event("input", { bubbles: true }));
}
onUiUpdate(function () {
let parentArea = gradioApp().querySelector("#toprow");
let generateBtn = gradioApp().querySelector("#txt2img_generate");
let beforeElement = gradioApp().querySelector("#roll_col");
if (parentArea == null || generateBtn == null || beforeElement == null) return;
if (gradioApp().querySelector("#nai2local") != null) return;
generateBtn.addEventListener("click", onClickGenerate);
let nai2LocalArea = document.createElement("div");
nai2LocalArea.id = "nai2local";
nai2LocalArea.className = "overflow-hidden flex flex-col relative col gap-4";
nai2LocalArea.style =
"min-width: min(110px, 100%); max-width: 120px; flex-grow: 1; padding: 0.4em; padding-bottom: 0.8em;";
let convertBtn = document.createElement("button");
convertBtn.id = "nai2localconvert";
convertBtn.type = "button";
convertBtn.innerHTML = "変換";
convertBtn.className = "gr-button gr-button-lg gr-button-secondary";
convertBtn.style =
"padding-left: 0.25em; padding-right: 0.25em; margin: 0.1em 0;max-height: 2em; max-width: 3em";
convertBtn.addEventListener("click", onClickConvert);
nai2LocalArea.appendChild(convertBtn);
let undoBtn = document.createElement("button");
undoBtn.id = "nai2localUndo";
undoBtn.type = "button";
undoBtn.innerHTML = "←";
undoBtn.className = "gr-button gr-button-lg gr-button-secondary";
undoBtn.style =
"padding-left: 0.25em; padding-right: 0.25em; margin: 0.1em 0;max-height: 2em; max-width: 3em";
undoBtn.addEventListener("click", onClickUndo);
nai2LocalArea.appendChild(undoBtn);
parentArea.insertBefore(nai2LocalArea, beforeElement.nextSibling);
});