faster

src/legisqa_local/app.py

@@ -31,13 +31,16 @@ def main():
     setup_environment()
     logger.info("✅ Environment setup complete")
     
-    # Setup ChromaDB (download if needed)
-    logger.info("💾 Inspecting ChromaDB...")
-    inspect_chromadb()
-    logger.info("✅ ChromaDB inspection complete")
+    # Setup ChromaDB (download if needed) - run once per session
+    if "chromadb_inspected" not in st.session_state:
+        logger.info("💾 Inspecting ChromaDB...")
+        inspect_chromadb()
+        logger.info("✅ ChromaDB inspection complete")
+        st.session_state.chromadb_inspected = True
     
     # Initialize vectorstore (load once and cache in session state)
-    initialize_vectorstore()
+    if "vectorstore" not in st.session_state:
+        initialize_vectorstore()
     
     # Main content
     st.title(":classical_building: LegisQA :classical_building:")

src/legisqa_local/components/sidebar.py

@@ -12,17 +12,17 @@ def render_chromadb_status():
     
     vectorstore = get_vectorstore()
     if vectorstore is not None:
-        try:
-            # Test the vectorstore to get document count
-            count = vectorstore._collection.count()
+        # Use cached count from session state to avoid network calls on every rerun
+        count = st.session_state.get("vectorstore_count")
+        if count is not None:
             st.success("✅ ChromaDB Ready")
             st.caption(f"📊 {count:,} documents loaded")
             
             config = get_chroma_config()
             st.caption(f"📁 Collection: {config['collection_name']}")
-        except Exception as e:
+        else:
             st.warning("⚠️ ChromaDB Loaded (verification failed)")
-            st.caption(f"Error: {str(e)[:50]}...")
+            st.caption("Could not retrieve document count")
     else:
         st.info("⏳ ChromaDB Loading...")
         st.caption("Vectorstore is being initialized")

src/legisqa_local/core/vectorstore.py

@@ -36,13 +36,15 @@ def initialize_vectorstore():
         vectorstore = load_vectorstore()
         st.session_state.vectorstore = vectorstore
         
-        # Test the vectorstore to verify it's working
+        # Test the vectorstore to verify it's working and cache the count
         collection = vectorstore._collection
         count = collection.count()
+        st.session_state.vectorstore_count = count  # Cache count to avoid repeated network calls
         logger.info(f"✅ Vectorstore initialized and cached - {count} documents available")
         
     except Exception as e:
         logger.error(f"❌ Error initializing vectorstore: {e}")
+        st.session_state.vectorstore_count = None  # Mark as failed
         # Don't raise - let the app continue and show loading message to users