Update app.py

app.py

@@ -170,22 +170,23 @@ def fetch_trustbuilders(user_id, keyword=None, brand=None, bucket=None):
     try:
         st.write(f"Fetching TrustBuilders with keyword='{keyword}', brand='{brand}', bucket='{bucket}'")
         trustbuilders = db.child("users").child(user_id).child("TrustBuilder").get().val()
-        st.write("TrustBuilders retrieved from database:", trustbuilders)
+        st.write("TrustBuilders retrieved from Firebase:", trustbuilders)
+
         if not trustbuilders:
-            st.write("No TrustBuilders found in the database.")
+            st.write("No TrustBuilders found in Firebase.")
             return {}
 
         filtered_trustbuilders = {}
-        for k, v in trustbuilders.items():
-            message = v.get("message", "").lower()
-            brand_field = v.get("brand", "").lower()
-            bucket_field = v.get("bucket", "")
-            st.write(f"Checking TrustBuilder {k}:")
+        for key, value in trustbuilders.items():
+            message = value.get("message", "").lower()
+            brand_field = value.get("brand", "").lower()
+            bucket_field = value.get("bucket", "")
+
+            st.write(f"Checking TrustBuilder {key}:")
             st.write(f"  Message: {message}")
             st.write(f"  Brand: {brand_field}")
             st.write(f"  Bucket: {bucket_field}")
 
-            # Apply filters
             if keyword and keyword.lower() not in message:
                 st.write("  Keyword does not match.")
                 continue
@@ -195,8 +196,10 @@ def fetch_trustbuilders(user_id, keyword=None, brand=None, bucket=None):
             if bucket and bucket != "All" and bucket != bucket_field:
                 st.write(f"  Bucket '{bucket_field}' does not match selected '{bucket}'.")
                 continue
+
             st.write("  TrustBuilder matches filters.")
-            filtered_trustbuilders[k] = v
+            filtered_trustbuilders[key] = value
+
         st.write("Filtered TrustBuilders:", filtered_trustbuilders)
         return filtered_trustbuilders
     except Exception as e:
@@ -563,21 +566,21 @@ def load_user_content(user_id):
 
 
 
-
 def save_content(user_id, brand, trust_builder_text, trust_bucket=None):
     try:
         trustbuilder_id = str(uuid.uuid4())
         data = {
-            "brand": brand,
-            "message": trust_builder_text,
-            "bucket": trust_bucket
+            "brand": str(brand),
+            "message": str(trust_builder_text),
+            "bucket": str(trust_bucket)
         }
-        # Output the data being saved
         st.write("Saving TrustBuilder with data:", data)
         db.child("users").child(user_id).child("TrustBuilder").child(trustbuilder_id).set(data)
         st.success(f"TrustBuilder saved to {trust_bucket} bucket.")
     except Exception as e:
         st.error(f"Error saving TrustBuilder: {e}")
+
+        
 def ai_allocate_trust_bucket(trust_builder_text):
     # Implement your AI allocation logic here
     return "Stability"