Update components/collection_manager.py

components/collection_manager.py

@@ -74,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()
         
@@ -133,19 +133,42 @@ def display_collection_content(collection):
         with filter_cols[2]:
             st.selectbox("View", ["List", "Grid"])
     
-    # Document list
+    # Document list with enhanced display
+    documents = get_collection_documents(st.session_state.db_conn, collection['id'])
     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']
-        )
+        documents = [doc for doc in documents 
+                    if search_query.lower() in doc['name'].lower() or 
+                       search_query.lower() in doc.get('content', '').lower()]
     
+    # Display documents in a clean, modern format
     for doc in documents:
-        with st.expander(f"📄 {doc['name']}", expanded=False):
-            enhanced_document_preview(doc)
+        with st.container():
+            doc_cols = st.columns([0.1, 2, 0.5])
+            with doc_cols[0]:
+                st.checkbox("", key=f"select_{doc['id']}")
+            with doc_cols[1]:
+                st.markdown(f"**{doc['name']}**")
+                st.caption(f"Uploaded: {doc.get('upload_date', 'N/A')}")
+            with doc_cols[2]:
+                st.button("Preview", key=f"preview_{doc['id']}", use_container_width=True)
+                st.button("Remove", key=f"remove_{doc['id']}", use_container_width=True)
+
+        # Document preview expander
+        if st.session_state.get(f"preview_{doc['id']}", False):
+            with st.expander("Document Preview", expanded=True):
+                st.text_area(
+                    "Content",
+                    value=doc['content'][:1000] + "..." if len(doc['content']) > 1000 else doc['content'],
+                    height=200,
+                    disabled=True
+                )
+
+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
+        )