Update utils/database.py

utils/database.py

@@ -255,6 +255,45 @@ def handle_document_upload(uploaded_files):
         if st.session_state.get('qa_system'):
             time.sleep(5)
             progress_container.empty()    
+
+def display_vector_store_info():
+    """Display information about the current vector store state."""
+    if 'vector_store' not in st.session_state:
+        st.info("ℹ️ No documents loaded yet.")
+        return
+        
+    try:
+        # Get basic stats
+        test_query = vector_store.similarity_search("test", k=1)
+        doc_count = len(test_query)
+        
+        # Create an expander for detailed info
+        with st.expander("📊 Knowledge Base Status"):
+            col1, col2 = st.columns(2)
+            
+            with col1:
+                st.metric(
+                    label="Documents Loaded",
+                    value=doc_count
+                )
+                
+            with col2:
+                st.metric(
+                    label="System Status",
+                    value="Ready" if verify_vector_store(vector_store) else "Not Ready"
+                )
+            
+            # Display sample queries
+            if verify_vector_store(vector_store):
+                st.markdown("### 🔍 Sample Document Snippets")
+                sample_docs = vector_store.similarity_search("", k=3)
+                for i, doc in enumerate(sample_docs, 1):
+                    with st.container():
+                        st.markdown(f"**Snippet {i}:**")
+                        st.text(doc.page_content[:200] + "...")
+                        
+    except Exception as e:
+        st.error(f"Error displaying vector store info: {e}")
             
 def initialize_qa_system(vector_store):
     """Initialize QA system with proper chat handling."""