Update app.py

app.py

@@ -29,12 +29,6 @@ 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)
@@ -54,21 +48,6 @@ 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'] = {}
@@ -111,12 +90,13 @@ def page_summarization(uploaded_files):
 def page_qna(uploaded_files):
     """Page for Q&A functionality."""
     st.title("Chat with RSCA")
-    user_query = st.text_area("Enter your question here:", height=200)
+    user_query = st.text_area("Enter your question here:", height=300)
     if st.button('Get Answer'):
         if user_query:
-            answer = handle_query(user_query)
-            st.session_state['data_bank_answer'] = answer  # Store the answer in session state
-            st.write(answer)
+            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'])
         else:
             st.error("Please enter a question to get an answer.")
     # Display stored answer if it exists
@@ -126,12 +106,13 @@ def page_qna(uploaded_files):
 def page_chat_with_uploaded_docs(uploaded_files):
     """Page for chatting with uploaded documents."""
     st.title("Chat with Uploaded Documents")
-    user_query = st.text_area("Enter your question here:", height=200)
+    user_query = st.text_area("Enter your question here:", height=300)
     if st.button('Get Answer'):
         if user_query:
-            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)
+            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'])
         else:
             st.error("Please enter a question to get an answer.")
     # Display stored answer if it exists
@@ -148,12 +129,13 @@ def page_chat_with_uploaded_docs(uploaded_files):
 def page_chat_with_voo(uploaded_files):
     """Page for chatting with VOO documents."""
     st.title("Chat with VOO")
-    user_query = st.text_area("Enter your question here:", height=200)
+    user_query = st.text_area("Enter your question here:", height=300)
     if st.button('Get Answer'):
         if user_query:
-            answer = handle_voo_query(user_query)
-            st.session_state['voo_answer'] = answer  # Store the answer in session state
-            st.write(answer)
+            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'])
         else:
             st.error("Please enter a question to get an answer.")
     # Display stored answer if it exists