Update components/document_store.py

components/document_store.py

@@ -13,123 +13,17 @@ from utils.database import (
 )
 
 def display_documents_tab():
-    """Display the documents management interface."""
     col1, col2 = st.columns([1, 2])
     
     with col1:
         st.header("Collections")
         
-        # Create new collection - now directly in the interface, not in an expander
+        # Create new collection
         st.subheader("Create New Collection")
-        col_name = st.text_input("Collection Name")
-        col_desc = st.text_area("Description", height=100)
-        if st.button("Create Collection", use_container_width=True):
+        col_name = st.text_input("Collection Name", key="doc_store_collection_name")
+        col_desc = st.text_area("Description", height=100, key="doc_store_collection_desc")
+        if st.button("Create Collection", use_container_width=True, key="doc_store_create_btn"):
             if col_name:
                 if create_collection(st.session_state.db_conn, col_name, col_desc):
                     st.success(f"Collection '{col_name}' created!")
-                    st.rerun()
-                
-        # Divider
-        st.divider()
-        
-        # List all collections
-        st.subheader("Existing 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()
-        else:
-            st.info("No collections yet. Create your first collection above.")
-    
-    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.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()
-            ]
-
-        if documents:
-            for doc in documents:
-                with st.expander(f"📄 {doc['name']}", expanded=False):
-                    # Create columns for metadata and actions
-                    meta_col, action_col = st.columns([2, 1])
-                    
-                    with meta_col:
-                        st.write(f"**Uploaded:** {doc['upload_date']}")
-                        if doc.get('collections'):
-                            st.write(f"**Collections:** {', '.join(doc['collections'])}")
-                    
-                    with action_col:
-                        # Preview button
-                        if st.button("Preview Content", key=f"preview_{doc['id']}"):
-                            st.session_state[f"show_preview_{doc['id']}"] = True
-                            
-                        # Use for Chat button
-                        if st.button("Use for Chat", key=f"chat_{doc['id']}", type="primary"):
-                            st.session_state.current_document = doc['id']
-                            st.session_state.chat_ready = True
-                            st.rerun()
-                    
-                    # Collection management
-                    if st.session_state.get('selected_collection'):
-                        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']}", use_container_width=True):
-                                add_document_to_collection(
-                                    st.session_state.db_conn,
-                                    doc['id'],
-                                    current_collection_id
-                                )
-                                st.rerun()
-                        else:
-                            if st.button("Remove from Collection", key=f"remove_{doc['id']}", use_container_width=True):
-                                remove_from_collection(
-                                    st.session_state.db_conn,
-                                    doc['id'],
-                                    current_collection_id
-                                )
-                                st.rerun()
-                    
-                    # Show preview if button was clicked
-                    if st.session_state.get(f"show_preview_{doc['id']}", False):
-                        st.text_area(
-                            "Document Content",
-                            value=doc['content'][:1000] + "..." if len(doc['content']) > 1000 else doc['content'],
-                            height=200,
-                            disabled=True
-                        )
-        else:
-            if filter_option == "Selected Collection":
-                st.info("No documents in this collection. Upload documents or add existing ones to this collection.")
-            elif filter_option == "Uncategorized":
-                st.info("No uncategorized documents found.")
-            else:
-                st.info("No documents found. Upload some documents to get started!")
+                    st.rerun()