Update app.py

app.py

@@ -21,14 +21,6 @@ def main():
     if 'uploaded_collection_name' not in st.session_state:
         st.session_state['uploaded_collection_name'] = None
 
-    # Initialize session state for storing answers if not already set
-    if 'data_bank_answer' not in st.session_state:
-        st.session_state['data_bank_answer'] = None
-    if 'uploaded_docs_answer' not in st.session_state:
-        st.session_state['uploaded_docs_answer'] = None
-    if 'voo_answer' not in st.session_state:
-        st.session_state['voo_answer'] = None
-
     if uploaded_files:
         if st.sidebar.button("Add Docs to Data Bank"):
             files_info = save_uploaded_files(uploaded_files)
@@ -93,15 +85,10 @@ def page_qna(uploaded_files):
     user_query = st.text_area("Enter your question here:", height=300)
     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.write(answer)
         else:
             st.error("Please enter a question to get an answer.")
-    # Display stored answer if it exists
-    if st.session_state['data_bank_answer']:
-        st.write(st.session_state['data_bank_answer'])
 
 def page_chat_with_uploaded_docs(uploaded_files):
     """Page for chatting with uploaded documents."""
@@ -109,15 +96,10 @@ def page_chat_with_uploaded_docs(uploaded_files):
     user_query = st.text_area("Enter your question here:", height=300)
     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.write(answer)
         else:
             st.error("Please enter a question to get an answer.")
-    # Display stored answer if it exists
-    if st.session_state['uploaded_docs_answer']:
-        st.write(st.session_state['uploaded_docs_answer'])
     
     if st.session_state['uploaded_collection_name']:
         if st.button('Delete Embedded Collection'):
@@ -132,15 +114,10 @@ def page_chat_with_voo(uploaded_files):
     user_query = st.text_area("Enter your question here:", height=300)
     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.write(answer)
         else:
             st.error("Please enter a question to get an answer.")
-    # Display stored answer if it exists
-    if st.session_state['voo_answer']:
-        st.write(st.session_state['voo_answer'])
 
 def embed_documents_to_data_bank(files_info):
     """Function to embed documents into the data bank."""