Update app.py

app.py

@@ -672,8 +672,6 @@ def side():
 
 
         st.sidebar.header("TrustVault®")
-       
-        st.subheader("Saved Documents")
         st.sidebar.markdown("""
             <style>
             .scrollable-container {
@@ -686,70 +684,105 @@ def side():
             }
             .button-container {
                 display: flex;
-                justify-content: flex-end;
+                flex-direction: column;
                 margin-top: 10px;
             }
             </style>
         """, unsafe_allow_html=True)
         
-        # Scrollable container for saved documents
+        # Sidebar: Saved Documents
         st.sidebar.markdown('<div class="scrollable-container">', unsafe_allow_html=True)
         try:
+            # Fetch and display saved documents
             docs = db.child("users").child(st.session_state["wix_user_id"]).child("KnowledgeBase").get().val()
+            doc_names = []
+            doc_ids = []
             if docs:
                 for doc_id, doc_data in docs.items():
-                    st.sidebar.markdown(f"**{doc_data.get('content', 'Document')}**")
-                    if st.sidebar.button("Delete", key=f"delete_{doc_id}"):
-                        delete_document(st.session_state["wix_user_id"], doc_id)
+                    doc_name = doc_data.get('content', 'Document')
+                    st.sidebar.markdown(f"**{doc_name}**")
+                    doc_names.append(doc_name)
+                    doc_ids.append(doc_id)
             else:
                 st.sidebar.write("No saved documents found.")
         except Exception as e:
             st.sidebar.error(f"Error fetching documents: {e}")
         st.sidebar.markdown('</div>', unsafe_allow_html=True)
         
-        # Buttons aligned beneath the text area
+        # Below saved documents: Upload and Delete buttons
         st.sidebar.markdown('<div class="button-container">', unsafe_allow_html=True)
-        uploaded_file = st.file_uploader("")
         
-        # Compact Upload Button
+        # File uploader
+        uploaded_file = st.sidebar.file_uploader("", key="file_uploader")
+        
+        # Upload button
         if st.sidebar.button("Upload"):
             if uploaded_file:
                 save_document(st.sess, uploaded_file.name)
             else:
                 st.sidebar.warning("Please select a file to upload.")
+        
+        # Multi-select for deleting documents
+        if docs:
+            selected_docs = st.sidebar.multiselect("Select documents to delete", options=doc_names)
+            if st.sidebar.button("Delete Selected"):
+                for doc_name in selected_docs:
+                    # Find the doc_id corresponding to the doc_name
+                    doc_id_to_delete = None
+                    for doc_id, doc_data in docs.items():
+                        if doc_data.get('content', '') == doc_name:
+                            doc_id_to_delete = doc_id
+                            break
+                    if doc_id_to_delete:
+                        delete_document(st.session_state["wix_user_id"], doc_id_to_delete)
+                st.sidebar.success("Selected documents deleted.")
+        else:
+            st.sidebar.write("No documents to delete.")
+        
         st.sidebar.markdown('</div>', unsafe_allow_html=True)
-
         
         # Show Trust Builders Section
         st.subheader("Show My TrustBuilders®")
-        search_query = st.text_input("Search by keyword")
+        
+        # Arrange text inputs and Show button
+        col1, col2, col3 = st.columns([1, 1, 0.5])
+        with col1:
+            search_query = st.text_input("Search by keyword")
+        with col2:
+            brand = st.text_input("Brand/Product/Person", key="brand_input_show")
+        with col3:
+            st.write("")  # Empty space to align the button
+            if st.button("Show"):
+                st.write(f"Filtering TrustBuilders for keyword='{search_query}', brand='{brand}', and bucket='{selected_bucket}'")
+                trustbuilders = fetch_trustbuilders(st.session_state["wix_user_id"])
+                results = []
+                for trustbuilder_id, data in trustbuilders.items():
+                    content = data.get("content", "")
+                    bucket = data.get("bucket", "")
+                    tb_brand = data.get("brand", "")
+                    if ((selected_bucket == bucket or not selected_bucket) and
+                        (not search_query or search_query.lower() in content.lower()) and
+                        (not brand or brand.lower() in tb_brand.lower())):
+                        results.append(f"{bucket}: {content}")
+                if results:
+                    for result in results:
+                        st.markdown(f"- {result}")
+                else:
+                    st.write("No TrustBuilders found.")
+        
+        # Trust Buckets selectbox below
         trust_buckets = ["Stability", "Development", "Relationship", "Benefit", "Vision", "Competence"]
         selected_bucket = st.selectbox("Select Trust Bucket", trust_buckets)
         
-        if st.button("Show"):
-            st.write(f"Filtering TrustBuilders for keyword='{search_query}' and bucket='{selected_bucket}'")
-            trustbuilders = fetch_trustbuilders(st.session_state["wix_user_id"])
-            results = []
-            for trustbuilder_id, data in trustbuilders.items():
-                content = data.get("content", "")
-                bucket = data.get("bucket", "")
-                if (selected_bucket == bucket or not selected_bucket) and (not search_query or search_query.lower() in content.lower()):
-                    results.append(f"{bucket}: {content}")
-            if results:
-                for result in results:
-                    st.markdown(f"- {result}")
-            else:
-                st.write("No TrustBuilders found.")
-        
         # Save Trust Builders Section
         st.subheader("Save TrustBuilders®")
-        brand = st.text_input("Brand/Product/Person", key="brand_input")
+        brand_save = st.text_input("Brand/Product/Person", key="brand_input_save")
         trust_builder_text = st.text_area("Type/paste Trust Builder®")
         selected_save_bucket = st.selectbox("Choose Trust Bucket to Save", trust_buckets, key="save_bucket")
         
         if st.button("Save"):
-            if trust_builder_text.strip() and selected_save_bucket:
-                save_trustbuilder(st.session_state["wix_user_id"], trust_builder_text.strip(), selected_save_bucket)
+            if trust_builder_text.strip() and selected_save_bucket and brand_save.strip():
+                save_trustbuilder(st.session_state["wix_user_id"], trust_builder_text.strip(), selected_save_bucket, brand_save.strip())
             else:
                 st.warning("Please fill all fields before saving.")