Update app.py

app.py

@@ -1720,8 +1720,11 @@ if st.session_state["used_messages"] < st.session_state["message_limit"]:
         if not any(msg["content"] == prompt for msg in st.session_state["chat_history"]):
             st.session_state.chat_history.append({"role": "user", "content": prompt})
 
-        # Handle missing Trust Bucket
-        if st.session_state.get("missing_trustbucket_content"):
+        # Introduce a flag to prevent further processing if a specific flow is handled
+        st.session_state["handled"] = False
+
+        # Check if a missing Trust Bucket needs to be handled
+        if st.session_state.get("missing_trustbucket_content") and not st.session_state["handled"]:
             bucket = prompt.strip().capitalize()
             valid_buckets = ["Stability", "Development", "Relationship", "Benefit", "Vision", "Competence"]
 
@@ -1733,47 +1736,75 @@ if st.session_state["used_messages"] < st.session_state["message_limit"]:
             else:
                 with st.chat_message("assistant"):
                     st.markdown("That doesn't seem to be a valid Trust Bucket. Please choose from Stability, Development, Relationship, Benefit, Vision, or Competence.")
-            # Exit the flow here to prevent triggering additional AI responses
-
-        # Handle TrustBuilder or Brand Tonality save commands
-        memory_response = handle_memory_queries(prompt)
-        if memory_response:
-            with st.chat_message("assistant"):
-                st.markdown(memory_response)
-            # Exit after handling the specific save command
-            
+            st.session_state["handled"] = True
+
+        # Handle saved data usage
+        if "use my saved" in prompt.lower() and not st.session_state["handled"]:
+            trustbuilders = st.session_state.get("TrustBuilder", {})
+            if trustbuilders:
+                # Retrieve all saved TrustBuilders
+                saved_content = "\n".join([entry["message"] for entry in trustbuilders.values()])
+                
+                # Append saved content to the chat history for contextual AI responses
+                st.session_state.chat_history.append({"role": "assistant", "content": saved_content})
 
-        # General AI query handling
-        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 = ""
-
-        try:
-            # Generate AI response for general queries
-            output = agent_executor.invoke({
-                "input": f"{prompt} Be in natural tone, avoid flowery language or typical AI words.Use knowledge directly from our base; if unavailable, use Google with sources.Avoid AI jargon and provide concise, conversational headings for each section.",
-                "chat_history": st.session_state.chat_history
-            })
-            full_response = output["output"]
-            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}"
+                # Inform the user that saved content is being used
+                with st.chat_message("assistant"):
+                    st.markdown("Using your saved TrustBuilders to generate a response...")
+                st.session_state["handled"] = True
+            else:
+                with st.chat_message("assistant"):
+                    st.markdown("You haven't saved any TrustBuilders yet. Please save some to use them in responses.")
+                st.session_state["handled"] = True
 
+        # Handle save commands or memory queries
+        if not st.session_state["handled"]:
+            memory_response = handle_memory_queries(prompt)
+            if memory_response:
+                with st.chat_message("assistant"):
+                    st.markdown(memory_response)
+                st.session_state["handled"] = True
+
+        # Handle general AI queries
+        if not st.session_state["handled"]:
+            with st.chat_message("user"):
+                st.markdown(prompt)
+            response_placeholder = st.empty()
             with response_placeholder:
                 with st.chat_message("assistant"):
-                    st.markdown(combined_text, unsafe_allow_html=True)
+                    add_dot_typing_animation()
+                    display_typing_indicator()
+
+            try:
+                # Include saved TrustBuilders in the prompt context
+                trustbuilders = st.session_state.get("TrustBuilder", {})
+                saved_content = "\n".join([entry["message"] for entry in trustbuilders.values()]) if trustbuilders else ""
+                
+                # Generate AI response with saved content as context
+                ai_prompt = f"{saved_content}\n\nUser Input: {prompt}\nBe in natural tone, avoid flowery language or typical AI words. Use knowledge directly from our base; if unavailable, use Google with sources. Avoid AI jargon and provide concise, conversational headings for each section."
 
-            update_message_usage(st.session_state["wix_user_id"])
+                output = agent_executor.invoke({
+                    "input": ai_prompt,
+                    "chat_history": st.session_state.chat_history
+                })
+                full_response = output["output"]
+                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)
+
+                update_message_usage(st.session_state["wix_user_id"])
+
+            except Exception as e:
+                logger.error(f"Error generating response: {e}")
+                st.error("An error occurred while generating the response. Please try again.")
 
-        except Exception as e:
-            logger.error(f"Error generating response: {e}")
-            st.error("An error occurred while generating the response. Please try again.")
+            # Update chat history with AI response
+            st.session_state.chat_history.append({"role": "assistant", "content": cleaned_text})
+            copy_to_clipboard(cleaned_text)
 
-        # Update chat history with AI response
-        st.session_state.chat_history.append({"role": "assistant", "content": cleaned_text})
-        copy_to_clipboard(cleaned_text)
+        # Reset the handled flag for the next prompt
+        st.session_state["handled"] = False