From 1eec1763137a5c262a0d73c1f0872ad596ba0f40 Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Thu, 18 Jan 2024 02:17:31 -0800
Subject: [PATCH] feat: save tags to chat data

---
 backend/apps/web/models/tags.py      | 2 --
 backend/apps/web/routers/chats.py    | 1 -
 src/routes/(app)/+page.svelte        | 9 +++++++++
 src/routes/(app)/c/[id]/+page.svelte | 8 ++++++++
 4 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/backend/apps/web/models/tags.py b/backend/apps/web/models/tags.py
index ef21ca08..5c6c094e 100644
--- a/backend/apps/web/models/tags.py
+++ b/backend/apps/web/models/tags.py
@@ -101,7 +101,6 @@ class TagTable:
         if tag == None:
             tag = self.insert_new_tag(form_data.tag_name, user_id)
 
-        print(tag)
         id = str(uuid.uuid4())
         chatIdTag = ChatIdTagModel(
             **{
@@ -131,7 +130,6 @@ class TagTable:
             .order_by(ChatIdTag.timestamp.desc())
         ]
 
-        print(tag_names)
         return [
             TagModel(**model_to_dict(tag))
             for tag in Tag.select().where(Tag.name.in_(tag_names))
diff --git a/backend/apps/web/routers/chats.py b/backend/apps/web/routers/chats.py
index 38685826..c4d9ebab 100644
--- a/backend/apps/web/routers/chats.py
+++ b/backend/apps/web/routers/chats.py
@@ -134,7 +134,6 @@ async def get_chat_tags_by_id(id: str, user=Depends(get_current_user)):
     tags = Tags.get_tags_by_chat_id_and_user_id(id, user.id)
 
     if tags != None:
-        print(tags)
         return tags
     else:
         raise HTTPException(
diff --git a/src/routes/(app)/+page.svelte b/src/routes/(app)/+page.svelte
index 66e8bc62..29e4f201 100644
--- a/src/routes/(app)/+page.svelte
+++ b/src/routes/(app)/+page.svelte
@@ -189,6 +189,7 @@
 						},
 						messages: messages,
 						history: history,
+						tags: [],
 						timestamp: Date.now()
 					});
 					await chats.set(await getChatList(localStorage.token));
@@ -690,11 +691,19 @@
 	const addTag = async (tagName) => {
 		const res = await addTagById(localStorage.token, $chatId, tagName);
 		tags = await getTags();
+
+		chat = await updateChatById(localStorage.token, $chatId, {
+			tags: tags
+		});
 	};
 
 	const deleteTag = async (tagName) => {
 		const res = await deleteTagById(localStorage.token, $chatId, tagName);
 		tags = await getTags();
+
+		chat = await updateChatById(localStorage.token, $chatId, {
+			tags: tags
+		});
 	};
 
 	const setChatTitle = async (_chatId, _title) => {
diff --git a/src/routes/(app)/c/[id]/+page.svelte b/src/routes/(app)/c/[id]/+page.svelte
index 481dc996..37f6f39c 100644
--- a/src/routes/(app)/c/[id]/+page.svelte
+++ b/src/routes/(app)/c/[id]/+page.svelte
@@ -707,11 +707,19 @@
 	const addTag = async (tagName) => {
 		const res = await addTagById(localStorage.token, $chatId, tagName);
 		tags = await getTags();
+
+		chat = await updateChatById(localStorage.token, $chatId, {
+			tags: tags.map((tag) => tag.name)
+		});
 	};
 
 	const deleteTag = async (tagName) => {
 		const res = await deleteTagById(localStorage.token, $chatId, tagName);
 		tags = await getTags();
+
+		chat = await updateChatById(localStorage.token, $chatId, {
+			tags: tags.map((tag) => tag.name)
+		});
 	};
 
 	onMount(async () => {