1580158
/
sd-webui-prompt-format
mirror of https://github.com/Haoming02/sd-webui-prompt-format.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Minor Changes 

Yeet~
pull/4/head
Haoming 2023-06-02 12:18:38 +08:00
parent 61d8512c3a
commit 010b78662f
2 changed files with 37 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
LICENSE
View File

@ -1,6 +1,6 @@
MIT License
Copyright (c) 2023
Copyright (c) 2023 Haoming
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal

65
javascript/prompt_format.js
View File

@ -27,40 +27,40 @@ class LeFormatter {
return label
}
static injectBracketEscape(id) {
const textarea = gradioApp().getElementById(id).querySelector('textarea')
}
textarea.addEventListener('keydown', (event) => {
if (event.ctrlKey && event.key === '\\') {
event.preventDefault()
function injectBracketEscape(id) {
const textarea = gradioApp().getElementById(id).querySelector('textarea')
let cursorPosition = textarea.selectionStart;
textarea.addEventListener('keydown', (event) => {
if (event.ctrlKey && event.key === '\\') {
event.preventDefault()
if (textarea.selectionStart !== textarea.selectionEnd)
cursorPosition++
let cursorPosition = textarea.selectionStart;
let result = pf_GrabBrackets(textarea.value, cursorPosition)
if (textarea.selectionStart !== textarea.selectionEnd)
cursorPosition++
if (result) {
const original = textarea.value
let result = pf_GrabBrackets(textarea.value, cursorPosition)
if (result[0] !== 0 && textarea.value[result[0] - 1] === '\\' && textarea.value[result[1] - 1] === '\\') {
textarea.value = original.slice(0, result[0] - 1) + original.slice(result[0] - 1, result[1]).replace(/\\/g, '') + original.slice(result[1])
textarea.selectionStart = result[0] - 1
textarea.selectionEnd = result[1] - 1
}
else {
textarea.value = original.slice(0, result[0]) + '\\' + original.slice(result[0], result[1]) + '\\' + original.slice(result[1])
textarea.selectionStart = result[0]
textarea.selectionEnd = result[1] + 3
}
if (result) {
const original = textarea.value
updateInput(textarea)
if (result[0] !== 0 && textarea.value[result[0] - 1] === '\\' && textarea.value[result[1] - 1] === '\\') {
textarea.value = original.slice(0, result[0] - 1) + original.slice(result[0] - 1, result[1]).replace(/\\/g, '') + original.slice(result[1])
textarea.selectionStart = result[0] - 1
textarea.selectionEnd = result[1] - 1
}
else {
textarea.value = original.slice(0, result[0]) + '\\' + original.slice(result[0], result[1]) + '\\' + original.slice(result[1])
textarea.selectionStart = result[0]
textarea.selectionEnd = result[1] + 3
}
}
})
}
updateInput(textarea)
}
}
})
}
function pf_GrabBrackets(str, index) {
@ -118,6 +118,11 @@ function formatString(input, dedupe, deunderline) {
}
onUiLoaded(async () => {
// SETTINGS
const iterations = 1
// SETTINGS
let dedupe = false
let deunderline = false
@ -152,10 +157,12 @@ onUiLoaded(async () => {
let lines = [textAreas[0].value.split('\n'), textAreas[1].value.split('\n')]
for (let i = 0; i < lines[0].length; i++)
lines[0][i] = formatString(lines[0][i], dedupe, deunderline)
for (let it = 0; it < iterations; it++)
lines[0][i] = formatString(lines[0][i], dedupe, deunderline)
for (let i = 0; i < lines[1].length; i++)
lines[1][i] = formatString(lines[1][i], dedupe, deunderline)
for (let it = 0; it < iterations; it++)
lines[1][i] = formatString(lines[1][i], dedupe, deunderline)
textAreas[0].value = lines[0].join('\n')
@ -166,8 +173,8 @@ onUiLoaded(async () => {
}
})
LeFormatter.injectBracketEscape(mode + '2img_prompt')
LeFormatter.injectBracketEscape(mode + '2img_neg_prompt')
injectBracketEscape(mode + '2img_prompt')
injectBracketEscape(mode + '2img_neg_prompt')
})
})