Create components/document_store.py

components/components/document_store.py

@@ -0,0 +1,96 @@
+# components/document_store.py
+
+import streamlit as st
+from typing import Dict, List, Optional, Any
+from utils.database import (
+    get_all_documents,
+    get_collections,
+    get_collection_documents,
+    create_collection,
+    add_document_to_collection,
+    remove_from_collection
+)
+
+def display_documents_tab():
+    """Display the documents management interface."""
+    col1, col2 = st.columns([1, 2])
+    
+    with col1:
+        st.header("Collections")
+        
+        # Create new collection
+        with st.expander("Create New Collection", expanded=False):
+            col_name = st.text_input("Collection Name")
+            col_desc = st.text_area("Description")
+            if st.button("Create Collection"):
+                if col_name:
+                    if create_collection(st.session_state.db_conn, col_name, col_desc):
+                        st.success(f"Collection '{col_name}' created!")
+                        st.rerun()
+        
+        # List all collections
+        collections = get_collections(st.session_state.db_conn)
+        if collections:
+            for col in collections:
+                if st.button(f"📁 {col['name']} ({col['doc_count']} documents)", 
+                           key=f"col_{col['id']}", use_container_width=True):
+                    st.session_state.selected_collection = col['id']
+                    st.rerun()
+    
+    with col2:
+        st.header("Documents")
+        
+        # Filter controls
+        col_filter, doc_search = st.columns([1, 2])
+        with col_filter:
+            filter_option = st.selectbox(
+                "View",
+                ["All Documents", "Uncategorized", "Selected Collection"]
+            )
+        with doc_search:
+            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'):
+            documents = get_collection_documents(st.session_state.db_conn, st.session_state.selected_collection)
+        else:
+            documents = get_all_documents(st.session_state.db_conn)
+            if filter_option == "Uncategorized":
+                documents = [doc for doc in documents if not doc['collections']]
+
+        # Display documents
+        for doc in documents:
+            with st.expander(f"📄 {doc['name']}", expanded=False):
+                # Document metadata
+                st.write(f"Uploaded: {doc['upload_date']}")
+                if doc.get('collections'):
+                    st.write(f"Collections: {', '.join(doc['collections'])}")
+                
+                # 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', [])]:
+                            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
+                                )
+                                st.rerun()
+                        else:
+                            if st.button("Remove from Collection", key=f"remove_{doc['id']}"):
+                                remove_from_collection(
+                                    st.session_state.db_conn,
+                                    doc['id'],
+                                    st.session_state.selected_collection
+                                )
+                                st.rerun()
+                
+                with col2:
+                    # Preview/Chat actions
+                    if st.button("Use for Chat", key=f"chat_{doc['id']}"):
+                        st.session_state.current_document = doc['id']
+                        st.session_state.chat_ready = True
+                        st.rerun()