diff --git a/public/script.js b/public/script.js index b410625b5..e9bb2add6 100644 --- a/public/script.js +++ b/public/script.js @@ -2035,18 +2035,19 @@ function addOneMessage(mes, { type = 'normal', insertAfter = null, scroll = true const messageId = forceId ?? chat.length - 1; $('#chat').find(`[mesid="${messageId}"]`).find('.mes_text').append(messageText); appendMediaToMessage(mes, newMessage); - hideSwipeButtons(newMessageId); + hideSwipeButtons(); } addCopyToCodeBlocks(newMessage); + $('#chat .mes').last().addClass('last_mes'); + $('#chat .mes').eq(-2).removeClass('last_mes'); + + hideSwipeButtons(); + showSwipeButtons(); + // Don't scroll if not inserting last if (!insertAfter && !insertBefore && scroll) { - $('#chat .mes').last().addClass('last_mes'); - $('#chat .mes').eq(-2).removeClass('last_mes'); - - hideSwipeButtons(newMessageId); - showSwipeButtons(newMessageId); scrollChatToBottom(); } } @@ -2139,10 +2140,42 @@ function scrollChatToBottom() { * @param {*} _name2 - The name of the character. Uses global name2 if not provided. * @param {*} _original - The original message for {{original}} substitution. * @param {*} _group - The group members list for {{group}} substitution. + * @param {boolean} _replaceCharacterCard - Whether to replace character card macros. * @returns {string} The string with substituted parameters. */ function substituteParams(content, _name1, _name2, _original, _group, _replaceCharacterCard = true) { - return evaluateMacros(content, _name1 ?? name1, _name2 ?? name2, _original, _group ?? name2, _replaceCharacterCard); + const environment = {}; + + if (typeof _original === 'string') { + let originalSubstituted = false; + environment.original = () => { + if (originalSubstituted) { + return ''; + } + + originalSubstituted = true; + return _original; + }; + } + + if (_replaceCharacterCard) { + const fields = getCharacterCardFields(); + environment.charPrompt = fields.system || ''; + environment.charJailbreak = fields.jailbreak || ''; + environment.description = fields.description || ''; + environment.personality = fields.personality || ''; + environment.scenario = fields.scenario || ''; + environment.persona = fields.persona || ''; + environment.mesExamples = fields.mesExamples || ''; + } + + // Must be substituted last so that they're replaced inside {{description}} + // TODO: evaluate macros recursively so we don't need to rely on substitution order + environment.user = _name1 ?? name1; + environment.char = _name2 ?? name2; + environment.group = environment.charIfNotGroup = _group ?? name2; + + return evaluateMacros(content, environment); } @@ -2442,7 +2475,7 @@ function showStopButton() { function hideStopButton() { // prevent NOOP, because hideStopButton() gets called multiple times - if($('#mes_stop').css('display') !== 'none') { + if ($('#mes_stop').css('display') !== 'none') { $('#mes_stop').css({ 'display': 'none' }); eventSource.emit(event_types.GENERATION_ENDED, chat.length); } @@ -6475,7 +6508,7 @@ function callPopup(text, type, inputValue = '', { okButton, rows, wide, large } }); } -function showSwipeButtons(id = chat.length - 1) { +function showSwipeButtons() { if (chat.length === 0) { return; } @@ -6483,10 +6516,9 @@ function showSwipeButtons(id = chat.length - 1) { if ( chat[chat.length - 1].is_system || !swipes || - $('.mes:last').attr('mesid') < 0 || + Number($('.mes:last').attr('mesid')) < 0 || chat[chat.length - 1].is_user || chat[chat.length - 1].extra?.image || - id < 0 || (selected_group && is_group_generating) ) { return; } @@ -6505,7 +6537,7 @@ function showSwipeButtons(id = chat.length - 1) { chat[chat.length - 1]['swipes'][0] = chat[chat.length - 1]['mes']; //assign swipe array with last message from chat } - const currentMessage = $('#chat').children().filter(`[mesid="${id}"]`); + const currentMessage = $('#chat').children().filter(`[mesid="${chat.length - 1}"]`); const swipeId = chat[chat.length - 1].swipe_id; var swipesCounterHTML = (`${(swipeId + 1)}/${(chat[chat.length - 1].swipes.length)}`); @@ -6530,10 +6562,10 @@ function showSwipeButtons(id = chat.length - 1) { //console.log(chat[chat.length - 1].swipes.length); } -function hideSwipeButtons(id = chat.length - 1) { +function hideSwipeButtons() { //console.log('hideswipebuttons entered'); - $('#chat').children().filter(`[mesid="${id}"]`).children('.swipe_right').css('display', 'none'); - $('#chat').children().filter(`[mesid="${id}"]`).children('.swipe_left').css('display', 'none'); + $('#chat').find('.swipe_right').css('display', 'none'); + $('#chat').find('.swipe_left').css('display', 'none'); } export async function saveMetadata() { diff --git a/public/scripts/macros.js b/public/scripts/macros.js index 2734a3a04..e88971de1 100644 --- a/public/scripts/macros.js +++ b/public/scripts/macros.js @@ -1,4 +1,4 @@ -import { chat, main_api, getMaxContextSize, getCharacterCardFields } from '../script.js'; +import { chat, main_api, getMaxContextSize } from '../script.js'; import { timestampToMoment, isDigitsOnly } from './utils.js'; import { textgenerationwebui_banned_in_macros } from './textgen-settings.js'; import { replaceInstructMacros } from './instruct-mode.js'; @@ -201,56 +201,41 @@ function diceRollReplace(input, invalidRollPlaceholder = '') { /** * Substitutes {{macro}} parameters in a string. * @param {string} content - The string to substitute parameters in. - * @param {*} _name1 - The name of the user. - * @param {*} _name2 - The name of the character. - * @param {*} _original - The original message for {{original}} substitution. - * @param {*} _group - The group members list for {{group}} substitution. - * @param {boolean} _replaceCharacterCard - Whether to replace character card macros. + * @param {Object} env - Map of macro names to the values they'll be substituted with. If the param + * values are functions, those functions will be called and their return values are used. * @returns {string} The string with substituted parameters. */ -export function evaluateMacros(content, _name1, _name2, _original, _group, _replaceCharacterCard = true) { +export function evaluateMacros(content, env) { if (!content) { return ''; } - // Replace {{original}} with the original message - // Note: only replace the first instance of {{original}} - // This will hopefully prevent the abuse - if (typeof _original === 'string') { - content = content.replace(/{{original}}/i, _original); - } content = diceRollReplace(content); content = replaceInstructMacros(content); content = replaceVariableMacros(content); content = content.replace(/{{newline}}/gi, '\n'); content = content.replace(/{{input}}/gi, String($('#send_textarea').val())); - if (_replaceCharacterCard) { - const fields = getCharacterCardFields(); - content = content.replace(/{{charPrompt}}/gi, fields.system || ''); - content = content.replace(/{{charJailbreak}}/gi, fields.jailbreak || ''); - content = content.replace(/{{description}}/gi, fields.description || ''); - content = content.replace(/{{personality}}/gi, fields.personality || ''); - content = content.replace(/{{scenario}}/gi, fields.scenario || ''); - content = content.replace(/{{persona}}/gi, fields.persona || ''); - content = content.replace(/{{mesExamples}}/gi, fields.mesExamples || ''); + // Substitute passed-in variables + for (const varName in env) { + if (!Object.hasOwn(env, varName)) continue; + + const param = env[varName]; + content = content.replace(new RegExp(`{{${varName}}}`, 'gi'), param); } content = content.replace(/{{maxPrompt}}/gi, () => String(getMaxContextSize())); - content = content.replace(/{{user}}/gi, _name1); - content = content.replace(/{{char}}/gi, _name2); - content = content.replace(/{{charIfNotGroup}}/gi, _group); - content = content.replace(/{{group}}/gi, _group); content = content.replace(/{{lastMessage}}/gi, getLastMessage()); content = content.replace(/{{lastMessageId}}/gi, getLastMessageId()); content = content.replace(/{{firstIncludedMessageId}}/gi, getFirstIncludedMessageId()); content = content.replace(/{{lastSwipeId}}/gi, getLastSwipeId()); content = content.replace(/{{currentSwipeId}}/gi, getCurrentSwipeId()); - content = content.replace(//gi, _name1); - content = content.replace(//gi, _name2); - content = content.replace(//gi, _group); - content = content.replace(//gi, _group); + // Legacy non-macro substitutions + content = content.replace(//gi, typeof env.user === 'function' ? env.user() : env.user); + content = content.replace(//gi, typeof env.char === 'function' ? env.char() : env.char); + content = content.replace(//gi, typeof env.group === 'function' ? env.group() : env.group); + content = content.replace(//gi, typeof env.group === 'function' ? env.group() : env.group); content = content.replace(/\{\{\/\/([\s\S]*?)\}\}/gm, '');