Highlight just imported chats in the list

87d33b4d0befbb4fab22316846956adbb4348679

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

public/script.js+24 -6 Viewed

  * The function first fetches the chats, processes them, and then displays them in
  * the HTML. It also has a built-in search functionality that allows filtering the
  * displayed chats based on a search query.
 + * @param {string[]} hightlightNames - An array of chat names to highlight
  */
 export async function displayPastChats(hightlightNames = []) {
     $('#select_chat_div').empty();
     $('#select_chat_search').val('').off('input');

     const displayName = chatDetails.characterName;
     const avatarImg = chatDetails.avatarImgURL;
     await displayChats('', currentChat, displayName, avatarImg, selected_group, hightlightNames);
     const debouncedDisplay = debounce((searchQuery) => {
         displayChats(searchQuery, currentChat, displayName, avatarImg, selected_group, []);
     });
     // Define the search input listener

     }, 200);
+}
 async function displayChats(searchQuery, currentChat, displayName, avatarImg, selected_group, highlightNames) {
     try {
         const trimExtension = (fileName) => String(fileName).replace('.jsonl', '');

+            }
             $('#select_chat_div').append(template);
++
 +            if (Array.isArray(highlightNames) && highlightNames.includes(chat.file_name)) {
 +                const templateOffset = template.offset().top - template.parent().offset().top;
 +                $('#select_chat_div').scrollTop(templateOffset);
 +                flashHighlight(template, debounce_timeout.extended);
 +            }
+        }
     } catch (error) {
         console.error('Error loading chats:', error);

  * @param {FormData} formData Form data to send to the server.
  * @param {object} [options={}] Options for the import
  * @param {boolean} [options.refresh] Whether to refresh the group chat list after import
 + * @returns {Promise<string[]>} List of imported file names.
  */
 async function importCharacterChat(formData, { refresh = true } = {}) {
     const fetchResult = await fetch('/api/chats/import', {

@@ -8067,7 +8075,10 @@ async function importCharacterChat(formData, { refresh = true } = {}) {
8067	8075	if (data.res && refresh) {
8068	8076	await displayPastChats();
8069	8077	}
	8078	+ return data?.fileNames \|\| [];
8070	8079	}
	8080	+
	8081	+ return [];
8071	8082	}
8072	8083
8073	8084	function updateViewMessageIds(startFromZero = false) {

             return;
+        }
 +        const importedFileNames = [];
++
         for (const file of targetElement.files) {
             const ext = file.name.match(/\.(\w+)$/);
             const format = ext?.[1]?.toLowerCase();

             formData.set('user_name', name1);
             const importFn = selected_group ? importGroupChat : importCharacterChat;
             const result = await importFn(formData, { refresh: false });
 +            importedFileNames.push(...result);
+        }
 -        await displayPastChats();
 +        if (importedFileNames.length > 0) {
 +            toastr.success(t`Successfully imported ${importedFileNames.length} chat(s).`);
 +        }
++
 +        await displayPastChats(importedFileNames);
         targetElement.value = '';
     });

public/scripts/group-chats.js+5 -0 Viewed

  * @param {FormData} formData Form data to send to the server
  * @param {object} [options={}] Options for the import
  * @param {boolean} [options.refresh] Whether to refresh the group chat list after import
 + * @returns {Promise<string[]>} List of imported file names
  */
 export async function importGroupChat(formData, { refresh = true } = {}) {
     const fetchResult = await fetch('/api/chats/group/import', {

@@ -2089,7 +2090,11 @@ export async function importGroupChat(formData, { refresh = true } = {}) {
2089	2090	}
2090	2091	}
2091	2092	}
	2093	+
	2094	+ return data?.fileNames \|\| [];
2092	2095	}
	2096	+
	2097	+ return [];
2093	2098	}
2094	2099
2095	2100	export async function saveGroupBookmarkChat(groupId, name, metadata, mesId) {

src/endpoints/chats.js+5 -2 Viewed

     const avatarUrl = (request.body.avatar_url).replace('.png', '');
     const characterName = request.body.character_name;
     const userName = request.body.user_name || 'User';
 +    const fileNames = [];
     if (!request.file) {
         return response.sendStatus(400);

             const handleChat = (chat) => {
                 const fileName = `${characterName} - ${humanizedISO8601DateTime()} imported.jsonl`;
                 const filePath = path.join(request.user.directories.chats, avatarUrl, fileName);
 +                fileNames.push(fileName);
                 writeFileAtomicSync(filePath, chat, 'utf8');
             };

@@ -673,7 +675,7 @@ router.post('/import', validateAvatarUrlMiddleware, function (request, response)
673	675	handleChat(chat);
674	676	}
675	677
676	678	return response.send({ res: true, fileNames });
677	679	}
678	680
679	681	if (format === 'jsonl') {

             const fileName = `${characterName} - ${humanizedISO8601DateTime()} imported.jsonl`;
             const filePath = path.join(request.user.directories.chats, avatarUrl, fileName);
 +            fileNames.push(fileName);
             if (flattenedChat !== data) {
                 writeFileAtomicSync(filePath, flattenedChat, 'utf8');
             } else {
                 fs.copyFileSync(pathToUpload, filePath);
+            }
             fs.unlinkSync(pathToUpload);
             response.send({ res: true, fileNames });
+        }
     } catch (error) {
         console.error(error);