Update utils/database.py

utils/database.py

@@ -236,6 +236,26 @@ def force_recreate_collections_tables(conn: sqlite3.Connection) -> bool:
         st.error(f"Error recreating collections tables: {e}")
         return False
 
+
+def get_existing_vector_store(document_ids: List[int]) -> Optional[FAISS]:
+    """Retrieve existing vector store if available."""
+    try:
+        if st.session_state.get('vector_store'):
+            current_docs = set(document_ids)
+            stored_docs = set(
+                metadata['document_id'] 
+                for metadata in st.session_state.vector_store.docstore.get_all_metadatas()
+            )
+            
+            # If the document sets match, reuse existing vector store
+            if current_docs == stored_docs:
+                return st.session_state.vector_store
+                
+        return None
+    except Exception as e:
+        st.error(f"Error checking vector store: {e}")
+        return None
+        
 def get_uncategorized_documents(conn: sqlite3.Connection) -> List[Dict]:
     """Get documents that aren't in any collection."""
     try: