Create components/collection_manager.py

components/collection_manager.py

@@ -0,0 +1,95 @@
+# components/collection_manager.py
+
+def display_enhanced_collections():
+    """Enhanced collections management interface."""
+    col1, col2 = st.columns([1, 2])
+    
+    with col1:
+        st.subheader("Collections")
+        
+        # Search collections
+        search_query = st.text_input("🔍 Search collections", key="collection_search")
+        
+        # Create new collection button
+        if st.button("+ New Collection", type="primary", use_container_width=True):
+            st.session_state.show_collection_dialog = True
+        
+        # Collection list with stats
+        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()]
+        
+        st.divider()
+        
+        for collection in collections:
+            with st.container():
+                col_button = st.button(
+                    f"📁 {collection['name']}",
+                    key=f"col_{collection['id']}",
+                    use_container_width=True
+                )
+                
+                # Show collection stats
+                stats_cols = st.columns(3)
+                with stats_cols[0]:
+                    st.caption(f"📄 {collection['doc_count']}")
+                with stats_cols[1]:
+                    st.caption(f"📅 {collection['created_at'][:10]}")
+                with stats_cols[2]:
+                    st.caption("⚙️ Manage")
+                
+                if col_button:
+                    st.session_state.selected_collection = collection
+                    st.rerun()
+    
+    with col2:
+        if st.session_state.get('selected_collection'):
+            display_collection_content(st.session_state.selected_collection)
+        else:
+            st.info("Select a collection to view its contents")
+
+def display_collection_content(collection):
+    """Display enhanced collection contents."""
+    st.subheader(collection['name'])
+    
+    # Collection actions
+    col1, col2, col3, col4 = st.columns(4)
+    with col1:
+        st.button("📥 Add Documents", use_container_width=True)
+    with col2:
+        st.button("✏️ Edit Collection", use_container_width=True)
+    with col3:
+        st.button("📤 Export", use_container_width=True)
+    with col4:
+        st.button("🗑️ Delete", use_container_width=True)
+    
+    # Document search within collection
+    search_query = st.text_input("🔍 Search documents in collection")
+    
+    # Filters
+    with st.expander("Filters"):
+        filter_cols = st.columns(3)
+        with filter_cols[0]:
+            st.date_input("Date Range")
+        with filter_cols[1]:
+            st.selectbox("Sort By", ["Name", "Date", "Size"])
+        with filter_cols[2]:
+            st.selectbox("View", ["List", "Grid"])
+    
+    # Document list
+    if search_query:
+        documents = search_documents(
+            st.session_state.db_conn,
+            search_query,
+            collection['id']
+        )
+    else:
+        documents = get_collection_documents(
+            st.session_state.db_conn,
+            collection['id']
+        )
+    
+    for doc in documents:
+        with st.expander(f"📄 {doc['name']}", expanded=False):
+            enhanced_document_preview(doc)