Update components/collection_manager.py

components/collection_manager.py

@@ -1,7 +1,15 @@
 # components/collection_manager.py
 
+import streamlit as st  # Import Streamlit here
 
-# components/collection_manager.py
+from utils.database import (
+    get_all_documents,
+    create_collection,
+    add_document_to_collection,
+    get_collections,
+    search_documents,
+    get_collection_documents,
+)
 
 def show_collection_creation_dialog():
     """Show dialog for creating a new collection."""
@@ -66,7 +74,7 @@ def display_enhanced_collections():
         collections = get_collections(st.session_state.db_conn)
         if search_query:
             collections = [c for c in collections 
-                         if search_query.lower() in c['name'].lower()]
+                if search_query.lower() in c['name'].lower()]
         
         st.divider()
         
@@ -140,4 +148,4 @@ def display_collection_content(collection):
     
     for doc in documents:
         with st.expander(f"📄 {doc['name']}", expanded=False):
-            enhanced_document_preview(doc)
+            enhanced_document_preview(doc)