Update app.py

app.py

@@ -4,16 +4,14 @@ import uuid
 import streamlit as st
 from dotenv import load_dotenv
 from qdrant_client import models
-from langchain_community.vectorstores import Qdrant
 
-
-from utils import setup_openai_embeddings,setup_qdrant_client,delete_collection,is_document_embedded
+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,retrieve_documents_from_collection
+from preprocess import split_documents, update_metadata, load_documents_OCR
+from retrieve import retrieve_documents, retrieve_documents_from_collection
 from summarize import summarize_documents
 
-
+# Load environment variables
 load_dotenv()
 
 def main():
@@ -23,6 +21,12 @@ 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 uploaded_files:
         if st.sidebar.button("Add Docs to Data Bank"):
             files_info = save_uploaded_files(uploaded_files)
@@ -34,7 +38,7 @@ def main():
 
     pages = {
         "Lex Document Summarization": page_summarization,
-        "Chat with RSCA": page_qna,
+        "Chat with Data Bank": page_qna,
         "Chat with Uploaded Docs": page_chat_with_uploaded_docs
     }
 
@@ -75,7 +79,10 @@ def page_summarization(uploaded_files):
                             st.session_state['summaries'][original_name] = summary  # Store summary in session state
                         except Exception as e:
                             st.error(f"Failed to summarize {original_name}: {str(e)}")
-                    st.text_area("", value=st.session_state['summaries'][original_name], height=200, key=f"summary_{original_name}")
+                    if original_name in st.session_state['summaries']:
+                        st.text_area("", value=st.session_state['summaries'][original_name], height=1000, key=f"summary_{original_name}")
+                    else:
+                        st.error(f"No summary found for {original_name}. Please click the summarize button.")
 
 def page_qna(uploaded_files):
     """Page for Q&A functionality."""
@@ -84,10 +91,14 @@ def page_qna(uploaded_files):
     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)
         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."""
     st.title("Chat with Uploaded Documents")
@@ -95,9 +106,13 @@ def page_chat_with_uploaded_docs(uploaded_files):
     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)
         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'):
@@ -179,4 +194,4 @@ def delete_collection(collection_name, qdrant_url, qdrant_api_key):
         print("Failed to delete collection:", e)
 
 if __name__ == "__main__":
-    main()
+    main()