Update app.py

app.py

@@ -191,24 +191,25 @@ def ask_bot(query):
     full_query = standard_prompt + query
     return full_query
 
-def save_conversation(chat_history, session_id, page_number):
+def save_conversation(chat_histories, session_id):
     base_path = "Chat_Store/conversation_logs"
     if not os.path.exists(base_path):
         os.makedirs(base_path)
 
-    filename = f"{base_path}/{session_id}_page{page_number}.json"
+    filename = f"{base_path}/{session_id}.json"
 
     # Check if the log file already exists
-    existing_data = []
+    existing_data = {"page1": [], "page2": [], "page3": []}
     if os.path.exists(filename):
         with open(filename, 'r', encoding='utf-8') as file:
             existing_data = json.load(file)
 
-    # Append the new chat history to the existing data
-    full_chat_history = existing_data + chat_history
+    # Append the new chat history to the existing data for each page
+    for page_number, chat_history in enumerate(chat_histories, start=1):
+        existing_data[f"page{page_number}"] += chat_history
 
     with open(filename, 'w', encoding='utf-8') as file:
-        json.dump(full_chat_history, file, indent=4, ensure_ascii=False)
+        json.dump(existing_data, file, indent=4, ensure_ascii=False)
 
     # Git operations
     try:
@@ -217,7 +218,7 @@ def save_conversation(chat_history, session_id, page_number):
         os.chdir('Chat_Store')
     
         # Correct file path relative to the Git repository's root
-        git_file_path = f"conversation_logs/{session_id}_page{page_number}.json"
+        git_file_path = f"conversation_logs/{session_id}.json"
     
         repo2.git_add(git_file_path)
         repo2.git_commit(f"Add/update conversation log for session {session_id}")
@@ -326,8 +327,17 @@ def page1():
                 background_color = "#ffeecf" if chat[2] == "new" else "#ffeecf" if chat[0] == "User" else "#ffeecf"
                 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
-            save_conversation(st.session_state['chat_history_page1'], st.session_state['session_id'], 1)
+
+             # Combine chat histories from all pages
+            all_chat_histories = [
+                st.session_state['chat_history_page1'],
+                st.session_state['chat_history_page2'],
+                st.session_state['chat_history_page3']
+            ]
+        
+            # Save the combined chat histories
+            save_conversation(all_chat_histories, st.session_state['session_id'])
+
 
             
             # Display the current working directory after save_conversation
@@ -435,7 +445,14 @@ def page2():
 
             st.session_state['chat_history_page2'].append(("Bot", response, "new"))
 
-            #save_conversation(st.session_state['chat_history_page2'], st.session_state['session_id'], 2)
+            # Combine chat histories from all pages
+            all_chat_histories = [
+                st.session_state['chat_history_page1'],
+                st.session_state['chat_history_page2'],
+                st.session_state['chat_history_page3']
+            ]
+            
+            save_conversation(st.session_state['chat_history_page2'], st.session_state['session_id'], 2)
 
             # Display new messages at the bottom
             new_messages = st.session_state['chat_history_page2'][-2:]
@@ -551,8 +568,15 @@ def page3():
             for chat in new_messages:
                 background_color = "#ffeecf" if chat[2] == "new" else "#ffeecf" if chat[0] == "User" else "#ffeecf"
                 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(st.session_state['chat_history_page3'], st.session_state['session_id'], 3)
+
+            # Combine chat histories from all pages
+            all_chat_histories = [
+                st.session_state['chat_history_page1'],
+                st.session_state['chat_history_page2'],
+                st.session_state['chat_history_page3']
+            ]
+            
+            save_conversation(st.session_state['chat_history_page3'], st.session_state['session_id'], 3)
 
             # Clear the input field after the query is made
             query = ""