Update app.py

app.py

@@ -1835,13 +1835,18 @@ def handle_prompt(prompt):
 
                 if bucket in valid_buckets:
                     content_to_save = st.session_state.pop("missing_trustbucket_content")
-                    handle_save_trustbuilder(content_to_save, bucket)
+                    response = handle_save_trustbuilder(content_to_save, bucket)
+                    if response:
+                        st.session_state.chat_history.append({"role": "assistant", "content": response})
+                        with st.chat_message("assistant"):
+                            st.markdown(response)
                 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
+                return  # Exit to prevent further processing
 
-            # Handle fetching saved TrustBuilders when user asks
+            # Handle fetching saved TrustBuilders
             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.get("wix_user_id", "default_user"))
                 if trustbuilders:
@@ -1849,14 +1854,12 @@ def handle_prompt(prompt):
                     assistant_response = f"Here are your saved TrustBuilders:\n{saved_content}"
                 else:
                     assistant_response = "You haven't saved any TrustBuilders yet."
-            
-                # Append assistant's response to chat history
+
                 st.session_state.chat_history.append({"role": "assistant", "content": assistant_response})
-            
-                # Display the assistant's response
                 with st.chat_message("assistant"):
                     st.markdown(assistant_response)
                 st.session_state["handled"] = True
+                return  # Exit to prevent further processing
 
             # Handle save TrustBuilder command
             if not st.session_state["handled"]:
@@ -1865,7 +1868,7 @@ def handle_prompt(prompt):
                     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
+                    # Check for explicit bucket mention
                     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()
@@ -1876,42 +1879,33 @@ def handle_prompt(prompt):
                             st.session_state.chat_history.append({"role": "assistant", "content": response})
                             with st.chat_message("assistant"):
                                 st.markdown(response)
-                                st.session_state["handled"] = True
-                                return
-                      
-                        else:
-                            # Prompt user to provide content if missing
-                            with st.chat_message("assistant"):
-                                st.markdown("Please provide the content to save as a TrustBuilder.")
+                    else:
+                        with st.chat_message("assistant"):
+                            st.markdown("Please provide the content to save as a TrustBuilder.")
 
-                    # Mark as handled and exit to prevent further processing
                     st.session_state["handled"] = True
-                    return  # Exit here to avoid triggering normal AI response
+                    return  # Exit to prevent further processing
 
-            # Handle other memory queries if any
+            # Handle other memory queries
             if not st.session_state["handled"]:
                 memory_response = handle_memory_queries(prompt)
-                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:
+                if memory_response:
+                    st.session_state.chat_history.append({"role": "assistant", "content": memory_response})
                     with st.chat_message("assistant"):
                         st.markdown(memory_response)
                     st.session_state["handled"] = True
+                    return  # Exit to prevent further processing
 
-            # If not handled yet, proceed to the existing AI response generation
+            # If no specific handling, generate a general AI response
             if not st.session_state["handled"]:
-                # 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
+                        display_typing_indicator()                # Specialized responses if keywords detected
                 try:
                     output = agent_executor.invoke({
                         "input": f"{prompt} Avoid flowery language and AI-Jargons.Only when asked to write  an article, blog, newsletter, adhere to the following structure and ensure trust builders are fluidly interconnected for cohesive and logical flow. Use clear subheaders for each paragraph to structure the content effectively and Use explicit transitions to show how one section leads to or supports the next.Do-not mention trustbucket names literally and word 'trust' in the copy applications.*Use only knowledgebase when asked about top trust buckets and builders give exactly those*. ",