adding session cache

app.py

@@ -1,5 +1,6 @@
 import openai 
 import streamlit as st
+import extra_streamlit_components as stx
 import uuid
 import time
 
@@ -7,38 +8,47 @@ openai.api_key = st.secrets["OPENAI_API_KEY"]
 assistant_id = st.secrets["OPENAI_ASSISTANT_ID"]
 client = openai
 
+st.set_page_config(page_title="Assistant API Chat", page_icon=":speech_balloon:")
+
+@st.cache_resource(experimental_allow_widgets=True)
+def get_manager():
+    return stx.CookieManager()
+cookie_manager = get_manager()
+
+st.session_state.thread_id = cookie_manager.get('thread_id')
+
 if "start_chat" not in st.session_state:
     st.session_state.start_chat = False
 if "session_id" not in st.session_state:
     st.session_state.session_id = str(uuid.uuid4())
-if "thread_id" not in st.session_state:
-    st.session_state.thread_id = None
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
-st.set_page_config(page_title="Assistant API Chat", page_icon=":speech_balloon:")
 st.title(":speech_balloon: Assistant API Chat")
 
-st.sidebar.caption(f"**Session ID**:  \n  {st.session_state.session_id}")
-st.sidebar.header("Configuration")
-thread_id = st.sidebar.text_input("Enter your Thread ID:")
-    
-if st.sidebar.button("Start Chat"):
-
-    st.session_state.start_chat = True
-    if thread_id:
-        st.session_state.thread_id = thread_id
-    else:
-        thread = client.beta.threads.create(
-            metadata={
-                'session_id': st.session_state.session_id,
-            }
-        )
-        st.session_state.thread_id = thread.id
+with st.sidebar:
+    st.caption(f"**Session ID**:  \n  {st.session_state.session_id}")
+    st.header("Configuration")
+    thread_id = st.sidebar.text_input("Enter your Thread ID:", placeholder="Leave empty to start a new thread")
+    c1, c2 = st.columns(2)
+
+    if c1.button("Start Chat", use_container_width=True):
+        st.session_state.start_chat = True
+        if thread_id:
+            st.session_state.thread_id = thread_id
+        else:
+            thread = client.beta.threads.create(
+                metadata={
+                    'session_id': st.session_state.session_id,
+                }
+            )
+            st.session_state.thread_id = thread.id
+        cookie_manager.set('thread_id', st.session_state.thread_id)
 
-    st.session_state.messages = client.beta.threads.messages.list(
-        thread_id=st.session_state.thread_id
-    )
+    if c2.button("Clear Chat", use_container_width=True):
+        st.session_state.start_chat = False
+        st.session_state.thread_id = None
+        cookie_manager.delete('thread_id')
 
 #def process_message(message):
 #    message_content = message.content[0].text
@@ -47,8 +57,11 @@ if st.sidebar.button("Start Chat"):
 #    full_response = message_content.value + '\n\n' + '\n'.join(citations)
 #    return full_response
 
-if st.session_state.start_chat:
+if st.session_state.thread_id:
 
+    st.session_state.messages = client.beta.threads.messages.list(
+        thread_id=st.session_state.thread_id
+    )
     st.caption(f"**Thread ID**: {st.session_state.thread_id}")
     for message in reversed(st.session_state.messages.data):
         with st.chat_message(message.role):
@@ -86,4 +99,4 @@ if st.session_state.start_chat:
                 with st.chat_message("assistant"):
                     st.markdown(message.content[0].text.value)
 else:
-    st.write("Click on 'Start Chat' to start/continue a thread.")
+    st.write("Click on 'Start Chat' to start a new thread.")

requirements.txt

@@ -1,2 +1,3 @@
+extra_streamlit_components==0.1.60
 streamlit==1.28.2
 openai==1.2.4