Update app.py

app.py

@@ -169,12 +169,19 @@ def fetch_trustbuilders(user_id, keyword=None, brand=None, bucket=None):
     try:
         trustbuilders = db.child("users").child(user_id).child("TrustBuilder").get().val()
         if trustbuilders:
-            filtered_trustbuilders = {
-                k: v for k, v in trustbuilders.items()
-                if (not keyword or keyword.lower() in v.get("message", "").lower())
-                   and (not brand or brand.lower() in v.get("brand", "").lower())
-                   and (not bucket or bucket == v.get("bucket", ""))
-            }
+            filtered_trustbuilders = {}
+            for k, v in trustbuilders.items():
+                message = v.get("message", "").lower()
+                brand_field = v.get("brand", "").lower()
+                bucket_field = v.get("bucket", "")
+                
+                if keyword and keyword.lower() not in message:
+                    continue
+                if brand and brand.lower() not in brand_field:
+                    continue
+                if bucket and bucket != bucket_field:
+                    continue
+                filtered_trustbuilders[k] = v
             return filtered_trustbuilders
         return {}
     except Exception as e:
@@ -522,11 +529,12 @@ def save_content(user_id, brand, trust_builder_text, trust_bucket=None):
             "message": trust_builder_text,
             "bucket": 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"