Update app.py

app.py

@@ -1633,8 +1633,30 @@ def download_trustbuilder_as_md(content, trustbuilder_id):
     download_link = f'<a href="data:text/markdown;base64,{b64_content}" download="TrustBuilder_{trustbuilder_id}.md">Download</a>'
     st.sidebar.markdown(download_link, unsafe_allow_html=True)
 
-def handle_save_trustbuilder_or_tonality(user_id, content, type_name):
-    save_content(user_id, content)
+def handle_save_trustbuilder(content, specified_bucket=None):
+    """
+    Handles saving TrustBuilders by detecting or prompting for the Trust Bucket.
+    """
+    trust_buckets = ["Stability", "Development", "Relationship", "Benefit", "Vision", "Competence"]
+    bucket = specified_bucket
+
+    # Detect bucket if not explicitly provided
+    if not bucket:
+        for tb in trust_buckets:
+            if tb.lower() in content.lower():
+                bucket = tb
+                break
+
+    if not bucket:
+        st.warning("No Trust Bucket specified. Please indicate the Trust Bucket.")
+        st.session_state["missing_trustbucket_content"] = content
+        return
+
+    # Save TrustBuilder with detected/provided bucket
+    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)
+    st.success(f"TrustBuilder saved under '{bucket}' Trust.")
 
 
 
@@ -1708,6 +1730,23 @@ if st.session_state["used_messages"] < st.session_state["message_limit"]:
     if prompt:
         st.session_state.chat_started = True
         st.session_state.chat_history.append({"role": "user", "content": prompt})
+        if "save this as trustbuilder" in prompt.lower():
+            content_to_save = prompt.replace("save this as trustbuilder", "").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
+
+            if content_to_save:
+                handle_save_trustbuilder(content_to_save, specified_bucket)
+            else:
+                st.warning("Please provide content to save as TrustBuilder.")
+
+
 
         # Handle save commands based on the user prompt
         memory_response = handle_memory_queries(prompt)