Create components/document_store.py

components/components/document_store.py

@@ -1,14 +1,15 @@
 # components/document_store.py
 
 import streamlit as st
-from typing import Dict, List, Optional, Any
+from typing import Dict, List, Optional
 from utils.database import (
     get_all_documents,
     get_collections,
     get_collection_documents,
     create_collection,
     add_document_to_collection,
-    remove_from_collection
+    remove_from_collection,
+    search_documents
 )
 
 def display_documents_tab():
@@ -48,7 +49,7 @@ def display_documents_tab():
                 ["All Documents", "Uncategorized", "Selected Collection"]
             )
         with doc_search:
-            search_query = st.text_input("Search documents", placeholder="Enter search terms...")
+            search_query = st.text_input("🔍 Search documents", placeholder="Enter search terms...")
 
         # Get and display documents based on filter
         if filter_option == "Selected Collection" and st.session_state.get('selected_collection'):
@@ -56,9 +57,17 @@ def display_documents_tab():
         else:
             documents = get_all_documents(st.session_state.db_conn)
             if filter_option == "Uncategorized":
-                documents = [doc for doc in documents if not doc['collections']]
+                documents = [doc for doc in documents if not doc.get('collections')]
 
         # Display documents
+        if search_query:
+            # Filter documents based on search query
+            documents = [
+                doc for doc in documents 
+                if search_query.lower() in doc['name'].lower() or 
+                   search_query.lower() in doc['content'].lower()
+            ]
+
         for doc in documents:
             with st.expander(f"📄 {doc['name']}", expanded=False):
                 # Document metadata
@@ -66,17 +75,29 @@ def display_documents_tab():
                 if doc.get('collections'):
                     st.write(f"Collections: {', '.join(doc['collections'])}")
                 
+                # Document preview
+                with st.expander("Preview Content"):
+                    st.text_area(
+                        "",
+                        value=doc['content'][:1000] + "..." if len(doc['content']) > 1000 else doc['content'],
+                        height=200,
+                        disabled=True
+                    )
+                
                 # Document actions
                 col1, col2 = st.columns(2)
                 with col1:
                     # Collection management
                     if st.session_state.get('selected_collection'):
-                        if st.session_state.selected_collection not in [c['id'] for c in doc.get('collections', [])]:
+                        current_collection_id = st.session_state.selected_collection
+                        doc_collections = [c['id'] for c in doc.get('collections', [])]
+                        
+                        if current_collection_id not in doc_collections:
                             if st.button("Add to Collection", key=f"add_{doc['id']}"):
                                 add_document_to_collection(
                                     st.session_state.db_conn,
                                     doc['id'],
-                                    st.session_state.selected_collection
+                                    current_collection_id
                                 )
                                 st.rerun()
                         else:
@@ -84,12 +105,12 @@ def display_documents_tab():
                                 remove_from_collection(
                                     st.session_state.db_conn,
                                     doc['id'],
-                                    st.session_state.selected_collection
+                                    current_collection_id
                                 )
                                 st.rerun()
                 
                 with col2:
-                    # Preview/Chat actions
+                    # Chat action
                     if st.button("Use for Chat", key=f"chat_{doc['id']}"):
                         st.session_state.current_document = doc['id']
                         st.session_state.chat_ready = True