Update app.py

app.py

@@ -672,6 +672,8 @@ def side():
 
 
         st.sidebar.header("TrustVault®")
+        st.sidebar.subheader("Saved Documents")
+        
         st.sidebar.markdown("""
             <style>
             .scrollable-container {
@@ -686,65 +688,76 @@ def side():
             .button-container {
                 display: flex;
                 justify-content: space-between;
-                gap: 10px;
             }
             </style>
         """, unsafe_allow_html=True)
         
-        # Sidebar: Saved Documents Section
-        st.sidebar.subheader("Saved Documents")
+        # Scrollable container for 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():
-                    doc_name = doc_data.get('content', 'Document')
+                    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)
         
-        st.sidebar.markdown('</div>', unsafe_allow_html=True)
-        
-        # Below saved documents: Upload and Delete buttons
+        # Button container for Upload and Delete actions
         st.sidebar.markdown('<div class="button-container">', unsafe_allow_html=True)
         
-        # File uploader
+        # File uploader button
         uploaded_file = st.sidebar.file_uploader("", key="file_uploader", label_visibility="collapsed")
-        
-        # Upload button
         if st.sidebar.button("Upload", key="upload_button"):
             if uploaded_file:
-                upload_to_firebase(st.session_state["wix_user_id"], uploaded_file)
-                st.sidebar.success(f"Document '{uploaded_file.name}' uploaded successfully.")
+                try:
+                    upload_to_firebase(st.session_state["wix_user_id"], uploaded_file)
+                    st.sidebar.success(f"Document '{uploaded_file.name}' uploaded successfully!")
+                except Exception as e:
+                    st.sidebar.error(f"Error uploading document: {e}")
             else:
                 st.sidebar.warning("Please select a file to upload.")
         
-        # Delete Selected Button
+        # Delete selected documents
         if docs:
-            selected_docs = st.sidebar.multiselect("Select documents to delete", options=doc_names)
+            delete_options = [doc_data.get("content", "Document") for doc_id, doc_data in docs.items()]
+            selected_docs = st.sidebar.multiselect("Delete documents", options=delete_options, key="delete_docs")
             if st.sidebar.button("Delete Selected", key="delete_button"):
-                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)
+                try:
+                    for doc_name in selected_docs:
+                        for doc_id, doc_data in docs.items():
+                            if doc_data.get("content") == doc_name:
+                                delete_document(st.session_state["wix_user_id"], doc_id)
+                    st.sidebar.success("Selected documents deleted successfully!")
+                    st.experimental_rerun()
+                except Exception as e:
+                    st.sidebar.error(f"Error deleting documents: {e}")
+        st.sidebar.markdown("</div>", unsafe_allow_html=True)
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
 
         
         # Show Trust Builders Section