Update app.py

app.py

@@ -185,7 +185,15 @@ def fetch_trustbuilders(user_id):
     try:
         trustbuilders = db.child("users").child(user_id).child("TrustBuilders").get().val()
         if trustbuilders:
-            return trustbuilders
+            # Ensure all entries are in the expected dictionary format
+            processed_trustbuilders = {}
+            for key, value in trustbuilders.items():
+                if isinstance(value, dict):
+                    processed_trustbuilders[key] = value
+                else:
+                    # Assuming value is just the content string and reconstructing the expected format
+                    processed_trustbuilders[key] = {"content": value}
+            return processed_trustbuilders
         else:
             st.warning("No TrustBuilders found.")
             return {}
@@ -193,7 +201,6 @@ def fetch_trustbuilders(user_id):
         st.error(f"Error fetching TrustBuilders: {e}")
         return {}
 
-
 def delete_trustbuilder(user_id, trustbuilder_id):
     try:
         db.child("users").child(user_id).child("TrustBuilder").child(trustbuilder_id).remove()
@@ -775,7 +782,7 @@ def side():
     
         if st.button("Show TrustBuilders", key="show_trustbuilders"):
             # Fetch TrustBuilders from Firebase
-            trustbuilders = fetch_trustbuilders(st.session_state.get("user_id", "demo_user"))
+            trustbuilders = fetch_trustbuilders(st.session_state.get("wix_user_id"))
             st.write(f"Debugging: Retrieved TrustBuilders - {trustbuilders}")  # Debug output to check what is being fetched
     
             results = []
@@ -803,7 +810,7 @@ def side():
         if st.button("Save TrustBuilder", key="save_trustbuilder"):
             if trust_builder_text.strip() and selected_save_bucket:
                 content_to_save = f"{selected_save_bucket}: Brand: {brand_save.strip()} | {trust_builder_text.strip()}"
-                save_content(st.session_state.get("wix_user_id", "demo_user"), content_to_save)  # Assume function signature has been corrected
+                save_content(st.session_state.get("wix_user_id"), content_to_save)  # Assume function signature has been corrected
                 st.success("TrustBuilder saved successfully!")
             else:
                 st.warning("Please fill all fields before saving.")