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() {
+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);
+    await displayChats('', currentChat, displayName, avatarImg, selected_group, hightlightNames);
     const debouncedDisplay = debounce((searchQuery) => {
-        displayChats(searchQuery, currentChat, displayName, avatarImg, selected_group);
+        displayChats(searchQuery, currentChat, displayName, avatarImg, selected_group, []);
     });
     // Define the search input listener

     }, 200);
+}
-async function displayChats(searchQuery, currentChat, displayName, avatarImg, selected_group) {
+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	if (data.res && refresh) {	8075	if (data.res && refresh) {
8068	await displayPastChats();	8076	await displayPastChats();
8069	}	8077	}
		8078	return data?.fileNames \|\| [];
8070	}	8079	}
		8080
		8081	return [];
8071	}	8082	}
8072		8083
8073	function updateViewMessageIds(startFromZero = false) {	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;
-            await importFn(formData, { refresh: false });
+            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	export async function saveGroupBookmarkChat(groupId, name, metadata, mesId) {	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	handleChat(chat);	675	handleChat(chat);
674	}	676	}
675		677
676	return response.send({ res: true });	678	return response.send({ res: true, fileNames });
677	}	679	}
678		680
679	if (format === 'jsonl') {	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 });
+            response.send({ res: true, fileNames });
+        }
     } catch (error) {
         console.error(error);