From 03bb36e6dd5b2ebe8ef92ad1b3a2c81e861298b3 Mon Sep 17 00:00:00 2001
From: Alexis VIALARET <alexis.vialaret@artefact.com>
Date: Mon, 29 Jan 2024 14:02:54 +0100
Subject: [PATCH] fix: feedbacks

---
 frontend/lib/chat.py    | 11 +++++++++--
 frontend/lib/sidebar.py |  3 ++-
 2 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/frontend/lib/chat.py b/frontend/lib/chat.py
index cba9621..f4c6141 100644
--- a/frontend/lib/chat.py
+++ b/frontend/lib/chat.py
@@ -57,10 +57,13 @@ def chat():
             len(st.session_state.get("messages", [])) > 0
             and len(st.session_state.get("messages")) % 2 == 0
         ):
+            chat_id = st.session_state.get("chat_id")
+            get_chat(st.session_state.get("messages")[-1].chat_id)
+            messages = [Message(**message) for message in get_chat(chat_id)["messages"]]
             streamlit_feedback(
-                key=str(len(st.session_state.get("messages"))),
+                key=str(len(messages)),
                 feedback_type="thumbs",
-                on_submit=lambda feedback: send_feedback(st.session_state.get("messages")[-1].id, feedback),
+                on_submit=lambda feedback: send_feedback(messages[-1].id, feedback),
             )
 
 
@@ -80,3 +83,7 @@ def send_prompt(message: Message):
 def send_feedback(message_id: str, feedback: str):
     feedback = "thumbs_up" if feedback["score"] == "👍" else "thumbs_down"
     return query("post", f"/feedback/{message_id}/{feedback}").text
+
+def get_chat(chat_id: str):
+    chat = query("get", f"/chat/{chat_id}").json()
+    return chat
\ No newline at end of file
diff --git a/frontend/lib/sidebar.py b/frontend/lib/sidebar.py
index f5a1961..8490742 100644
--- a/frontend/lib/sidebar.py
+++ b/frontend/lib/sidebar.py
@@ -39,4 +39,5 @@ def list_chats():
     return query("get", "/chat/list").json()
 
 def get_chat(chat_id: str):
-    return query("get", f"/chat/{chat_id}").json()
+    chat = query("get", f"/chat/{chat_id}").json()
+    return chat