Update app.py

app.py

@@ -7,12 +7,17 @@ messages_dir = "messages"
 if not os.path.exists(messages_dir):
     os.makedirs(messages_dir)
 
+# Initialize session state to store the list of threads
+if 'thread_list' not in st.session_state:
+    st.session_state.thread_list = []
+
 # Create a dictionary to store messages and threads
 messages = {}
 
 def create_thread(thread_id, user_input):
     thread = {'messages': [user_input], 'created_at': datetime.now()}
     messages[thread_id] = thread
+    st.session_state.thread_list.append(thread_id)
 
 def add_message(thread_id, user_input):
     if thread_id in messages:
@@ -38,11 +43,11 @@ def main():
 
     # Display existing threads
     st.subheader("Existing Threads")
-    for thread_id in messages.keys():
+    for thread_id in st.session_state.thread_list:
         st.write(f"- {thread_id}")
 
     # Allow user to select a thread and add messages
-    thread_to_post = st.selectbox("Select a thread to post:", list(messages.keys()), key='thread_to_post')
+    thread_to_post = st.selectbox("Select a thread to post:", st.session_state.thread_list, key='thread_to_post')
     if thread_to_post:
         user_input = st.text_area("Write your message:", key='user_input')
         if user_input: