76 lines
3 KiB
JavaScript
76 lines
3 KiB
JavaScript
addEventListener('keydown', (event) => {
|
|
let target = event.originalTarget || event.composedPath()[0];
|
|
if (!target.hasAttribute("placeholder")) return;
|
|
if (!target.placeholder.toLowerCase().includes("prompt")) return;
|
|
if (! (event.metaKey || event.ctrlKey)) return;
|
|
|
|
|
|
let plus = "ArrowUp"
|
|
let minus = "ArrowDown"
|
|
if (event.key != plus && event.key != minus) return;
|
|
|
|
let selectionStart = target.selectionStart;
|
|
let selectionEnd = target.selectionEnd;
|
|
// If the user hasn't selected anything, let's select their current parenthesis block
|
|
if (selectionStart === selectionEnd) {
|
|
// Find opening parenthesis around current cursor
|
|
const before = target.value.substring(0, selectionStart);
|
|
let beforeParen = before.lastIndexOf("(");
|
|
if (beforeParen == -1) return;
|
|
let beforeParenClose = before.lastIndexOf(")");
|
|
while (beforeParenClose !== -1 && beforeParenClose > beforeParen) {
|
|
beforeParen = before.lastIndexOf("(", beforeParen - 1);
|
|
beforeParenClose = before.lastIndexOf(")", beforeParenClose - 1);
|
|
}
|
|
|
|
// Find closing parenthesis around current cursor
|
|
const after = target.value.substring(selectionStart);
|
|
let afterParen = after.indexOf(")");
|
|
if (afterParen == -1) return;
|
|
let afterParenOpen = after.indexOf("(");
|
|
while (afterParenOpen !== -1 && afterParen > afterParenOpen) {
|
|
afterParen = after.indexOf(")", afterParen + 1);
|
|
afterParenOpen = after.indexOf("(", afterParenOpen + 1);
|
|
}
|
|
if (beforeParen === -1 || afterParen === -1) return;
|
|
|
|
// Set the selection to the text between the parenthesis
|
|
const parenContent = target.value.substring(beforeParen + 1, selectionStart + afterParen);
|
|
const lastColon = parenContent.lastIndexOf(":");
|
|
selectionStart = beforeParen + 1;
|
|
selectionEnd = selectionStart + lastColon;
|
|
target.setSelectionRange(selectionStart, selectionEnd);
|
|
}
|
|
|
|
event.preventDefault();
|
|
|
|
if (selectionStart == 0 || target.value[selectionStart - 1] != "(") {
|
|
target.value = target.value.slice(0, selectionStart) +
|
|
"(" + target.value.slice(selectionStart, selectionEnd) + ":1.0)" +
|
|
target.value.slice(selectionEnd);
|
|
|
|
target.focus();
|
|
target.selectionStart = selectionStart + 1;
|
|
target.selectionEnd = selectionEnd + 1;
|
|
|
|
} else {
|
|
end = target.value.slice(selectionEnd + 1).indexOf(")") + 1;
|
|
weight = parseFloat(target.value.slice(selectionEnd + 1, selectionEnd + 1 + end));
|
|
if (isNaN(weight)) return;
|
|
if (event.key == minus) weight -= 0.1;
|
|
if (event.key == plus) weight += 0.1;
|
|
|
|
weight = parseFloat(weight.toPrecision(12));
|
|
|
|
target.value = target.value.slice(0, selectionEnd + 1) +
|
|
weight +
|
|
target.value.slice(selectionEnd + 1 + end - 1);
|
|
|
|
target.focus();
|
|
target.selectionStart = selectionStart;
|
|
target.selectionEnd = selectionEnd;
|
|
}
|
|
// Since we've modified a Gradio Textbox component manually, we need to simulate an `input` DOM event to ensure its
|
|
// internal Svelte data binding remains in sync.
|
|
target.dispatchEvent(new Event("input", { bubbles: true }));
|
|
});
|