Update app.py

app.py

@@ -32,6 +32,10 @@ if 'chat_history_page2' not in st.session_state:
 if 'chat_history_page3' not in st.session_state:
     st.session_state['chat_history_page3'] = []
 
+# This session ID will be unique per user session and consistent across all pages.
+if 'session_id' not in st.session_state:
+    st.session_state['session_id'] = str(uuid.uuid4())
+
 
 
 # Step 1: Clone the Dataset Repository
@@ -189,18 +193,18 @@ def ask_bot(query):
     full_query = standard_prompt + query
     return 
 
-def save_conversation(chat_history, session_id):
-    # Define the base path for conversation logs
-    base_path = "Private_Book/conversation_logs"
-    
-    # Create the directory if it doesn't exist
+def save_conversation(chat_history, session_id, page_number):
+    base_path = "Chat_Store/conversation_logs"
     if not os.path.exists(base_path):
         os.makedirs(base_path)
 
-    # Create a unique filename using session_id
-    filename = f"{base_path}/{session_id}.json"
+    filename = f"{base_path}/{session_id}_page{page_number}.json"
+
+    if os.path.exists(filename):
+        with open(filename, 'r') as file:
+            existing_data = json.load(file)
+        chat_history = existing_data + chat_history
 
-    # Save the chat history as JSON
     with open(filename, 'w') as file:
         json.dump(chat_history, file, indent=4)
 
@@ -320,8 +324,7 @@ def page1():
                     new_messages_placeholder.markdown(f"<div style='background-color: {background_color}; padding: 10px; border-radius: 10px; margin: 10px;'>{chat[0]}: {chat[1]}</div>", unsafe_allow_html=True)
     
                 # Save conversation after chat interaction
-                session_id = str(uuid.uuid4())  # Generate a unique session ID
-                save_conversation(st.session_state['chat_history_page1'], session_id)
+                save_conversation(st.session_state['chat_history_page1'], st.session_state['session_id'], 1)
                                   
                 # Clear the input field after the query is made
                 query = ""