Update app.py

app.py

@@ -168,17 +168,18 @@ def upload_to_firebase(user_id, content):
 
 def fetch_trustbuilders(user_id):
     """
-    Fetch TrustBuilders for a specific user from Firebase.
+    Fetch all TrustBuilders as plain text from Firebase.
     """
     try:
         trustbuilders = db.child("users").child(user_id).child("TrustBuilders").get()
         if trustbuilders.each():
-            return {item.key(): item.val() for item in trustbuilders.each()}
+            return [item.val() for item in trustbuilders.each()]
         else:
-            return {}
+            st.write("No TrustBuilders found in Firebase.")
+            return []
     except Exception as e:
         st.error(f"Error fetching TrustBuilders: {e}")
-        return {}
+        return []
 
 
 def delete_trustbuilder(user_id, trustbuilder_id):
@@ -499,15 +500,15 @@ def load_user_content(user_id):
 
 def save_content(user_id, content, bucket):
     """
-    Save TrustBuilder to Firebase under a specified bucket.
+    Save TrustBuilder as plain text in Firebase.
+    The bucket name is prefixed to the content for easy searching.
     """
     try:
-        entry_id = str(uuid.uuid4())
-        full_message = f"[{bucket}] {content}"  # Append bucket name to the content
-        db.child("users").child(user_id).child("TrustBuilders").child(entry_id).set({"message": full_message})
-        st.success(f"TrustBuilder saved under '{bucket}'!")
+        trustbuilder_text = f"{bucket}: {content}"
+        db.child("users").child(user_id).child("TrustBuilders").push(trustbuilder_text)
+        st.success("TrustBuilder saved successfully!")
     except Exception as e:
-        st.error(f"Could not save TrustBuilder. Error: {e}")
+        st.error(f"Error saving TrustBuilder: {e}")
 
         
 def ai_allocate_trust_bucket(trust_builder_text):
@@ -752,7 +753,7 @@ def side():
         if st.button("Save TrustBuilder"):
             if bucket_to_save and trust_builder_message.strip():
                 save_content(
-                    user_id=st.session_state.get("wix_user_id"),  # Replace "demo_user" with actual user ID logic
+                    user_id=st.session_state["wix_user_id"],  # Replace "demo_user" with actual user ID logic
                     content=trust_builder_message.strip(),
                     bucket=bucket_to_save
                 )
@@ -765,7 +766,7 @@ def side():
         selected_bucket = st.selectbox("Select Trust Bucket", ["All"] + trust_buckets)
         
         if st.button("Show TrustBuilders"):
-            trustbuilders = fetch_trustbuilders(st.session_state.get("wix_user_id", "demo_user"))  # Replace "demo_user" with actual user ID logic
+            trustbuilders = fetch_trustbuilders(st.session_state["wix_user_id"])  # Replace "demo_user" with actual user ID logic
             st.write(f"Filtering TrustBuilders for keyword='{search_query}' and bucket='{selected_bucket}'")
         
             if trustbuilders:
@@ -1484,7 +1485,7 @@ def initialize_user_session():
         user_id = st.session_state["wix_user_id"]
         email = st.session_state["email"]
         user_data = {
-            "user_name": "Demo",
+            "user_name":user_id,
             "email": email,
             "message_limit": 1000,
             "used_messages": 0,