Update components/document_store.py

components/document_store.py

@@ -19,17 +19,21 @@ def display_documents_tab():
     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()
+        # Create new collection - now directly in the interface, not in an expander
+        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):
+            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:
@@ -37,6 +41,8 @@ def display_documents_tab():
                            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")
@@ -68,32 +74,35 @@ def display_documents_tab():
                    search_query.lower() in doc['content'].lower()
             ]
 
-        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 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:
+        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']}"):
+                            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'],
@@ -101,17 +110,26 @@ def display_documents_tab():
                                 )
                                 st.rerun()
                         else:
-                            if st.button("Remove from Collection", key=f"remove_{doc['id']}"):
+                            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()
-                
-                with col2:
-                    # 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
-                        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!")