Update app.py

app.py

@@ -29,6 +29,12 @@ def main():
     if 'voo_answer' not in st.session_state:
         st.session_state['voo_answer'] = None
 
+    # Initialize session state for tracking page navigation
+    if 'current_page' not in st.session_state:
+        st.session_state['current_page'] = None
+    if 'previous_page' not in st.session_state:
+        st.session_state['previous_page'] = None
+
     if uploaded_files:
         if st.sidebar.button("Add Docs to Data Bank"):
             files_info = save_uploaded_files(uploaded_files)
@@ -48,6 +54,21 @@ def main():
     st.sidebar.title("Page Navigation")
     page = st.sidebar.radio("Select a page", tuple(pages.keys()))
 
+    # Update current page in session state
+    st.session_state['current_page'] = page
+
+    # Check if the user has navigated to a different page
+    if st.session_state['current_page'] != st.session_state['previous_page']:
+        # Display session state content if the user has navigated away and returned
+        if st.session_state['data_bank_answer'] and st.session_state['current_page'] == "Chat with RSCA":
+            st.write(st.session_state['data_bank_answer'])
+        if st.session_state['uploaded_docs_answer'] and st.session_state['current_page'] == "Chat with Uploaded Docs":
+            st.write(st.session_state['uploaded_docs_answer'])
+        if st.session_state['voo_answer'] and st.session_state['current_page'] == "Chat with VOO":
+            st.write(st.session_state['voo_answer'])
+        # Update the previous page to the current page
+        st.session_state['previous_page'] = st.session_state['current_page']
+
     # Initialize session state for summarization results if not already set
     if 'summaries' not in st.session_state:
         st.session_state['summaries'] = {}
@@ -93,10 +114,9 @@ def page_qna(uploaded_files):
     user_query = st.text_area("Enter your question here:", height=200)
     if st.button('Get Answer'):
         if user_query:
-            if 'data_bank_answer' not in st.session_state or st.session_state['data_bank_answer'] is None:
-                answer = handle_query(user_query)
-                st.session_state['data_bank_answer'] = answer  # Store the answer in session state
-            st.write(st.session_state['data_bank_answer'])
+            answer = handle_query(user_query)
+            st.session_state['data_bank_answer'] = answer  # Store the answer in session state
+            st.write(answer)
         else:
             st.error("Please enter a question to get an answer.")
     # Display stored answer if it exists
@@ -109,10 +129,9 @@ def page_chat_with_uploaded_docs(uploaded_files):
     user_query = st.text_area("Enter your question here:", height=200)
     if st.button('Get Answer'):
         if user_query:
-            if 'uploaded_docs_answer' not in st.session_state or st.session_state['uploaded_docs_answer'] is None:
-                answer = handle_uploaded_docs_query(user_query, st.session_state['uploaded_collection_name'])
-                st.session_state['uploaded_docs_answer'] = answer  # Store the answer in session state
-            st.write(st.session_state['uploaded_docs_answer'])
+            answer = handle_uploaded_docs_query(user_query, st.session_state['uploaded_collection_name'])
+            st.session_state['uploaded_docs_answer'] = answer  # Store the answer in session state
+            st.write(answer)
         else:
             st.error("Please enter a question to get an answer.")
     # Display stored answer if it exists
@@ -132,10 +151,9 @@ def page_chat_with_voo(uploaded_files):
     user_query = st.text_area("Enter your question here:", height=200)
     if st.button('Get Answer'):
         if user_query:
-            if 'voo_answer' not in st.session_state or st.session_state['voo_answer'] is None:
-                answer = handle_voo_query(user_query)
-                st.session_state['voo_answer'] = answer  # Store the answer in session state
-            st.write(st.session_state['voo_answer'])
+            answer = handle_voo_query(user_query)
+            st.session_state['voo_answer'] = answer  # Store the answer in session state
+            st.write(answer)
         else:
             st.error("Please enter a question to get an answer.")
     # Display stored answer if it exists