Indicate welcome page assistant in the list

d5c56fa40531e61cca7baaa87e2ccaa4a2890604

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

public/index.html+3 -0 Viewed

                     <div class="wide100p character_name_block">
                         <span class="ch_name"></span>
                         <small class="ch_additional_info ch_add_placeholder">+++</small>
+                        <small class="ch_assistant" title="This character will be used as a welcome page assistant." data-i18n="[title]This character will be used as a welcome page assistant.">
+                            <i class="fa-solid fa-sm fa-user-graduate"></i>
+                        </small>
                         <small class="ch_additional_info character_version"></small>
                         <small class="ch_additional_info ch_avatar_url"></small>
                     </div>

public/script.js+6 -1 Viewed

 import { getContext } from './scripts/st-context.js';
 import { extractReasoningFromData, initReasoning, parseReasoningInSwipes, PromptReasoning, ReasoningHandler, removeReasoningFromString, updateReasoningUI } from './scripts/reasoning.js';
 import { accountStorage } from './scripts/util/AccountStorage.js';
-import { initWelcomeScreen, openPermanentAssistantChat, openPermanentAssistantCard } from './scripts/welcome-screen.js';
+import { initWelcomeScreen, openPermanentAssistantChat, openPermanentAssistantCard, getPermanentAssistantAvatar } from './scripts/welcome-screen.js';
 // API OBJECT FOR EXTERNAL WIRING
 globalThis.SillyTavern = {

     template.toggleClass('is_fav', item.fav || item.fav == 'true');
     template.find('.ch_fav').val(item.fav);
+    const isAssistant = item.avatar === getPermanentAssistantAvatar();
+    if (!isAssistant) {
+        template.find('.ch_assistant').remove();
+    }
     const description = item.data?.creator_notes || '';
     if (description) {
         template.find('.ch_description').text(description);

public/scripts/welcome-screen.js+29 -14 Viewed

 import { is_group_generating } from './group-chats.js';
 import { t } from './i18n.js';
 import { renderTemplateAsync } from './templates.js';
+import { accountStorage } from './util/AccountStorage.js';
 import { timestampToMoment } from './utils.js';
-const permanentAssistantAvatar = 'default_Assistant.png';
+const assistantAvatarKey = 'assistant';
+const defaultAssistantAvatar = 'default_Assistant.png';
+export function getPermanentAssistantAvatar() {
+    const assistantAvatar = accountStorage.getItem(assistantAvatarKey);
+    if (assistantAvatar === null) {
+        return defaultAssistantAvatar;
+    }
+    const character = characters.find(x => x.avatar === assistantAvatar);
+    if (character === undefined) {
+        accountStorage.removeItem(assistantAvatarKey);
+        return defaultAssistantAvatar;
+    }
+    return assistantAvatar;
+}
 export async function openWelcomeScreen() {
     const currentChatId = getCurrentChatId();

     /** @type {RecentChat[]} */
     const data = await response.json();
-    data.sort((a, b) => b.last_mes - a.last_mes).forEach((chat, index) => {
+    data.sort((a, b) => b.last_mes - a.last_mes)
-        const character = characters.find(x => x.avatar === chat.avatar);
+        .map(chat => ({ chat, character: characters.find(x => x.avatar === chat.avatar) }))
-        if (!character) {
+        .filter(t => t.character)
-            console.warn(`Character not found for chat: ${chat.file_name}`);
+        .forEach(({ chat, character }) => {
-            data.splice(index, 1);
-            return;
-        }
             const chatTimestamp = timestampToMoment(chat.last_mes);
             chat.char_name = character.name;
             chat.date_short = chatTimestamp.format('l');

@@ -141,15 +154,16 @@ async function getRecentChats() {
141	}	154	}
142		155
143	export async function openPermanentAssistantChat({ tryCreate = true } = {}) {	156	export async function openPermanentAssistantChat({ tryCreate = true } = {}) {
144	const characterId = characters.findIndex(x => x.avatar === permanentAssistantAvatar);	157	const avatar = getPermanentAssistantAvatar();
		158	const characterId = characters.findIndex(x => x.avatar === avatar);
145	if (characterId === -1) {	159	if (characterId === -1) {
146	if (!tryCreate) {	160	if (!tryCreate) {
147	console.error(`Character not found for avatar ID: ${permanentAssistantAvatar}. Cannot create.`);	161	console.error(`Character not found for avatar ID: ${avatar}. Cannot create.`);
148	return;	162	return;
149	}	163	}
150		164
151	try {	165	try {
152	console.log(`Character not found for avatar ID: ${permanentAssistantAvatar}. Creating new assistant.`);	166	console.log(`Character not found for avatar ID: ${avatar}. Creating new assistant.`);
153	await createPermanentAssistant();	167	await createPermanentAssistant();
154	return openPermanentAssistantChat({ tryCreate: false });	168	return openPermanentAssistantChat({ tryCreate: false });
155	}	169	}

     const formData = new FormData();
     formData.append('ch_name', neutralCharacterName);
-    formData.append('file_name', permanentAssistantAvatar.replace('.png', ''));
+    formData.append('file_name', defaultAssistantAvatar.replace('.png', ''));
     const headers = getRequestHeaders();
     delete headers['Content-Type'];

+}
 export async function openPermanentAssistantCard() {
-    const characterId = characters.findIndex(x => x.avatar === permanentAssistantAvatar);
+    const avatar = getPermanentAssistantAvatar();
+    const characterId = characters.findIndex(x => x.avatar === avatar);
     if (characterId === -1) {
         toastr.info(t`Assistant not found. Try sending a chat message.`);
         return;