Update app.py

app.py

@@ -180,18 +180,18 @@ def upload_to_firebase(user_id, file):
 
 def fetch_trustbuilders(user_id):
     """
-    Fetch all TrustBuilders as plain text from Firebase.
+    Fetch all TrustBuilders for a user from Firebase.
     """
     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 trustbuilders:
+            return trustbuilders
         else:
-            st.warning("No TrustBuilders found in Firebase.")
-            return []
+            st.warning("No TrustBuilders found.")
+            return {}
     except Exception as e:
         st.error(f"Error fetching TrustBuilders: {e}")
-        return []
+        return {}
 
 
 def delete_trustbuilder(user_id, trustbuilder_id):
@@ -491,19 +491,19 @@ def load_user_content(user_id):
          st.info("not loaded ")
 
 
-def save_content(user_id, content, bucket):
+def save_content(user_id, content):
     """
-    Save TrustBuilder as plain text in Firebase.
+    Save TrustBuilder in Firebase under the correct structure.
     """
     try:
-        trustbuilder_text = f"{bucket}: {content}"
-        db.child("users").child(user_id).child("TrustBuilders").push(trustbuilder_text)
-        st.success("TrustBuilder saved successfully!")
+        trustbuilder_id = str(uuid.uuid4())
+        trustbuilder_data = {"content": content}
+        db.child("users").child(user_id).child("TrustBuilders").child(trustbuilder_id).set(trustbuilder_data)
+        st.success(f"TrustBuilder '{content}' saved successfully!")
     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"
@@ -769,52 +769,44 @@ def side():
         trust_buckets = ["Stability", "Development", "Relationship", "Benefit", "Vision", "Competence"]
 
         st.subheader("Show My TrustBuilders®")
-        search_query = st.text_input("Search by keyword")
-        selected_bucket = st.selectbox("Select Trust Bucket", ["All"] + trust_buckets)
-        
-        if st.button("Show TrustBuilders"):
-            trustbuilders = fetch_trustbuilders(st.session_state.get("user_id", "demo_user"))  # Replace with actual user ID
-            st.write(f"Filtering TrustBuilders for keyword='{search_query}' and bucket='{selected_bucket}'")
-        
+        # Ensure unique keys for each text input by providing explicit keys
+        search_query = st.text_input("Search by keyword", key="search_query")
+        selected_bucket = st.selectbox("Select Trust Bucket", ["All"] + trust_buckets, key="selected_bucket")
+    
+        if st.button("Show TrustBuilders", key="show_trustbuilders"):
+            # Fetch TrustBuilders from Firebase
+            trustbuilders = fetch_trustbuilders(st.session_state.get("user_id", "demo_user"))
+            st.write(f"Debugging: Retrieved TrustBuilders - {trustbuilders}")  # Debug output to check what is being fetched
+    
+            results = []
             if trustbuilders:
-                filtered = [
-                    tb for tb in trustbuilders
-                    if (selected_bucket == "All" or tb.startswith(selected_bucket)) and
-                       (not search_query or search_query.lower() in tb.lower())
-                ]
-                if filtered:
-                    for tb in filtered:
-                        st.write(tb)
+                for tb_id, tb_data in trustbuilders.items():
+                    content = tb_data.get("content", "")
+                    bucket, text = content.split(": ", 1) if ": " in content else ("", content)
+                    if (selected_bucket == "All" or bucket == selected_bucket) and (not search_query or search_query.lower() in text.lower()):
+                        results.append(content)
+    
+                if results:
+                    st.write("### Results:")
+                    for result in results:
+                        st.write(f"- {result}")
                 else:
                     st.write("No TrustBuilders found matching the criteria.")
             else:
                 st.write("No TrustBuilders found.")
-
-        
-        # Save TrustBuilders Section
+    
         st.subheader("Save TrustBuilders®")
-
-        st.subheader("Show My TrustBuilders®")
-        search_query = st.text_input("Search by keyword")
-        selected_bucket = st.selectbox("Select Trust Bucket", ["All"] + trust_buckets)
-        
-        if st.button("Show TrustBuilders"):
-            trustbuilders = fetch_trustbuilders(st.session_state.get("user_id", "demo_user"))  # Replace with actual user ID
-            st.write(f"Filtering TrustBuilders for keyword='{search_query}' and bucket='{selected_bucket}'")
-        
-            if trustbuilders:
-                filtered = [
-                    tb for tb in trustbuilders
-                    if (selected_bucket == "All" or tb.startswith(selected_bucket)) and
-                       (not search_query or search_query.lower() in tb.lower())
-                ]
-                if filtered:
-                    for tb in filtered:
-                        st.write(tb)
-                else:
-                    st.write("No TrustBuilders found matching the criteria.")
+        brand_save = st.text_input("Brand/Product/Person", key="brand_input_save")
+        trust_builder_text = st.text_area("Type/paste Trust Builder®", key="trust_builder_text")
+        selected_save_bucket = st.selectbox("Choose Trust Bucket to Save", trust_buckets, key="save_bucket")
+    
+        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
+                st.success("TrustBuilder saved successfully!")
             else:
-                st.write("No TrustBuilders found.")
+                st.warning("Please fill all fields before saving.")