Update app.py

app.py

@@ -1721,13 +1721,20 @@ if st.session_state["used_messages"] < st.session_state["message_limit"]:
 
         # Handle missing Trust Bucket if needed
         if st.session_state.get("missing_trustbucket_content") and not st.session_state["handled"]:
-            # [Your existing missing bucket handling code]
-            # ...
+            bucket = prompt.strip().capitalize()
+            valid_buckets = ["Stability", "Development", "Relationship", "Benefit", "Vision", "Competence"]
+
+            if bucket in valid_buckets:
+                content_to_save = st.session_state.pop("missing_trustbucket_content")
+                handle_save_trustbuilder(content_to_save, bucket)
+            else:
+                with st.chat_message("assistant"):
+                    st.markdown("Invalid Trust Bucket. Please choose from Stability, Development, Relationship, Benefit, Vision, or Competence.")
             st.session_state["handled"] = True
 
         # Handle fetching saved TrustBuilders when user asks
         if ("find my saved trustbuilders" in prompt.lower() or "show my saved trustbuilders" in prompt.lower()) and not st.session_state["handled"]:
-            trustbuilders = fetch_trustbuilders(st.session_state["wix_user_id"])
+            trustbuilders = fetch_trustbuilders(st.session_state.get("wix_user_id", "default_user"))
             if trustbuilders:
                 saved_content = "\n".join([f"- {entry}" for entry in trustbuilders])
                 with st.chat_message("assistant"):
@@ -1737,10 +1744,37 @@ if st.session_state["used_messages"] < st.session_state["message_limit"]:
                     st.markdown("You haven't saved any TrustBuilders yet.")
             st.session_state["handled"] = True
 
+        # Handle save TrustBuilder command
+        if not st.session_state["handled"]:
+            save_match = re.search(r"\b(save|add|keep|store)\s+(this)?\s*(as)?\s*(\w+\s*trustbuilder|trustbuilder)\s*:?(.+)?", prompt, re.IGNORECASE)
+            if save_match:
+                content_to_save = save_match.group(5).strip() if save_match.group(5) else None
+                specified_bucket = None
+
+                # Check for explicit bucket mention in the same prompt
+                bucket_match = re.search(r"\b(stability|development|relationship|benefit|vision|competence)\b", prompt, re.IGNORECASE)
+                if bucket_match:
+                    specified_bucket = bucket_match.group(1).capitalize()
+
+                if content_to_save:
+                    handle_save_trustbuilder(content_to_save, specified_bucket)
+                else:
+                    # If content is not provided after the command, extract from prompt
+                    content_to_save = re.sub(r"\b(save|add|keep|store)\s+(this)?\s*(as)?\s*(\w+\s*trustbuilder|trustbuilder)\b", "", prompt, flags=re.IGNORECASE).strip()
+                    if content_to_save:
+                        handle_save_trustbuilder(content_to_save, specified_bucket)
+                    else:
+                        with st.chat_message("assistant"):
+                            st.markdown("Please provide the content to save as a TrustBuilder.")
+                st.session_state["handled"] = True
+
         # Handle other memory queries if any
         if not st.session_state["handled"]:
             memory_response = handle_memory_queries(prompt)
-            if memory_response:
+            if memory_response == "find_my_saved_trustbuilders":
+                # This case is already handled above, so we can set handled to True
+                st.session_state["handled"] = True
+            elif memory_response:
                 with st.chat_message("assistant"):
                     st.markdown(memory_response)
                 st.session_state["handled"] = True
@@ -1750,39 +1784,38 @@ if st.session_state["used_messages"] < st.session_state["message_limit"]:
             # Your existing code block (kept unchanged)
             # -----------------------------------------------------------
             
-                # Generate a response with AI for other types of queries
-                with st.chat_message("user"):
-                    st.markdown(prompt)
-                response_placeholder = st.empty()
+            # Generate a response with AI for other types of queries
+            with st.chat_message("user"):
+                st.markdown(prompt)
+            response_placeholder = st.empty()
+            with response_placeholder:
+                with st.chat_message("assistant"):
+                    add_dot_typing_animation()
+                    display_typing_indicator()
+            cleaned_text = ""
+            # Specialized responses if keywords detected
+            try:
+                output = agent_executor.invoke({
+                     "input": f"{prompt} Be in natural tone, doesn’t use flowery language and typical AI words. Sources should be accurate, latest, and relevant. Avoid AI jargon. Give short headings in conversational language with each paragraph in content of articles, blogs, write-ups, and newsletters, keeping the mentioned format only when asked.",
+                     "chat_history": st.session_state.chat_history
+                })
+                full_response = output["output"]
+                # full_response = replace_terms(full_response)
+    
+                cleaned_text = clean_text(full_response)
+                trust_tip, suggestion = get_trust_tip_and_suggestion()
+                combined_text = f"{cleaned_text}\n\n---\n\n**Trust Tip**: {trust_tip}\n\n**Suggestion**: {suggestion}"
                 with response_placeholder:
                     with st.chat_message("assistant"):
-                        add_dot_typing_animation()
-                        display_typing_indicator()
-                cleaned_text = ""
-                # Specialized responses if keywords detected
-                try:
-                    #if "trustbuilder" in prompt.lower() or "trust" in prompt.lower():
-                    output = agent_executor.invoke({
-                         "input": f"{prompt} Be in natural tone,doesn’t use flowery language and typical ai words. Sources should be accurate latest and relevant.Avoid AI jargons.Give short headings in conversational language with each paragraph in content of article,blogs,writeup and newsletter keeping mentioned format only when asked .",
-                         "chat_history": st.session_state.chat_history
-                    })
-                    full_response = output["output"]
-                    #full_response= replace_terms(full_response)
-        
-                    cleaned_text = clean_text(full_response)
-                    trust_tip, suggestion = get_trust_tip_and_suggestion()
-                    combined_text = f"{cleaned_text}\n\n---\n\n**Trust Tip**: {trust_tip}\n\n**Suggestion**: {suggestion}"
-                    with response_placeholder:
-                        with st.chat_message("assistant"):
-                            st.markdown(combined_text, unsafe_allow_html=True)
+                        st.markdown(combined_text, unsafe_allow_html=True)
 
-                    update_message_usage(st.session_state["wix_user_id"])
+                update_message_usage(st.session_state.get("wix_user_id", "default_user"))
 
-                except Exception as e:
-                    logger.error(f"Error generating response: {e}")
-                    st.error("An error occurred while generating the response. Please try again.")
-                
-                st.session_state.chat_history.append({"role": "assistant", "content": cleaned_text})
-                copy_to_clipboard(cleaned_text) 
+            except Exception as e:
+                logging.error(f"Error generating response: {e}")
+                st.error("An error occurred while generating the response. Please try again.")
+            
+            st.session_state.chat_history.append({"role": "assistant", "content": cleaned_text})
+            copy_to_clipboard(cleaned_text) 
             # -----------------------------------------------------------
-        st.session_state["handled"] = True  # Mark as handled
+            st.session_state["handled"] = True  # Mark as handled