Update app.py

app.py

@@ -20,8 +20,14 @@ import pydeck as pdk
 from urllib.error import URLError
 
 # Initialize session state variables
-if 'chat_history' not in st.session_state:
-    st.session_state['chat_history'] = []
+if 'chat_history_page1' not in st.session_state:
+    st.session_state['chat_history_page1'] = []
+
+if 'chat_history_page2' not in st.session_state:
+    st.session_state['chat_history_page2'] = []
+
+# Then use st.session_state['chat_history_page1'] within page1()
+
 
 st.set_page_config(layout="wide")
 
@@ -49,6 +55,7 @@ api_key = os.getenv("OPENAI_API_KEY")
 # Updated caching mechanism using st.cache_data
 @st.cache_data(persist="disk")  # Using persist="disk" to save cache across sessions
 def load_vector_store(file_path, store_name, force_reload=False):
+
         # Check if we need to force reload the vector store (e.g., when the PDF changes)
         if force_reload or not os.path.exists(f"{store_name}.pkl"):
             text_splitter = RecursiveCharacterTextSplitter(
@@ -118,13 +125,9 @@ def page1():
                 st.error("File not found. Please check the file path.")
                 return
     
-            VectorStore = load_vector_store(pdf_path, "my_vector_store", force_reload=False)
-    
-    
-            if "chat_history" not in st.session_state:
-                st.session_state['chat_history'] = []
-    
-            display_chat_history(st.session_state['chat_history'])
+            VectorStore = load_vector_store(pdf_path, "vector_store_page1", force_reload=False)
+
+            display_chat_history(st.session_state['chat_history1'])
     
             st.write("<!-- Start Spacer -->", unsafe_allow_html=True)
             st.write("<div style='flex: 1;'></div>", unsafe_allow_html=True)
@@ -180,11 +183,11 @@ def page1():
                 # You can use Streamlit's text function to display the timing
                 st.text(f"Response time: {duration:.2f} seconds")
     
-                st.session_state['chat_history'].append(("Bot", response, "new"))
+                st.session_state['chat_history1'].append(("Bot", response, "new"))
     
     
                 # Display new messages at the bottom
-                new_messages = st.session_state['chat_history'][-2:]
+                new_messages = st.session_state['chat_history1'][-2:]
                 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)
@@ -194,7 +197,7 @@ def page1():
                 query = ""
     
             # Mark all messages as old after displaying
-            st.session_state['chat_history'] = [(sender, msg, "old") for sender, msg, _ in st.session_state['chat_history']]
+            st.session_state['chat_history_page1'] = [(sender, msg, "old") for sender, msg, _ in st.session_state['chat_history_page1']]
 
     except Exception as e:
         st.error(f"Upsi, an unexpected error occurred: {e}")
@@ -232,13 +235,11 @@ def page2():
                 st.error("File not found. Please check the file path.")
                 return
     
-            VectorStore = load_vector_store(pdf_path2, "my_vector_store", force_reload=False)
-    
+            VectorStore = load_vector_store(pdf_path2, "vector_store_page2", force_reload=False)
     
-            if "chat_history" not in st.session_state:
-                st.session_state['chat_history'] = []
+
     
-            display_chat_history(st.session_state['chat_history'])
+            display_chat_history(st.session_state['chat_history_page1'])
     
             st.write("<!-- Start Spacer -->", unsafe_allow_html=True)
             st.write("<div style='flex: 1;'></div>", unsafe_allow_html=True)
@@ -272,7 +273,7 @@ def page2():
 
         
             if query:
-                st.session_state['chat_history'].append(("User", query, "new"))
+                st.session_state['chat_history_page2'].append(("User", query, "new"))
 
                 # Start timing
                 start_time = time.time()
@@ -294,11 +295,11 @@ def page2():
                 # You can use Streamlit's text function to display the timing
                 st.text(f"Response time: {duration:.2f} seconds")
     
-                st.session_state['chat_history'].append(("Bot", response, "new"))
+                st.session_state['chat_history_page2'].append(("Bot", response, "new"))
     
     
                 # Display new messages at the bottom
-                new_messages = st.session_state['chat_history'][-2:]
+                new_messages = st.session_state['chat_history_page2'][-2:]
                 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)
@@ -308,7 +309,7 @@ def page2():
                 query = ""
     
             # Mark all messages as old after displaying
-            st.session_state['chat_history'] = [(sender, msg, "old") for sender, msg, _ in st.session_state['chat_history']]
+            st.session_state['chat_history_page2'] = [(sender, msg, "old") for sender, msg, _ in st.session_state['chat_history_page2']]
 
     except Exception as e:
         st.error(f"Upsi, an unexpected error occurred: {e}")
@@ -329,9 +330,9 @@ def main():
         st.write('Made with ❤️ by BinDoc GmbH')
 
     # Main area content based on page selection
-    if page == "Main ChatBot":
+    if page == "Page 1":
         page1()
-    elif page == "Kodierhilfe":
+    elif page == "Page 2":
         page2()
 
 if __name__ == "__main__":