Update app.py

app.py

@@ -382,40 +382,34 @@ def load_user_data_source(user_id):
 
 
 
+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}
+    except Exception as e:
+        st.error(f"Error saving TrustBuilder: {e}")
 
+# Function to store Brand Tonality
 def store_brand_tonality(user_id, message):
     try:
         tonality_id = str(uuid.uuid4())
-        db.child("users").child(user_id).child("brandTonality").child(tonality_id).set({"message": message})
-        
-        # Update st.session_state for instant display in sidebar
+        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("users").child(user_id).child("trustBuilders").child(trustbuilder_id).set({"message": message})
-        
-        # Update st.session_state for instant display in sidebar
-        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):
-    """Load TrustBuilders and Brand Tonalities from Firebase."""
     try:
-        trustbuilders = db.child("users").child(user_id).child("trustBuilders").get()
-        brand_tonalities = db.child("users").child(user_id).child("brandTonality").get()
-        st.session_state["trustBuilders"] = [item.val()["content"] for item in trustbuilders.each()] if trustbuilders.each() else []
-        st.session_state["brand_tonality"] = [item.val()["content"] for item in brand_tonalities.each()] if brand_tonalities.each() else []
+        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", {})
+            st.session_state["brand_tonality"] = user_data.get("brandTonality", {})
     except Exception as e:
         st.error(f"Error loading saved content: {e}")
         
@@ -1202,7 +1196,11 @@ def download_trustbuilder_as_md(content, trustbuilder_id):
     st.sidebar.markdown(download_link, unsafe_allow_html=True)
 
 
-
+def display_save_confirmation(type_saved):
+    """
+    Display a confirmation message when content is saved.
+    """
+    st.info(f"Content successfully saved as **{type_saved}**!")