Update pages/document_store.py

pages/document_store.py

@@ -1,8 +1,13 @@
 # pages/document_store.py
-
 import streamlit as st
-from utils.collections import *
-from utils.database import create_connection
+from utils.database import (
+    create_collection,
+    get_collections,
+    get_collection_documents,
+    delete_collection,
+    add_document_to_collection,
+    process_and_store_document
+)
 
 def display_collection_manager():
     """Display the collection management interface."""
@@ -15,10 +20,10 @@ def display_collection_manager():
     with st.expander("Create New Collection", expanded=False):
         col1, col2 = st.columns([3, 1])
         with col1:
-            name = st.text_input("Collection Name")
-            description = st.text_area("Description")
+            name = st.text_input("Collection Name", key="new_collection_name")
+            description = st.text_area("Description", key="new_collection_desc")
         with col2:
-            if st.button("Create Collection") and name:
+            if st.button("Create Collection", key="create_collection_btn") and name:
                 if create_collection(conn, name, description):
                     st.success(f"Collection '{name}' created!")
                     st.rerun()
@@ -63,7 +68,7 @@ def handle_document_upload_with_collection(uploaded_files, collection_id):
     conn = st.session_state.db_conn
     
     for uploaded_file in uploaded_files:
-        # Your existing document processing code here
+        # Process and store the document
         document_id = process_and_store_document(uploaded_file)
         
         if document_id:
@@ -79,4 +84,11 @@ if 'show_upload' not in st.session_state:
 if 'selected_collection' not in st.session_state:
     st.session_state.selected_collection = None
 if 'editing_collection' not in st.session_state:
-    st.session_state.editing_collection = None
+    st.session_state.editing_collection = None
+
+# Add this if you need it for the page
+def main():
+    display_collection_manager()
+
+if __name__ == "__main__":
+    main()