Merge pull request #2560 from SillyTavern/dupe-persona Dupe persona

a2661b2c486f2b509f096a3ce8e59fedaddbbb92

Cohee <18619528+Cohee1207@users.noreply.github.com>

Signed

Parents: 6f58f9c599fe3cdc2436589bd081f335a17cf83b 9cf53c6a557962d1b352e45220c9c508566401c9

public/index.html+3 -0 Viewed

                         <button class="menu_button set_default_persona" title="Select this as default persona for the new chats." data-i18n="[title]Select this as default persona for the new chats.">
                             <i class="fa-fw fa-solid fa-crown"></i>
                         </button>
+                        <button class="menu_button duplicate_persona" title="Duplicate persona" data-i18n="[title]Duplicate persona">
+                            <i class="fa-fw fa-solid fa-clone"></i>
+                        </button>
                         <button class="menu_button delete_avatar" title="Delete persona" data-i18n="[title]Delete persona">
                             <i class="fa-fw fa-solid fa-trash-alt"></i>
                         </button>

public/scripts/personas.js+70 -15 Viewed

 import {
-    callPopup,
     characters,
     chat,
     chat_metadata,

 import { debounce_timeout } from './constants.js';
 import { FILTER_TYPES, FilterHelper } from './filters.js';
 import { selected_group } from './group-chats.js';
-import { POPUP_TYPE, Popup } from './popup.js';
+import { POPUP_RESULT, POPUP_TYPE, Popup } from './popup.js';
 let savePersonasPage = 0;
 const GRID_STORAGE_KEY = 'Personas_GridView';

  * @returns {Promise} Promise that resolves when the persona is set
  */
 export async function createPersona(avatarId) {
-    const personaName = await callPopup('<h3>Enter a name for this persona:</h3>Cancel if you\'re just uploading an avatar.', 'input', '');
+    const personaName = await Popup.show.input('Enter a name for this persona:', 'Cancel if you\'re just uploading an avatar.', '');
     if (!personaName) {
         console.debug('User cancelled creating a persona');
         return;
+    }
-    await delay(500);
+    const personaDescription = await Popup.show.input('Enter a description for this persona:', 'You can always add or change it later.', '', { rows: 4 });
-    const personaDescription = await callPopup('<h3>Enter a description for this persona:</h3>You can always add or change it later.', 'input', '', { rows: 4 });
     initPersona(avatarId, personaName, personaDescription);
     if (power_user.persona_show_notifications) {

+}
 async function createDummyPersona() {
-    const personaName = await callPopup('<h3>Enter a name for this persona:</h3>', 'input', '');
+    const personaName = await Popup.show.input('Enter a name for this persona:', null);
     if (!personaName) {
         console.debug('User cancelled creating dummy persona');

         return;
+    }
+    let personaUnbind = false;
     const existingPersona = power_user.personas[avatarId];
-    const personaName = await callPopup('<h3>Enter a name for this persona:</h3>(If empty name is provided, this will unbind the name from this avatar)', 'input', existingPersona || '');
+    const personaName = await Popup.show.input(
+        'Enter a name for this persona:',
+        '(If empty name is provided, this will unbind the name from this avatar)',
+        existingPersona || '',
+        { onClose: (p) => { personaUnbind = p.value === '' && p.result === POPUP_RESULT.AFFIRMATIVE; } });
     // If the user clicked cancel, don't do anything
-    if (personaName === false) {
+    if (personaName === null && !personaUnbind) {
         return;
+    }
-    if (personaName.length > 0) {
+    if (personaName && personaName.length > 0) {
         // If the user clicked ok and entered a name, bind the name to the persona
         console.log(`Binding persona ${avatarId} to name ${personaName}`);
         power_user.personas[avatarId] = personaName;

             );
+        }
         power_user.personas[user_avatar] = name1;
-        power_user.persona_descriptions[user_avatar] = { description: '', position: persona_description_positions.IN_PROMPT };
+        power_user.persona_descriptions[user_avatar] = {
+            description: '',
+            position: persona_description_positions.IN_PROMPT,
+            depth: DEFAULT_DEPTH,
+            role: DEFAULT_ROLE,
+        };
+    }
     chat_metadata['persona'] = user_avatar;

         return;
+    }
-    const confirm = await callPopup('<h3>Are you sure you want to delete this avatar?</h3>All information associated with its linked persona will be lost.', 'confirm');
+    const confirm = await Popup.show.confirm('Are you sure you want to delete this avatar?', 'All information associated with its linked persona will be lost.');
     if (!confirm) {
         console.debug('User cancelled deleting avatar');

     const personaName = power_user.personas[avatarId];
     if (avatarId === currentDefault) {
-        const confirm = await callPopup('Are you sure you want to remove the default persona?', 'confirm');
+        const confirm = await Popup.show.confirm('Are you sure you want to remove the default persona?', personaName);
         if (!confirm) {
             console.debug('User cancelled removing default persona');

+        }
         delete power_user.default_persona;
     } else {
-        const confirm = await callPopup(`<h3>Are you sure you want to set "${personaName}" as the default persona?</h3>
+        const confirm = await Popup.show.confirm(`Are you sure you want to set "${personaName}" as the default persona?`, 'This name and avatar will be used for all new chats, as well as existing chats where the user persona is not locked.');
-        This name and avatar will be used for all new chats, as well as existing chats where the user persona is not locked.`, 'confirm');
         if (!confirm) {
             console.debug('User cancelled setting default persona');

+}
 async function syncUserNameToPersona() {
-    const confirmation = await callPopup(`<h3>Are you sure?</h3>All user-sent messages in this chat will be attributed to ${name1}.`, 'confirm');
+    const confirmation = await Popup.show.confirm('Are you sure?', `All user-sent messages in this chat will be attributed to ${name1}.`);
     if (!confirmation) {
         return;

+    }
+}
+/**
+ * Duplicates a persona.
+ * @param {string} avatarId
+ * @returns {Promise<void>}
+ */
+async function duplicatePersona(avatarId) {
+    const personaName = power_user.personas[avatarId];
+    if (!personaName) {
+        toastr.warning('Chosen avatar is not a persona');
+        return;
+    }
+    const confirm = await Popup.show.confirm('Are you sure you want to duplicate this persona?', personaName);
+    if (!confirm) {
+        console.debug('User cancelled duplicating persona');
+        return;
+    }
+    const newAvatarId = `${Date.now()}-${personaName.replace(/[^a-zA-Z0-9]/g, '')}.png`;
+    const descriptor = power_user.persona_descriptions[avatarId];
+    power_user.personas[newAvatarId] = personaName;
+    power_user.persona_descriptions[newAvatarId] = {
+        description: descriptor?.description ?? '',
+        position: descriptor?.position ?? persona_description_positions.IN_PROMPT,
+        depth: descriptor?.depth ?? DEFAULT_DEPTH,
+        role: descriptor?.role ?? DEFAULT_ROLE,
+    };
+    await uploadUserAvatar(getUserAvatar(avatarId), newAvatarId);
+    await getUserAvatars(true, newAvatarId);
+    saveSettingsDebounced();
+}
 export function initPersonas() {
     $(document).on('click', '.bind_user_name', bindUserNameToPersona);
     $(document).on('click', '.set_default_persona', setDefaultPersona);

         $('#avatar_upload_file').trigger('click');
     });
+    $(document).on('click', '#user_avatar_block .duplicate_persona', function (e) {
+        e.stopPropagation();
+        const avatarId = $(this).closest('.avatar-container').find('.avatar').attr('imgfile');
+        if (!avatarId) {
+            console.log('no imgfile');
+            return;
+        }
+        duplicatePersona(avatarId);
+    });
     $(document).on('click', '#user_avatar_block .set_persona_image', function (e) {
         e.stopPropagation();
         const avatarId = $(this).closest('.avatar-container').find('.avatar').attr('imgfile');

public/scripts/popup.js+5 -5 Viewed

     /**
      * Asynchronously displays an input popup with the given header and text, and returns the user's input.
+     *
-     * @param {string} header - The header text for the popup.
+     * @param {string?} header - The header text for the popup.
-     * @param {string} text - The main text for the popup.
+     * @param {string?} text - The main text for the popup.
      * @param {string} [defaultValue=''] - The default value for the input field.
      * @param {PopupOptions} [popupOptions={}] - Options for the popup.
      * @return {Promise<string?>} A Promise that resolves with the user's input.

@@ -591,15 +591,15 @@ class PopupUtils {
591	/**	591	/**
592	* Builds popup content with header and text below	592	* Builds popup content with header and text below
593	*	593	*
594	* @param {string} header - The header to be added to the text	594	* @param {string?} header - The header to be added to the text
595	* @param {string} text - The main text content	595	* @param {string?} text - The main text content
596	*/	596	*/
597	static BuildTextWithHeader(header, text) {	597	static BuildTextWithHeader(header, text) {
598	if (!header) {	598	if (!header) {
599	return text;	599	return text;
600	}	600	}
601	return `<h3>${header}</h3>	601	return `<h3>${header}</h3>
602	${text}`;	602	${text ?? ''}`; // Convert no text to empty string
603	}	603	}
604	}	604	}
605		605

public/style.css+4 -0 Viewed

     opacity: 1;
+}
+.avatar-container .avatar-buttons .menu_button {
+    padding: 3px;
+}
 /* Ross should be able to handle this later */
 /*.big-avatars .avatar-buttons{
     justify-content: center;