Indicate welcome page assistant in the list

d5c56fa40531e61cca7baaa87e2ccaa4a2890604

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

Parents: 7b9f5b3fb85b1a4a95e37bf6e1667b050ddf14bb

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+36 -21 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);
+            const chatTimestamp = timestampToMoment(chat.last_mes);
-            return;
+            chat.char_name = character.name;
-        }
+            chat.date_short = chatTimestamp.format('l');
+            chat.date_long = chatTimestamp.format('LL LT');
-        const chatTimestamp = timestampToMoment(chat.last_mes);
+            chat.chat_name = chat.file_name.replace('.jsonl', '');
-        chat.char_name = character.name;
+            chat.char_thumbnail = getThumbnailUrl('avatar', character.avatar);
-        chat.date_short = chatTimestamp.format('l');
+        });
-        chat.date_long = chatTimestamp.format('LL LT');
-        chat.chat_name = chat.file_name.replace('.jsonl', '');
-        chat.char_thumbnail = getThumbnailUrl('avatar', character.avatar);
-    });
     return data;
+}
 export async function openPermanentAssistantChat({ tryCreate = true } = {}) {
-    const characterId = characters.findIndex(x => x.avatar === permanentAssistantAvatar);
+    const avatar = getPermanentAssistantAvatar();
+    const characterId = characters.findIndex(x => x.avatar === avatar);
     if (characterId === -1) {
         if (!tryCreate) {
-            console.error(`Character not found for avatar ID: ${permanentAssistantAvatar}. Cannot create.`);
+            console.error(`Character not found for avatar ID: ${avatar}. Cannot create.`);
             return;
+        }
         try {
-            console.log(`Character not found for avatar ID: ${permanentAssistantAvatar}. Creating new assistant.`);
+            console.log(`Character not found for avatar ID: ${avatar}. Creating new assistant.`);
             await createPermanentAssistant();
             return openPermanentAssistantChat({ tryCreate: false });
+        }

     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;