Update app.py

app.py

@@ -25,17 +25,19 @@ with st.sidebar:
 
     # Fetch collection names for dropdown
     try:
-        existing_collections = db.list_collection_names()
-        existing_collections.append("Create New Collection")
-        selected_collection = st.selectbox("Choose MongoDB Collection", existing_collections, index=existing_collections.index("default_collection") if "default_collection" in existing_collections else 0)
-    except Exception as e:
-        st.error(f"Failed to list collections: {e}")
-        selected_collection = "default_collection"
-
-    if selected_collection == "Create New Collection":
-        selected_collection = st.sidebar.text_input("Enter Collection Name:")
-        if not selected_collection:
-            st.warning("⚠️ Enter a collection name to proceed.")
+        existing_categories = db["final_chunks"].distinct("collection_category") or []
+    except Exception:
+        existing_categories = []
+    existing_categories=sorted([c for c in existing_categories if c])+["Create New Category"]
+    selected_category = st.selectbox(
+        "Choose Category (collection_category)",
+        existing_categories,
+        index=existing_categories.index("Create New Category") if "Create New Category" in existing_categories else 0
+    )
+    if selected_category == "Create New Category":
+        selected_category = st.sidebar.text_input("Enter Category Name:")
+        if not selected_category:
+            st.warning("⚠️ Enter a category name to proceed.")
             st.stop()
 
     is_grant_app = st.toggle("Is this a Grant Application?", value=False)
@@ -50,10 +52,14 @@ if uploaded_file:
     st.success(f"Uploaded `{uploaded_file.name}`")
 
     modified_time = datetime.now().isoformat()
-    collection = db[selected_collection]
-
-    if collection.find_one({"metadata.title": uploaded_file.name}):
-        st.warning("⚠️ This file already exists in the collection. Skipping...")
+    collection = db['final_chunks']
+    already = final_col.find_one({
+        "metadata.title": uploaded_file.name,
+        "collection_category": selected_category
+    })
+
+    if already:
+        st.warning(f"⚠️ `{uploaded_file.name}` already exists in category `{selected_category}`. Skipping…")
     else:
         st.write("⏳ Processing with DocumentChunker...")
         chunker = DocumentChunker()
@@ -61,6 +67,7 @@ if uploaded_file:
 
         if chunks:
             for chunk in chunks:
+                chunk['collection_category']=selected_category
                 chunk['metadata'].update({
                     "title": uploaded_file.name,
                     "uploaded_at": modified_time,
@@ -68,7 +75,7 @@ if uploaded_file:
                 })
                 collection.insert_one(chunk)
 
-            st.success(f"✅ {len(chunks)} chunks inserted into `{selected_collection}`")
+            st.success(f"✅ {len(chunks)} chunks inserted into `final_chunks` (category: `{selected_category}`)")
 
             # Show a few previews
             for i, c in enumerate(chunks[:3]):