Update app.py

app.py

@@ -1620,40 +1620,29 @@ def retrieve_user_data(user_id):
     except Exception as e:
         st.error(f"Error loading saved content: {e}")
 
-
 def handle_memory_queries(prompt):
     prompt = prompt.lower().strip()
 
-    # Save as TrustBuilder
-    if re.search(r"\b(keep|add|save|store)\s*(this)?\s*(as)?\s*(trust\s*builder|trustbuilder)\b", prompt, re.IGNORECASE):
-        content_to_save = re.sub(r"\b(keep|add|save|store)\s*(this)?\s*(as)?\s*(trust\s*builder|trustbuilder)\b", "", prompt, flags=re.IGNORECASE).strip()
-        specified_bucket = None
-
-        # Check for explicit bucket mention in the same prompt
-        match = re.search(r"under\s+(\w+\s*trust)", prompt, re.IGNORECASE)
-        if match:
-            bucket_name = match.group(1).replace("trust", "").strip()
-            if bucket_name in ["Stability", "Development", "Relationship", "Benefit", "Vision", "Competence"]:
-                specified_bucket = bucket_name
-
+    # Save and allocate TrustBuilder
+    if "save and allocate" in prompt:
+        content_to_save = prompt.replace("save and allocate:", "").strip()
         if content_to_save:
-            handle_save_trustbuilder(content_to_save, specified_bucket)
+            handle_save_trustbuilder(content_to_save)
         else:
             with st.chat_message("assistant"):
-                st.markdown("Please provide content to save as TrustBuilder.")
+                st.markdown("Please provide content to save and allocate.")
         return None
 
-    # Retrieve saved TrustBuilders
-    elif "show my saved trustbuilders" in prompt or "show my saved trustbuilders" in prompt:
-        trustbuilders = st.session_state.get("TrustBuilder", {})
+    # Show saved TrustBuilders
+    elif "show my saved trustbuilders" in prompt or "find my saved trustbuilders" in prompt:
+        trustbuilders = fetch_trustbuilders(st.session_state.get("wix_user_id", "default_user"))
         if trustbuilders:
-            response = "\n".join([f"- {entry['message']}" for entry in trustbuilders.values()])
-            return f"Here are your saved TrustBuilders:\n{response}"
+            saved_content = "\n".join([f"- {entry['message']}" for entry in trustbuilders.values()])
+            return f"Here are your saved TrustBuilders:\n{saved_content}"
         else:
             return "You haven't saved any TrustBuilders yet."
 
-    return None 
-
+    return None
     
 def delete_entry(category, entry_id):
     try:
@@ -1673,24 +1662,32 @@ def download_trustbuilder_as_md(content, trustbuilder_id):
 
 def handle_save_trustbuilder(content, specified_bucket=None):
     """
-    Handles saving TrustBuilders by detecting or prompting for the Trust Bucket.
+    Handles saving TrustBuilders by detecting or automatically allocating the Trust Bucket.
     """
-    trust_buckets = ["Stability", "Development", "Relationship", "Benefit", "Vision", "Competence"]
+    trust_buckets = {
+        "Stability": ["secure", "reliable", "consistent", "stable"],
+        "Development": ["growth", "future", "plan", "innovate", "strategy"],
+        "Relationship": ["trust", "partnership", "collaborate", "customer"],
+        "Benefit": ["value", "profit", "advantage", "benefit"],
+        "Vision": ["future", "goal", "mission", "aspire", "dream"],
+        "Competence": ["skill", "expertise", "knowledge", "competent", "ability"]
+    }
+
     bucket = specified_bucket
 
-    # Detect bucket if not explicitly provided
+    # Automatically allocate bucket if not provided
     if not bucket:
-        for tb in trust_buckets:
-            if tb.lower() in content.lower():
+        for tb, keywords in trust_buckets.items():
+            if any(keyword in content.lower() for keyword in keywords):
                 bucket = tb
                 break
 
+    # If still no bucket found, prompt the user
     if not bucket:
-        # Store content and prompt user for bucket
         st.session_state["missing_trustbucket_content"] = content
         with st.chat_message("assistant"):
             st.markdown(
-                "No Trust Bucket specified. Please indicate the Trust Bucket (e.g., Stability, Development, Relationship, Benefit, Vision, Competence)."
+                "No Trust Bucket could be allocated automatically. Please indicate the Trust Bucket (e.g., Stability, Development, Relationship, Benefit, Vision, Competence)."
             )
         return
 
@@ -1698,7 +1695,12 @@ def handle_save_trustbuilder(content, specified_bucket=None):
     brand = st.session_state.get("brand_input_save", "Unknown")
     content_to_save = f"{bucket}: Brand: {brand.strip()} | {content.strip()}"
     save_content(st.session_state["wix_user_id"], content_to_save)
-  
+
+    # Confirm saving to the user
+    with st.chat_message("assistant"):
+        st.markdown(f"TrustBuilder allocated to **{bucket}** and saved successfully!")
+
+
 
                         
 # Function to update the message counter in a static location