493 lines
16 KiB
JavaScript
493 lines
16 KiB
JavaScript
import {
|
|
chat_metadata,
|
|
substituteParams,
|
|
this_chid,
|
|
eventSource,
|
|
event_types,
|
|
saveSettingsDebounced,
|
|
} from '../script.js';
|
|
import { extension_settings, saveMetadataDebounced } from './extensions.js';
|
|
import { selected_group } from './group-chats.js';
|
|
import { getCharaFilename, delay } from './utils.js';
|
|
import { power_user } from './power-user.js';
|
|
|
|
const extensionName = 'cfg';
|
|
const defaultSettings = {
|
|
global: {
|
|
'guidance_scale': 1,
|
|
'negative_prompt': ''
|
|
},
|
|
chara: []
|
|
};
|
|
const settingType = {
|
|
guidance_scale: 0,
|
|
negative_prompt: 1,
|
|
positive_prompt: 2
|
|
};
|
|
|
|
// Used for character and chat CFG values
|
|
function updateSettings() {
|
|
saveSettingsDebounced();
|
|
loadSettings();
|
|
}
|
|
|
|
function setCharCfg(tempValue, setting) {
|
|
const avatarName = getCharaFilename();
|
|
|
|
// Assign temp object
|
|
let tempCharaCfg = {
|
|
name: avatarName
|
|
};
|
|
|
|
switch(setting) {
|
|
case settingType.guidance_scale:
|
|
tempCharaCfg['guidance_scale'] = Number(tempValue);
|
|
break;
|
|
case settingType.negative_prompt:
|
|
tempCharaCfg['negative_prompt'] = tempValue;
|
|
break;
|
|
case settingType.positive_prompt:
|
|
tempCharaCfg['positive_prompt'] = tempValue;
|
|
break;
|
|
default:
|
|
return false;
|
|
}
|
|
|
|
let existingCharaCfgIndex;
|
|
let existingCharaCfg;
|
|
|
|
if (extension_settings.cfg.chara) {
|
|
existingCharaCfgIndex = extension_settings.cfg.chara.findIndex((e) => e.name === avatarName);
|
|
existingCharaCfg = extension_settings.cfg.chara[existingCharaCfgIndex];
|
|
}
|
|
|
|
if (extension_settings.cfg.chara && existingCharaCfg) {
|
|
const tempAssign = Object.assign(existingCharaCfg, tempCharaCfg);
|
|
|
|
// If both values are default, remove the entry
|
|
if (!existingCharaCfg.useChara &&
|
|
(tempAssign.guidance_scale ?? 1.00) === 1.00 &&
|
|
(tempAssign.negative_prompt?.length ?? 0) === 0 &&
|
|
(tempAssign.positive_prompt?.length ?? 0) === 0)
|
|
{
|
|
extension_settings.cfg.chara.splice(existingCharaCfgIndex, 1);
|
|
}
|
|
} else if (avatarName && tempValue.length > 0) {
|
|
if (!extension_settings.cfg.chara) {
|
|
extension_settings.cfg.chara = [];
|
|
}
|
|
|
|
extension_settings.cfg.chara.push(tempCharaCfg);
|
|
} else {
|
|
console.debug('Character CFG error: No avatar name key could be found.');
|
|
|
|
// Don't save settings if something went wrong
|
|
return false;
|
|
}
|
|
|
|
updateSettings();
|
|
|
|
return true;
|
|
}
|
|
|
|
function setChatCfg(tempValue, setting) {
|
|
switch(setting) {
|
|
case settingType.guidance_scale:
|
|
chat_metadata[metadataKeys.guidance_scale] = tempValue;
|
|
break;
|
|
case settingType.negative_prompt:
|
|
chat_metadata[metadataKeys.negative_prompt] = tempValue;
|
|
break;
|
|
case settingType.positive_prompt:
|
|
chat_metadata[metadataKeys.positive_prompt] = tempValue;
|
|
break;
|
|
default:
|
|
return false;
|
|
}
|
|
|
|
saveMetadataDebounced();
|
|
|
|
return true;
|
|
}
|
|
|
|
// TODO: Only change CFG when character is selected
|
|
function onCfgMenuItemClick() {
|
|
if (selected_group || this_chid) {
|
|
//show CFG config if it's hidden
|
|
if ($('#cfgConfig').css('display') !== 'flex') {
|
|
$('#cfgConfig').addClass('resizing');
|
|
$('#cfgConfig').css('display', 'flex');
|
|
$('#cfgConfig').css('opacity', 0.0);
|
|
$('#cfgConfig').transition({
|
|
opacity: 1.0,
|
|
duration: 250,
|
|
}, async function () {
|
|
await delay(50);
|
|
$('#cfgConfig').removeClass('resizing');
|
|
});
|
|
|
|
//auto-open the main AN inline drawer
|
|
if ($('#CFGBlockToggle')
|
|
.siblings('.inline-drawer-content')
|
|
.css('display') !== 'block') {
|
|
$('#floatingPrompt').addClass('resizing');
|
|
$('#CFGBlockToggle').click();
|
|
}
|
|
} else {
|
|
//hide AN if it's already displayed
|
|
$('#cfgConfig').addClass('resizing');
|
|
$('#cfgConfig').transition({
|
|
opacity: 0.0,
|
|
duration: 250,
|
|
},
|
|
async function () {
|
|
await delay(50);
|
|
$('#cfgConfig').removeClass('resizing');
|
|
});
|
|
setTimeout(function () {
|
|
$('#cfgConfig').hide();
|
|
}, 250);
|
|
|
|
}
|
|
//duplicate options menu close handler from script.js
|
|
//because this listener takes priority
|
|
$('#options').stop().fadeOut(250);
|
|
} else {
|
|
toastr.warning('Select a character before trying to configure CFG', '', { timeOut: 2000 });
|
|
}
|
|
}
|
|
|
|
async function onChatChanged() {
|
|
loadSettings();
|
|
await modifyCharaHtml();
|
|
}
|
|
|
|
// Rearrange the panel if a group chat is present
|
|
async function modifyCharaHtml() {
|
|
if (selected_group) {
|
|
$('#chara_cfg_container').hide();
|
|
$('#groupchat_cfg_use_chara_container').show();
|
|
} else {
|
|
$('#chara_cfg_container').show();
|
|
$('#groupchat_cfg_use_chara_container').hide();
|
|
// TODO: Remove chat checkbox here
|
|
}
|
|
}
|
|
|
|
// Reloads chat-specific settings
|
|
function loadSettings() {
|
|
// Set chat CFG if it exists
|
|
$('#chat_cfg_guidance_scale').val(chat_metadata[metadataKeys.guidance_scale] ?? 1.0.toFixed(2));
|
|
$('#chat_cfg_guidance_scale_counter').val(chat_metadata[metadataKeys.guidance_scale]?.toFixed(2) ?? 1.0.toFixed(2));
|
|
$('#chat_cfg_negative_prompt').val(chat_metadata[metadataKeys.negative_prompt] ?? '');
|
|
$('#chat_cfg_positive_prompt').val(chat_metadata[metadataKeys.positive_prompt] ?? '');
|
|
$('#groupchat_cfg_use_chara').prop('checked', chat_metadata[metadataKeys.groupchat_individual_chars] ?? false);
|
|
if (chat_metadata[metadataKeys.prompt_combine]?.length > 0) {
|
|
chat_metadata[metadataKeys.prompt_combine].forEach((element) => {
|
|
$(`input[name="cfg_prompt_combine"][value="${element}"]`)
|
|
.prop('checked', true);
|
|
});
|
|
}
|
|
|
|
// Display the negative separator in quotes if not quoted already
|
|
let promptSeparatorDisplay = [];
|
|
const promptSeparator = chat_metadata[metadataKeys.prompt_separator];
|
|
if (promptSeparator) {
|
|
promptSeparatorDisplay.push(promptSeparator);
|
|
if (!promptSeparator.startsWith('"')) {
|
|
promptSeparatorDisplay.unshift('"');
|
|
}
|
|
|
|
if (!promptSeparator.endsWith('"')) {
|
|
promptSeparatorDisplay.push('"');
|
|
}
|
|
}
|
|
|
|
$('#cfg_prompt_separator').val(promptSeparatorDisplay.length === 0 ? '' : promptSeparatorDisplay.join(''));
|
|
|
|
$('#cfg_prompt_insertion_depth').val(chat_metadata[metadataKeys.prompt_insertion_depth] ?? 1);
|
|
|
|
// Set character CFG if it exists
|
|
if (!selected_group) {
|
|
const charaCfg = extension_settings.cfg.chara.find((e) => e.name === getCharaFilename());
|
|
$('#chara_cfg_guidance_scale').val(charaCfg?.guidance_scale ?? 1.00);
|
|
$('#chara_cfg_guidance_scale_counter').val(charaCfg?.guidance_scale?.toFixed(2) ?? 1.0.toFixed(2));
|
|
$('#chara_cfg_negative_prompt').val(charaCfg?.negative_prompt ?? '');
|
|
$('#chara_cfg_positive_prompt').val(charaCfg?.positive_prompt ?? '');
|
|
}
|
|
}
|
|
|
|
// Load initial extension settings
|
|
async function initialLoadSettings() {
|
|
// Create the settings if they don't exist
|
|
extension_settings[extensionName] = extension_settings[extensionName] || {};
|
|
if (Object.keys(extension_settings[extensionName]).length === 0) {
|
|
Object.assign(extension_settings[extensionName], defaultSettings);
|
|
saveSettingsDebounced();
|
|
}
|
|
|
|
// Set global CFG values on load
|
|
$('#global_cfg_guidance_scale').val(extension_settings.cfg.global.guidance_scale);
|
|
$('#global_cfg_guidance_scale_counter').val(extension_settings.cfg.global.guidance_scale.toFixed(2));
|
|
$('#global_cfg_negative_prompt').val(extension_settings.cfg.global.negative_prompt);
|
|
$('#global_cfg_positive_prompt').val(extension_settings.cfg.global.positive_prompt);
|
|
}
|
|
|
|
function migrateSettings() {
|
|
let performSettingsSave = false;
|
|
let performMetaSave = false;
|
|
|
|
if (power_user.guidance_scale) {
|
|
extension_settings.cfg.global.guidance_scale = power_user.guidance_scale;
|
|
delete power_user['guidance_scale'];
|
|
performSettingsSave = true;
|
|
}
|
|
|
|
if (power_user.negative_prompt) {
|
|
extension_settings.cfg.global.negative_prompt = power_user.negative_prompt;
|
|
delete power_user['negative_prompt'];
|
|
performSettingsSave = true;
|
|
}
|
|
|
|
if (chat_metadata['cfg_negative_combine']) {
|
|
chat_metadata[metadataKeys.prompt_combine] = chat_metadata['cfg_negative_combine'];
|
|
chat_metadata['cfg_negative_combine'] = undefined;
|
|
performMetaSave = true;
|
|
}
|
|
|
|
if (chat_metadata['cfg_negative_insertion_depth']) {
|
|
chat_metadata[metadataKeys.prompt_insertion_depth] = chat_metadata['cfg_negative_insertion_depth'];
|
|
chat_metadata['cfg_negative_insertion_depth'] = undefined;
|
|
performMetaSave = true;
|
|
}
|
|
|
|
if (chat_metadata['cfg_negative_separator']) {
|
|
chat_metadata[metadataKeys.prompt_separator] = chat_metadata['cfg_negative_separator'];
|
|
chat_metadata['cfg_negative_separator'] = undefined;
|
|
performMetaSave = true;
|
|
}
|
|
|
|
if (performSettingsSave) {
|
|
saveSettingsDebounced();
|
|
}
|
|
|
|
if (performMetaSave) {
|
|
saveMetadataDebounced();
|
|
}
|
|
}
|
|
|
|
// This function is called when the extension is loaded
|
|
export function initCfg() {
|
|
$('#CFGClose').on('click', function () {
|
|
$('#cfgConfig').transition({
|
|
opacity: 0,
|
|
duration: 200,
|
|
easing: 'ease-in-out',
|
|
});
|
|
setTimeout(function () { $('#cfgConfig').hide(); }, 200);
|
|
});
|
|
|
|
$('#chat_cfg_guidance_scale').on('input', function() {
|
|
const numberValue = Number($(this).val());
|
|
const success = setChatCfg(numberValue, settingType.guidance_scale);
|
|
if (success) {
|
|
$('#chat_cfg_guidance_scale_counter').val(numberValue.toFixed(2));
|
|
}
|
|
});
|
|
|
|
$('#chat_cfg_negative_prompt').on('input', function() {
|
|
setChatCfg($(this).val(), settingType.negative_prompt);
|
|
});
|
|
|
|
$('#chat_cfg_positive_prompt').on('input', function() {
|
|
setChatCfg($(this).val(), settingType.positive_prompt);
|
|
});
|
|
|
|
$('#chara_cfg_guidance_scale').on('input', function() {
|
|
const value = $(this).val();
|
|
const success = setCharCfg(value, settingType.guidance_scale);
|
|
if (success) {
|
|
$('#chara_cfg_guidance_scale_counter').val(Number(value).toFixed(2));
|
|
}
|
|
});
|
|
|
|
$('#chara_cfg_negative_prompt').on('input', function() {
|
|
setCharCfg($(this).val(), settingType.negative_prompt);
|
|
});
|
|
|
|
$('#chara_cfg_positive_prompt').on('input', function() {
|
|
setCharCfg($(this).val(), settingType.positive_prompt);
|
|
});
|
|
|
|
$('#global_cfg_guidance_scale').on('input', function() {
|
|
extension_settings.cfg.global.guidance_scale = Number($(this).val());
|
|
$('#global_cfg_guidance_scale_counter').val(extension_settings.cfg.global.guidance_scale.toFixed(2));
|
|
saveSettingsDebounced();
|
|
});
|
|
|
|
$('#global_cfg_negative_prompt').on('input', function() {
|
|
extension_settings.cfg.global.negative_prompt = $(this).val();
|
|
saveSettingsDebounced();
|
|
});
|
|
|
|
$('#global_cfg_positive_prompt').on('input', function() {
|
|
extension_settings.cfg.global.positive_prompt = $(this).val();
|
|
saveSettingsDebounced();
|
|
});
|
|
|
|
$('input[name="cfg_prompt_combine"]').on('input', function() {
|
|
const values = $('#cfgConfig').find('input[name="cfg_prompt_combine"]')
|
|
.filter(':checked')
|
|
.map(function() { return Number($(this).val()); })
|
|
.get()
|
|
.filter((e) => !Number.isNaN(e)) || [];
|
|
|
|
chat_metadata[metadataKeys.prompt_combine] = values;
|
|
saveMetadataDebounced();
|
|
});
|
|
|
|
$('#cfg_prompt_insertion_depth').on('input', function() {
|
|
chat_metadata[metadataKeys.prompt_insertion_depth] = Number($(this).val());
|
|
saveMetadataDebounced();
|
|
});
|
|
|
|
$('#cfg_prompt_separator').on('input', function() {
|
|
chat_metadata[metadataKeys.prompt_separator] = $(this).val();
|
|
saveMetadataDebounced();
|
|
});
|
|
|
|
$('#groupchat_cfg_use_chara').on('input', function() {
|
|
const checked = !!$(this).prop('checked');
|
|
chat_metadata[metadataKeys.groupchat_individual_chars] = checked;
|
|
|
|
if (checked) {
|
|
toastr.info('You can edit character CFG values in their respective character chats.');
|
|
}
|
|
|
|
saveMetadataDebounced();
|
|
});
|
|
|
|
initialLoadSettings();
|
|
|
|
if (extension_settings.cfg) {
|
|
migrateSettings();
|
|
}
|
|
|
|
$('#option_toggle_CFG').on('click', onCfgMenuItemClick);
|
|
|
|
// Hook events
|
|
eventSource.on(event_types.CHAT_CHANGED, async () => {
|
|
await onChatChanged();
|
|
});
|
|
}
|
|
|
|
export const cfgType = {
|
|
chat: 0,
|
|
chara: 1,
|
|
global: 2
|
|
};
|
|
|
|
export const metadataKeys = {
|
|
guidance_scale: 'cfg_guidance_scale',
|
|
negative_prompt: 'cfg_negative_prompt',
|
|
positive_prompt: 'cfg_positive_prompt',
|
|
prompt_combine: 'cfg_prompt_combine',
|
|
groupchat_individual_chars: 'cfg_groupchat_individual_chars',
|
|
prompt_insertion_depth: 'cfg_prompt_insertion_depth',
|
|
prompt_separator: 'cfg_prompt_separator'
|
|
};
|
|
|
|
// Gets the CFG guidance scale
|
|
// If the guidance scale is 1, ignore the CFG prompt(s) since it won't be used anyways
|
|
export function getGuidanceScale() {
|
|
if (!extension_settings.cfg) {
|
|
console.warn('CFG extension is not enabled. Skipping CFG guidance.');
|
|
return;
|
|
}
|
|
|
|
const charaCfg = extension_settings.cfg.chara?.find((e) => e.name === getCharaFilename(this_chid));
|
|
const chatGuidanceScale = chat_metadata[metadataKeys.guidance_scale];
|
|
const groupchatCharOverride = chat_metadata[metadataKeys.groupchat_individual_chars] ?? false;
|
|
|
|
if (chatGuidanceScale && chatGuidanceScale !== 1 && !groupchatCharOverride) {
|
|
return {
|
|
type: cfgType.chat,
|
|
value: chatGuidanceScale
|
|
};
|
|
}
|
|
|
|
if ((!selected_group && charaCfg || groupchatCharOverride) && charaCfg?.guidance_scale !== 1) {
|
|
return {
|
|
type: cfgType.chara,
|
|
value: charaCfg.guidance_scale
|
|
};
|
|
}
|
|
|
|
if (extension_settings.cfg.global && extension_settings.cfg.global?.guidance_scale !== 1) {
|
|
return {
|
|
type: cfgType.global,
|
|
value: extension_settings.cfg.global.guidance_scale
|
|
};
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Gets the CFG prompt separator.
|
|
* @returns {string} The CFG prompt separator
|
|
*/
|
|
function getCustomSeparator() {
|
|
const defaultSeparator = '\n';
|
|
|
|
try {
|
|
if (chat_metadata[metadataKeys.prompt_separator]) {
|
|
return JSON.parse(chat_metadata[metadataKeys.prompt_separator]);
|
|
}
|
|
|
|
return defaultSeparator;
|
|
} catch {
|
|
console.warn('Invalid JSON detected for prompt separator. Using default separator.');
|
|
return defaultSeparator;
|
|
}
|
|
}
|
|
|
|
// Gets the CFG prompt
|
|
export function getCfgPrompt(guidanceScale, isNegative) {
|
|
let splitCfgPrompt = [];
|
|
|
|
const cfgPromptCombine = chat_metadata[metadataKeys.prompt_combine] ?? [];
|
|
if (guidanceScale.type === cfgType.chat || cfgPromptCombine.includes(cfgType.chat)) {
|
|
splitCfgPrompt.unshift(
|
|
substituteParams(
|
|
chat_metadata[isNegative ? metadataKeys.negative_prompt : metadataKeys.positive_prompt]
|
|
)
|
|
);
|
|
}
|
|
|
|
const charaCfg = extension_settings.cfg.chara?.find((e) => e.name === getCharaFilename(this_chid));
|
|
if (guidanceScale.type === cfgType.chara || cfgPromptCombine.includes(cfgType.chara)) {
|
|
splitCfgPrompt.unshift(
|
|
substituteParams(
|
|
isNegative ? charaCfg.negative_prompt : charaCfg.positive_prompt
|
|
)
|
|
);
|
|
}
|
|
|
|
if (guidanceScale.type === cfgType.global || cfgPromptCombine.includes(cfgType.global)) {
|
|
splitCfgPrompt.unshift(
|
|
substituteParams(
|
|
isNegative ? extension_settings.cfg.global.negative_prompt : extension_settings.cfg.global.positive_prompt
|
|
)
|
|
);
|
|
}
|
|
|
|
const customSeparator = getCustomSeparator();
|
|
const combinedCfgPrompt = splitCfgPrompt.filter((e) => e.length > 0).join(customSeparator);
|
|
const insertionDepth = chat_metadata[metadataKeys.prompt_insertion_depth] ?? 1;
|
|
console.log(`Setting CFG with guidance scale: ${guidanceScale.value}, negatives: ${combinedCfgPrompt}`);
|
|
|
|
return {
|
|
value: combinedCfgPrompt,
|
|
depth: insertionDepth
|
|
};
|
|
}
|