Update src/app.py

src/app.py

@@ -357,64 +357,64 @@ with tab3:
 
 # --- TAB 4: KNOWLEDGE BASE ---
 with tab4:
-    st.header("🧠 Unit Knowledge Base")
+    st.header("🧠 Personal Knowledge Base")
+    st.info(f"Managing knowledge for: **{st.session_state.username}**")
     
-    is_admin = "admin" in st.session_state.roles
+    # We no longer check 'is_admin' for the whole tab
     kb_tab1, kb_tab2 = st.tabs(["📤 Add Documents", "🗂️ Manage Database"])
     
-    # --- SUB-TAB 1: UPLOAD ---
+    # --- SUB-TAB 1: UPLOAD (Unlocked for Everyone) ---
     with kb_tab1:
-        if is_admin:
-            st.subheader("Ingest New Knowledge")
-            uploaded_file = st.file_uploader("Upload Instructions, Manuals, or Logs", type=["pdf", "docx", "txt", "md"])
-            
-            col1, col2 = st.columns([1, 2])
-            with col1:
-                chunk_strategy = st.selectbox(
-                    "Chunking Strategy", 
-                    ["paragraph", "token", "page"],
-                    help="Paragraph: Manuals. Token: Dense text. Page: Forms."
+        st.subheader("Ingest New Knowledge")
+        uploaded_file = st.file_uploader("Upload Instructions, Manuals, or Logs", type=["pdf", "docx", "txt", "md"])
+        
+        col1, col2 = st.columns([1, 2])
+        with col1:
+            chunk_strategy = st.selectbox(
+                "Chunking Strategy", 
+                ["paragraph", "token", "page"],
+                help="Paragraph: Manuals. Token: Dense text. Page: Forms."
+            )
+        
+        if uploaded_file and st.button("Process & Add"):
+            with st.spinner("Analyzing and Indexing..."):
+                # 1. Save temp file
+                temp_path = rag_engine.save_uploaded_file(uploaded_file)
+                
+                # 2. Process into USER'S specific DB (st.session_state.username)
+                success, msg = rag_engine.process_and_add_document(
+                    temp_path, 
+                    st.session_state.username, 
+                    chunk_strategy
                 )
-            
-            if uploaded_file and st.button("Process & Add"):
-                with st.spinner("Analyzing and Indexing..."):
-                    # Use safe save + process
-                    temp_path = rag_engine.save_uploaded_file(uploaded_file)
-                    success, msg = rag_engine.process_and_add_document(
-                        temp_path, 
-                        st.session_state.username, 
-                        chunk_strategy
-                    )
-                    
-                    if success:
-                        st.success(msg)
-                        st.rerun() 
-                    else:
-                        st.error(f"Failed: {msg}")
-        else:
-            st.info("🔒 Only Admins can upload documents.")
+                
+                if success:
+                    st.success(msg)
+                    st.rerun() 
+                else:
+                    st.error(f"Failed: {msg}")
 
         st.divider()
         st.subheader("🔎 Quick Test")
-        test_query = st.text_input("Ask the brain something...")
+        test_query = st.text_input("Ask your brain something...")
         if test_query:
             results = rag_engine.search_knowledge_base(test_query, st.session_state.username)
+            if not results:
+                st.warning("No matches found.")
             for i, doc in enumerate(results):
-                # Using cleaned safe basename
                 src_name = os.path.basename(doc.metadata.get('source', '?'))
                 score = doc.metadata.get('relevance_score', 'N/A')
                 with st.expander(f"Match {i+1}: {src_name} (Score: {score})"):
                     st.write(doc.page_content)
 
-    # --- SUB-TAB 2: MANAGE ---
+    # --- SUB-TAB 2: MANAGE (Unlocked for Everyone) ---
     with kb_tab2:
         st.subheader("🗄️ Database Inventory")
         
-        # 1. Fetch current docs
         docs = rag_engine.list_documents(st.session_state.username)
         
         if not docs:
-            st.info("Knowledge Base is empty.")
+            st.info("Your Knowledge Base is empty.")
         else:
             st.markdown(f"**Total Documents:** {len(docs)}")
             
@@ -425,22 +425,20 @@ with tab4:
                 with c2:
                     st.caption(f"{doc['chunks']} chunks")
                 with c3:
-                    if is_admin:
-                        if st.button("🗑️ Delete", key=doc['source']):
-                            with st.spinner("Deleting..."):
-                                success, msg = rag_engine.delete_document(st.session_state.username, doc['source'])
-                                if success:
-                                    st.success(msg)
-                                    st.rerun()
-                                else:
-                                    st.error(msg)
-                    else:
-                        st.caption("Read Only")
+                    # Allow ANY user to delete their OWN documents
+                    if st.button("🗑️ Delete", key=doc['source']):
+                        with st.spinner("Deleting..."):
+                            success, msg = rag_engine.delete_document(st.session_state.username, doc['source'])
+                            if success:
+                                st.success(msg)
+                                st.rerun()
+                            else:
+                                st.error(msg)
 
-        if is_admin and docs:
             st.divider()
             with st.expander("🚨 Danger Zone"):
-                if st.button("☢️ RESET ENTIRE DATABASE", type="primary"):
+                # Allow ANY user to reset their OWN database
+                if st.button("☢️ RESET MY DATABASE", type="primary"):
                     success, msg = rag_engine.reset_knowledge_base(st.session_state.username)
                     if success:
                         st.success(msg)