Update components/collection_manager.py

components/collection_manager.py

@@ -1,6 +1,8 @@
 # components/collection_manager.py
 
 
+# components/collection_manager.py
+
 def show_collection_creation_dialog():
     """Show dialog for creating a new collection."""
     with st.form("create_collection"):
@@ -26,51 +28,25 @@ def show_collection_creation_dialog():
                 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.session_state.show_collection_dialog = False
             st.rerun()
 
-def show_collection_details(collection):
-    """Show collection details and management interface."""
-    st.markdown(f"### 📁 {collection['name']}")
+def enhanced_document_preview(doc):
+    """Enhanced document preview with actions."""
+    col1, col2 = st.columns([3, 1])
     
-    # 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,
+            "Preview",
+            value=doc['content'][:1000] + "..." if len(doc['content']) > 1000 else doc['content'],
+            height=200,
             disabled=True
         )
+    
+    with col2:
+        st.button("🗑️ Remove", key=f"remove_{doc['id']}", use_container_width=True)
+        st.button("📋 Copy", key=f"copy_{doc['id']}", use_container_width=True)
+        st.button("💬 Chat", key=f"chat_{doc['id']}", use_container_width=True)
 
 def display_enhanced_collections():
     """Enhanced collections management interface."""