Update app.py

app.py

@@ -180,18 +180,19 @@ def upload_to_firebase(user_id, file):
 
 def fetch_trustbuilders(user_id):
     """
-    Fetch all TrustBuilders as plain text from Firebase.
+    Fetch all TrustBuilders from Firebase and return as a list of dictionaries.
     """
     try:
-        trustbuilders = db.child("users").child(user_id).child("TrustBuilders").get()
-        if trustbuilders.each():
-            return [item.val() for item in trustbuilders.each()]
+        trustbuilders = db.child("users").child(user_id).child("TrustBuilders").get().val()
+        if isinstance(trustbuilders, dict):  # If it's a dictionary
+            return trustbuilders
+        elif isinstance(trustbuilders, list):  # If it's a list
+            return {str(idx): item for idx, item in enumerate(trustbuilders) if item}  # Convert list to dict
         else:
-            st.warning("No TrustBuilders found in Firebase.")
-            return []
+            return {}  # Return an empty dict if no data found
     except Exception as e:
         st.error(f"Error fetching TrustBuilders: {e}")
-        return []
+        return {}
 
 
 def delete_trustbuilder(user_id, trustbuilder_id):
@@ -788,13 +789,15 @@ def side():
                 trustbuilders = fetch_trustbuilders(st.session_state["wix_user_id"])
                 results = []
                 for trustbuilder_id, data in trustbuilders.items():
-                    content = data.get("content", "")
-                    bucket = data.get("bucket", "")
-                    tb_brand = data.get("brand", "")
-                    if ((selected_bucket == bucket or not selected_bucket) and
-                        (not search_query or search_query.lower() in content.lower()) and
-                        (not brand or brand.lower() in tb_brand.lower())):
-                        results.append(f"{bucket}: {content}")
+                    # Ensure data is a dictionary or handle accordingly
+                    if isinstance(data, dict):
+                        content = data.get("content", "")
+                        bucket = data.get("bucket", "")
+                        tb_brand = data.get("brand", "")
+                        if ((selected_bucket == bucket or not selected_bucket) and
+                            (not search_query or search_query.lower() in content.lower()) and
+                            (not brand or brand.lower() in tb_brand.lower())):
+                            results.append(f"{bucket}: {content}")
                 if results:
                     for result in results:
                         st.markdown(f"- {result}")