Update app.py

app.py

@@ -382,31 +382,35 @@ def load_user_data_source(user_id):
 
 
 
-def store_trustbuilder(user_id, message):
-    try:
-        trustbuilder_id = str(uuid.uuid4())
-        db.child(f"food_for_the_hungry/users/{user_id}/trustBuilders").child(trustbuilder_id).set({"message": message})
-        if "trustbuilders" not in st.session_state:
-            st.session_state["trustbuilders"] = {}
-        st.session_state["trustbuilders"][trustbuilder_id] = {"message": message}
-    except Exception as e:
-        st.error(f"Error saving TrustBuilder: {e}")
-
-# Updated store_brand_tonality function
 def store_brand_tonality(user_id, message):
     try:
         tonality_id = str(uuid.uuid4())
-        db.child(f"food_for_the_hungry/users/{user_id}/brandTonality").child(tonality_id).set({"message": message})
+        db.child(f"users/{user_id}/brandTonality").child(tonality_id).set({"message": message})
         if "brand_tonality" not in st.session_state:
             st.session_state["brand_tonality"] = {}
         st.session_state["brand_tonality"][tonality_id] = {"message": message}
+        display_save_confirmation("Brand Tonality")
+
     except Exception as e:
         st.error(f"Error saving Brand Tonality: {e}")
 
 
+def store_trustbuilder(user_id, message):
+    try:
+        trustbuilder_id = str(uuid.uuid4())
+        db.child(f"users/{user_id}/trustBuilders").child(trustbuilder_id).set({"message": message})
+        if "trustbuilders" not in st.session_state:
+            st.session_state["trustbuilders"] = {}
+        st.session_state["trustbuilders"][trustbuilder_id] = {"message": message}
+        display_save_confirmation("TrustBuilder")
+
+    except Exception as e:
+        st.error(f"Error saving TrustBuilder: {e}")
+
+
 def load_user_content(user_id):
     try:
-        user_data = db.child(f"food_for_the_hungry/users/{user_id}").get().val()
+        user_data = db.child(f"users/{user_id}").get().val()
         if user_data:
             st.session_state.update(user_data)
             st.session_state["trustbuilders"] = user_data.get("trustBuilders", {})
@@ -414,14 +418,13 @@ def load_user_content(user_id):
     except Exception as e:
         st.error(f"Error loading saved content: {e}")
         
+        
 def save_content(user_id, content, content_type):
     try:
         entry_id = str(uuid.uuid4())
-        db.child("food_for_the_hungry").child("users").child(user_id).child(content_type).child(entry_id).set({"message": content})
+        db.child("users").child(user_id).child(content_type).child(entry_id).set({"message": content})
         
         # Update session state directly after saving to avoid duplicates
-        if content_type.lower() not in st.session_state:
-            st.session_state[content_type.lower()] = {}
         st.session_state[content_type.lower()][entry_id] = {"message": content}
         display_save_confirmation(content_type)
     except Exception as e: