Update app.py

app.py

@@ -180,16 +180,11 @@ def upload_to_firebase(user_id, file):
 
 def fetch_trustbuilders(user_id):
     """
-    Fetch all TrustBuilders from Firebase and return as a list of dictionaries.
+    Fetch all TrustBuilders from Firebase.
     """
     try:
         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:
-            return {}  # Return an empty dict if no data found
+        return trustbuilders if isinstance(trustbuilders, dict) else {}
     except Exception as e:
         st.error(f"Error fetching TrustBuilders: {e}")
         return {}
@@ -495,18 +490,19 @@ def load_user_content(user_id):
 
 def save_content(user_id, content, bucket):
     """
-    Save TrustBuilder as plain text in Firebase.
+    Save TrustBuilder as normal text in Firebase with the associated bucket.
     """
     try:
-        trustbuilder_text = f"{bucket}: {content}"
+        trustbuilder_text = {
+            "content": content,
+            "bucket": bucket
+        }
         db.child("users").child(user_id).child("TrustBuilders").push(trustbuilder_text)
         st.success("TrustBuilder 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
@@ -772,33 +768,24 @@ def side():
         # Show Trust Builders Section
         st.subheader("Show My TrustBuilders®")
         
-        # Arrange text inputs and Show button
-        col1, col2, col3 = st.columns([1, 1, 0.5])
-        
         search_query = st.text_input("Search by keyword")
-        st.write("or")
         brand = st.text_input("Brand/Product/Person", key="brand_input_show")
-        
-        # Trust Buckets selectbox (define it before referencing)
         trust_buckets = ["Stability", "Development", "Relationship", "Benefit", "Vision", "Competence"]
-        selected_bucket = st.selectbox("Select Trust Bucket", trust_buckets, key="selected_bucket")
+        selected_bucket = st.selectbox("Select Trust Bucket", trust_buckets, key="selected_bucket_show")
         
         if st.button("Show"):
-            st.write(f"Filtering TrustBuilders for keyword='{search_query}', brand='{brand}', and bucket='{selected_bucket}'")
             try:
                 trustbuilders = fetch_trustbuilders(st.session_state["wix_user_id"])
                 results = []
                 for trustbuilder_id, data in trustbuilders.items():
-                    # 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}")
+                    content = data.get("content", "")
+                    bucket = data.get("bucket", "")
+                    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 content.lower())):
+                        results.append(f"{bucket}: {content}")
                 if results:
+                    st.markdown("### Results:")
                     for result in results:
                         st.markdown(f"- {result}")
                 else:
@@ -806,23 +793,41 @@ def side():
             except Exception as e:
                 st.error(f"Error fetching TrustBuilders: {e}")
         
-        # Save Trust Builders Section
+        # Save TrustBuilders Section
         st.subheader("Save TrustBuilders®")
+        
         brand_save = st.text_input("Brand/Product/Person", key="brand_input_save")
         trust_builder_text = st.text_area("Type/paste Trust Builder®")
         selected_save_bucket = st.selectbox("Choose Trust Bucket to Save", trust_buckets, key="save_bucket")
         
         if st.button("Save"):
             if trust_builder_text.strip() and selected_save_bucket:
-                # Concatenate brand and trust builder text into the content to match the save_content function
-                content_to_save = f"Brand: {brand_save.strip()} | {trust_builder_text.strip()}"
-                save_content(
-                    st.session_state["wix_user_id"],
-                    content_to_save,
-                    selected_save_bucket
-                )
+                save_content(st.session_state["wix_user_id"], trust_builder_text.strip(), selected_save_bucket)
             else:
                 st.warning("Please fill all fields before saving.")
+        
+        
+        
+        
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+