Update app.py

app.py

@@ -8,13 +8,12 @@ from qdrant_client import models
 from utils import setup_openai_embeddings, setup_qdrant_client, delete_collection, is_document_embedded
 from embed import embed_documents_into_qdrant
 from preprocess import split_documents, update_metadata, load_documents_OCR
-from retrieve import  retrieve_documents_from_collection
+from retrieve import retrieve_documents_from_collection
 from summarize import summarize_documents
 
 # Load environment variables
 load_dotenv()
 
-
 def main():
     st.sidebar.title("PDF Management")
     uploaded_files = st.sidebar.file_uploader("Upload PDF files", type=["pdf"], accept_multiple_files=True)
@@ -94,9 +93,10 @@ def page_qna(uploaded_files):
     user_query = st.text_area("Enter your question here:", height=200)
     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
@@ -109,9 +109,10 @@ 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:
-            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
@@ -131,9 +132,10 @@ 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:
-            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