Update app.py

app.py

@@ -56,9 +56,9 @@ INITIAL_MESSAGE_LIMIT = 100
 
 
 if "wix_user_id" not in st.session_state:
-    st.session_state["wix_user_id"] = str(uuid.uuid4())  # Assign a unique ID for each session
+    st.session_state["wix_user_id"] = str(uuid.uuid4())  # Assign unique user ID for the session
 if "email" not in st.session_state:
-    st.session_state["email"] = str(uuid.uuid4())  # Default email for demo
+    st.session_state["email"] = f"user_{uuid.uuid4()}@example.com"
 if "message_limit" not in st.session_state:
     st.session_state["message_limit"] = 1000
 if "used_messages" not in st.session_state:
@@ -180,22 +180,14 @@ def upload_to_firebase(user_id, file):
 
 def fetch_trustbuilders(user_id):
     """
-    Fetch all TrustBuilders for a user from Firebase.
+    Retrieve TrustBuilders from Firebase.
     """
     try:
         trustbuilders = db.child("users").child(user_id).child("TrustBuilders").get().val()
         if trustbuilders:
-            # Ensure all entries are in the expected dictionary format
-            processed_trustbuilders = {}
-            for key, value in trustbuilders.items():
-                if isinstance(value, dict):
-                    processed_trustbuilders[key] = value
-                else:
-                    # Assuming value is just the content string and reconstructing the expected format
-                    processed_trustbuilders[key] = {"content": value}
-            return processed_trustbuilders
+            return trustbuilders
         else:
-            st.warning("No TrustBuilders found.")
+            st.info("No TrustBuilders found.")
             return {}
     except Exception as e:
         st.error(f"Error fetching TrustBuilders: {e}")
@@ -497,20 +489,18 @@ def load_user_content(user_id):
     except Exception as e:
          st.info("not loaded ")
 
-
 def save_content(user_id, content):
     """
-    Save TrustBuilder in Firebase under the correct structure.
+    Save a TrustBuilder to Firebase.
     """
     try:
-        trustbuilder_id = str(uuid.uuid4())
+        trustbuilder_id = str(uuid.uuid4())  # Generate unique ID for each TrustBuilder
         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!")
+        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
     return "Stability"
@@ -774,47 +764,44 @@ def side():
 
         # Show Trust Builders Section
         trust_buckets = ["Stability", "Development", "Relationship", "Benefit", "Vision", "Competence"]
-
         st.subheader("Show My TrustBuilders®")
-        # Ensure unique keys for each text input by providing explicit keys
+        
+        # Ensure unique keys for widgets
         search_query = st.text_input("Search by keyword", key="search_query")
-        selected_bucket = st.selectbox("Select Trust Bucket", ["All"] + trust_buckets, key="selected_bucket")
-    
+        trust_buckets = ["All", "Stability", "Development", "Relationship", "Benefit", "Vision", "Competence"]
+        selected_bucket = st.selectbox("Select Trust Bucket", trust_buckets, key="selected_bucket")
+        
         if st.button("Show TrustBuilders", key="show_trustbuilders"):
-            # Fetch TrustBuilders from Firebase
-            trustbuilders = fetch_trustbuilders(st.session_state.get("wix_user_id"))
-            st.write(f"Debugging: Retrieved TrustBuilders - {trustbuilders}")  # Debug output to check what is being fetched
-    
+            user_id = st.session_state["wix_user_id"]
+            trustbuilders = fetch_trustbuilders(st.session_state["wix_user_id"])
             results = []
+        
             if trustbuilders:
                 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.")
+        
+            if results:
+                st.markdown("### Results:")
+                for result in results:
+                    st.markdown(f"- {result}")
             else:
-                st.write("No TrustBuilders found.")
-    
+                st.write("No TrustBuilders found matching the criteria.")
+        
         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®", key="trust_builder_text")
-        selected_save_bucket = st.selectbox("Choose Trust Bucket to Save", trust_buckets, key="save_bucket")
-    
+        selected_save_bucket = st.selectbox("Choose Trust Bucket to Save", trust_buckets[1:], 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"), content_to_save)  # Assume function signature has been corrected
-                st.success("TrustBuilder saved successfully!")
+                save_content(st.session_state["wix_user_id"], content_to_save)
             else:
                 st.warning("Please fill all fields before saving.")
-
+        
         
               
        
@@ -1515,26 +1502,20 @@ if "used_messages" not in st.session_state:
     st.session_state["used_messages"] = 0
 
 def initialize_user_session():
+    """
+    Initialize user data in Firebase and session state.
+    """
     try:
         user_id = st.session_state["wix_user_id"]
         email = st.session_state["email"]
         user_data = {
-            "user_name":user_id,
+            "user_name": user_id,
             "email": email,
             "message_limit": 1000,
             "used_messages": 0,
             "TrustBuilders": {}
-
         }
-
-        # Save user session to Firebase
-        db.child("users").child(user_id).set(user_data)
-
-        # Update session state
-        st.session_state["user_name"] = user_data["user_name"]
-        st.session_state["message_limit"] = user_data["message_limit"]
-        st.session_state["used_messages"] = user_data["used_messages"]
-
+        db.child("users").child(user_id).update(user_data)
         st.success("User session initialized successfully.")
     except Exception as e:
         st.error(f"Error initializing user session: {e}")