Update utils/database.py

utils/database.py

@@ -144,12 +144,16 @@ def verify_vector_store(vector_store):
         st.error(f"Vector store verification failed: {e}")
         return False
 
-
-
 def handle_document_upload(uploaded_files):
     """Handle document upload with improved chunking and progress tracking."""
     try:
         # Initialize session state variables
+        # Initialize persistence manager
+        persistence = PersistenceManager()
+        
+        # Generate a session ID based on timestamp and files
+        session_id = f"session_{datetime.now().strftime('%Y%m%d_%H%M%S')}"
+        
         if 'qa_system' not in st.session_state:
             st.session_state.qa_system = None
         if 'vector_store' not in st.session_state:
@@ -221,6 +225,8 @@ def handle_document_upload(uploaded_files):
             return
         
         progress_bar.progress(80)
+        # After chunking documents
+        persistence.save_chunks(chunks, chunk_metadatas, session_id)
         
         # Initialize vector store (90% progress)
         status_container.info("🔄 Initializing vector store...")
@@ -233,7 +239,9 @@ def handle_document_upload(uploaded_files):
         
         st.session_state.vector_store = vector_store
         progress_bar.progress(90)
-        
+        persistence.save_vector_store(vector_store, session_id)
+        # Store session ID in state
+        st.session_state.current_session_id = session_id
         # Initialize QA system (100% progress)
         status_container.info("🔄 Setting up QA system...")
         qa_system = initialize_qa_system(vector_store)
@@ -267,6 +275,7 @@ def handle_document_upload(uploaded_files):
         st.session_state.vector_store = None
         st.session_state.qa_system = None
         st.session_state.chat_ready = False
+        st.error(f"Error in document upload: {str(e)}")
         
     finally:
         if st.session_state.get('qa_system') is not None:
@@ -389,6 +398,7 @@ Accuracy: Double-check all information for accuracy and completeness before prov
         st.error(f"Error initializing QA system: {e}")
         return None
         
+
 # FAISS vector store initialization
 def initialize_faiss(embeddings, documents, document_names):
     """Initialize FAISS vector store."""