Update app.py

app.py

@@ -452,49 +452,34 @@ def load_user_content(user_id):
     except Exception as e:
          st.info("not loaded ")
         
-def save_content(user_id, content, content_type):
-    """
-    Save TrustBuilder or Brand Tonality to Firebase and update session state, with a clean confirmation.
-    """
+
+
+
+
+
+
+def save_content(user_id, content, trust_bucket, content_type="TrustBuilder"):
     try:
         entry_id = str(uuid.uuid4())
         firebase_key = "TrustBuilder" if content_type.lower() == "trustbuilder" else "BrandTonality"
 
-        # Save to Firebase
-        db.child("users").child(user_id).child(firebase_key).child(entry_id).set({"message": content})
+        # Save with metadata (trust_bucket for TrustBuilders)
+        data = {"message": content}
+        if content_type == "TrustBuilder":
+            data["trust_bucket"] = trust_bucket
+
+        db.child("users").child(user_id).child(firebase_key).child(entry_id).set(data)
 
-        # Update `st.session_state` for immediate display in sidebar and chatbot
         if firebase_key not in st.session_state:
             st.session_state[firebase_key] = {}
-        st.session_state[firebase_key][entry_id] = {"message": content}
-
-        # Display save confirmation without showing any ID or debug information
-        display_save_confirmation(content_type.capitalize())
+        st.session_state[firebase_key][entry_id] = data
 
+        st.success(f"{content_type.capitalize()} saved successfully!")
     except Exception as e:
-        st.error(f"Could not save {content_type}. Please try again. Error: {e}")
+        st.error(f"Error saving {content_type}: {e}")
 
-        
-def display_saved_trustbuilders():
-    
-    # Display TrustBuilders with delete buttons
-    st.sidebar.subheader("Saved TrustBuilders")
-    trustbuilders = st.session_state.get("TrustBuilder", {})
-    for trustbuilder_id, trustbuilder_data in trustbuilders.items():
-        message = trustbuilder_data.get("message", "No message available")
-        st.sidebar.markdown(f"- {message[:50]}...")
-        if st.sidebar.button("Delete", key=f"del_tb_{trustbuilder_id}"):
-            delete_entry("TrustBuilder", trustbuilder_id, "TrustBuilder")
-
-    # Display Brand Tonality entries with delete buttons
-    st.sidebar.subheader("Saved Brand Tonality")
-    brand_tonality = st.session_state.get("BrandTonality", {})
-    for tonality_id, tonality_data in brand_tonality.items():
-        message = tonality_data.get("message", "No message available")
-        st.sidebar.markdown(f"- {message[:50]}...")
-        if st.sidebar.button("Delete", key=f"del_bt_{tonality_id}"):
-            delete_entry("BrandTonality", tonality_id, "BrandTonality")
 
+        
 
 def download_link(content, filename):
     """
@@ -519,7 +504,7 @@ def delete_document(user_id, doc_id):
     try:
         db.child("users").child(user_id).child("KnowledgeBase").child(doc_id).remove()
         st.success("Document deleted successfully!")
-        st.experimental_rerun()  # Refresh the list after deletion
+        st.rerun()  # Refresh the list after deletion
     except Exception as e:
         st.error(f"Error deleting document: {e}")
 
@@ -690,21 +675,7 @@ def side():
             if st.button("Delete"):
                 st.warning("Select a document to delete from the list below.")
         
-        # Display saved documents in a scrollable container
-        st.sidebar.markdown(
-            """
-            <style>
-            .scrollable-container {
-                height: 200px;
-                overflow-y: auto;
-                border: 1px solid #ddd;
-                padding: 10px;
-                background-color: #f5f5f5;
-            }
-            </style>
-            """,
-            unsafe_allow_html=True
-        )
+        
     
         docs = fetch_documents(st.session_state["wix_user_id"])
         st.sidebar.markdown('<div class="scrollable-container">', unsafe_allow_html=True)
@@ -724,16 +695,13 @@ def side():
     
         # Show My TrustBuilders Section
         st.sidebar.subheader("Show my TrustBuilders®")
-        search_query = st.sidebar.text_input("Search by keyword",key="keyword")
+        search_query = st.sidebar.text_input("Search by keyword", key="keyword")
         if search_query:
             search_results = {k: v for k, v in st.session_state.get("TrustBuilder", {}).items() if search_query.lower() in v["message"].lower()}
             for result_id, result_data in search_results.items():
                 st.sidebar.write(f"- {result_data['message'][:50]}")
     
-        st.sidebar.text_input("Brand/Product/Person",key="hko")
-        if st.sidebar.button("Show"):
-            st.write("Show clicked")
-    
+        # Trust Bucket Selection
         st.sidebar.write("and choose the Trust Bucket®")
         trust_buckets = ["S", "D", "R", "B", "V", "C"]
         bucket_labels = ["Stability", "Development", "Relationship", "Benefit", "Vision", "Competence"]
@@ -753,12 +721,13 @@ def side():
         for idx, (bucket, label) in enumerate(zip(trust_buckets, bucket_labels)):
             with st.sidebar:
                 if st.button(f"{bucket}", key=f"allocate_{label}"):
-                    save_content(st.session_state["wix_user_id"], trust_builder_text, label)
+                    save_content(st.session_state["wix_user_id"], trust_builder_text, trust_bucket=label)
     
         if st.sidebar.button("Let AI allocate"):
             st.write("AI allocated the Trust Bucket")
 
 
+
 side()
 # Load knowledge base
 
@@ -1426,23 +1395,12 @@ def handle_memory_queries(prompt):
     return None # If no recognized command, proceed with general handling
 
 
-def delete_entry(category, entry_id, session_key):
-    """
-    Deletes an entry from Firebase and updates the session state to reflect the deletion.
-    """
+def delete_entry(category, entry_id):
     try:
-        # Get user ID from session state
         user_id = st.session_state["wix_user_id"]
-        
-        # Remove the entry from Firebase
         db.child("users").child(user_id).child(category).child(entry_id).remove()
-
-        # Remove from session state and refresh sidebar display
-        if session_key in st.session_state and entry_id in st.session_state[session_key]:
-            del st.session_state[session_key][entry_id]
-
-        # Refresh the app to immediately reflect the change in the sidebar
-        st.rerun()
+        st.session_state[category].pop(entry_id, None)
+        st.success(f"{category} entry deleted successfully!")
     except Exception as e:
         st.error(f"Error deleting entry: {e}")
 
@@ -1455,7 +1413,7 @@ def download_trustbuilder_as_md(content, trustbuilder_id):
 
 def handle_save_trustbuilder_or_tonality(user_id, content, type_name):
     save_content(user_id, content, type_name)
-    display_saved_trustbuilders()
+