Update components/collection_manager.py

components/collection_manager.py

@@ -1,5 +1,77 @@
 # components/collection_manager.py
 
+
+def show_collection_creation_dialog():
+    """Show dialog for creating a new collection."""
+    with st.form("create_collection"):
+        st.subheader("Create New Collection")
+        name = st.text_input("Collection Name", key="new_col_name")
+        description = st.text_area("Description", key="new_col_desc")
+        
+        # Document selection
+        existing_docs = get_all_documents(st.session_state.db_conn)
+        if existing_docs:
+            st.subheader("Add Existing Documents")
+            selected_docs = st.multiselect(
+                "Select documents to add",
+                options=[doc['id'] for doc in existing_docs],
+                format_func=lambda x: next(doc['name'] for doc in existing_docs if doc['id'] == x)
+            )
+        
+        submitted = st.form_submit_button("Create Collection")
+        
+        if submitted and name:
+            collection_id = create_collection(st.session_state.db_conn, name, description)
+            if collection_id and selected_docs:
+                for doc_id in selected_docs:
+                    add_document_to_collection(st.session_state.db_conn, doc_id, collection_id)
+            st.success(f"Collection '{name}' created!")
+            st.rerun()
+
+def show_collection_details(collection):
+    """Show collection details and management interface."""
+    st.markdown(f"### 📁 {collection['name']}")
+    
+    # Collection Analytics
+    col1, col2 = st.columns(2)
+    with col1:
+        st.metric("Total Documents", collection['doc_count'])
+    with col2:
+        st.metric("Last Updated", collection['last_updated'])
+    
+    # Document List
+    documents = get_collection_documents(st.session_state.db_conn, collection['id'])
+    if documents:
+        with st.expander("Manage Documents", expanded=True):
+            selected_docs = []
+            for doc in documents:
+                col1, col2, col3 = st.columns([0.1, 0.7, 0.2])
+                with col1:
+                    selected = st.checkbox("", key=f"select_{doc['id']}")
+                    if selected:
+                        selected_docs.append(doc['id'])
+                with col2:
+                    st.write(f"📄 {doc['name']}")
+                with col3:
+                    if st.button("Preview", key=f"preview_{doc['id']}"):
+                        show_document_preview(doc)
+            
+            if selected_docs:
+                if st.button("Remove Selected", key="remove_selected"):
+                    for doc_id in selected_docs:
+                        remove_from_collection(st.session_state.db_conn, doc_id, collection['id'])
+                    st.rerun()
+
+def show_document_preview(document):
+    """Show document preview in a modal."""
+    with st.expander(f"Preview: {document['name']}", expanded=True):
+        st.text_area(
+            "Content",
+            value=document['content'][:1000] + "..." if len(document['content']) > 1000 else document['content'],
+            height=300,
+            disabled=True
+        )
+
 def display_enhanced_collections():
     """Enhanced collections management interface."""
     col1, col2 = st.columns([1, 2])