Update app.py

app.py

@@ -243,6 +243,33 @@ def display_collections_tab():
         if st.session_state.get('selected_collection'):
             display_collection_details(st.session_state.selected_collection)
 
+def show_collection_dialog():
+    """Show dialog for creating/editing collections."""
+    with st.sidebar:
+        st.header("Collection Details")
+        
+        name = st.text_input("Collection Name")
+        description = st.text_area("Description")
+        
+        col1, col2 = st.columns(2)
+        with col1:
+            if st.button("Save", type="primary"):
+                if name:
+                    create_collection(
+                        st.session_state.db_conn,
+                        name,
+                        description
+                    )
+                    st.session_state.show_collection_dialog = False
+                    st.rerun()
+                else:
+                    st.error("Please enter a collection name")
+        
+        with col2:
+            if st.button("Cancel"):
+                st.session_state.show_collection_dialog = False
+                st.rerun()
+                
 def display_collection_details(collection):
     """Display details and documents for a selected collection."""
     st.header(collection['name'])
@@ -396,8 +423,63 @@ def get_response_with_sources(prompt, qa_system):
     
     return list(sources), response.content
     
+def display_documents_tab():
+    """Enhanced documents tab with search and preview."""
+    st.header("Document Library")
+    
+    # Search and filter section
+    col1, col2, col3 = st.columns([2, 1, 1])
+    with col1:
+        search_query = st.text_input("🔍 Search documents")
+    with col2:
+        date_filter = st.date_input("Date Filter")
+    with col3:
+        sort_by = st.selectbox("Sort By", ["Name", "Date", "Size"])
+    
+    # Advanced filters
+    with st.expander("Advanced Filters"):
+        filter_cols = st.columns(3)
+        with filter_cols[0]:
+            collections_filter = st.multiselect(
+                "Collections",
+                options=[c['name'] for c in get_collections(st.session_state.db_conn)]
+            )
+        with filter_cols[1]:
+            date_range = st.date_input("Date Range", value=())
+        with filter_cols[2]:
+            st.checkbox("Show only uncategorized")
+    
+    # Build filters dictionary
+    filters = {
+        'collections': collections_filter,
+        'date_range': date_range if len(date_range) == 2 else None
+    }
+    
+    # Get and display documents
+    if search_query:
+        documents = search_documents(
+            st.session_state.db_conn,
+            search_query,
+            filters=filters
+        )
+        if documents:
+            st.success(f"Found {len(documents)} matching documents")
+        else:
+            st.info("No documents found matching your search")
+    else:
+        documents = get_all_documents(st.session_state.db_conn)
+    
+    # Display documents
+    for doc in documents:
+        with st.expander(f"📄 {doc['name']}", expanded=False):
+            enhanced_document_preview(doc)
+
 def main():
-    st.set_page_config(layout="wide", page_title="SYNAPTYX - RFP Analysis Agent")
+    st.set_page_config(
+        layout="wide",
+        page_title="SYNAPTYX - RFP Analysis Agent",
+        initial_sidebar_state="expanded"
+    )
     
     if not initialize_database():
         st.error("Failed to initialize database. Please contact support.")
@@ -406,16 +488,23 @@ def main():
     initialize_session_state()
     display_top_bar()
 
-    # Main navigation tabs
+    # Main tabs
     tab1, tab2, tab3 = st.tabs(["Chat", "Collections", "Documents"])
-
+    
     with tab1:
         display_chat_interface_with_sources()
     
     with tab2:
-        display_collections_tab()
+        display_enhanced_collections()
     
     with tab3:
         display_documents_tab()
+
+    # Handle collection dialog if needed
+    if st.session_state.get('show_collection_dialog'):
+        show_collection_dialog()
+
+
+
 if __name__ == "__main__":
-    main()
+    main()