Update app.py

app.py

@@ -1722,115 +1722,117 @@ user_name = extract_name(st.session_state["email"])
 
 prompt = st.chat_input("")
 global combined_text
+def handle_prompt(prompt):
+    if st.session_state["used_messages"] < st.session_state["message_limit"]:
+        if prompt:
+            st.session_state.chat_started = True
+
+            # Prevent duplicate messages in chat history
+            if not any(msg["content"] == prompt for msg in st.session_state["chat_history"]):
+                st.session_state.chat_history.append({"role": "user", "content": prompt})
+
+            # Introduce a flag to track if a specific flow is handled
+            st.session_state["handled"] = False
+
+            # Handle missing Trust Bucket if needed
+            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"]
+
+                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
 
-if st.session_state["used_messages"] < st.session_state["message_limit"]:
-    if prompt:
-        st.session_state.chat_started = True
-        # Prevent duplicate messages in chat history
-        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 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.get("wix_user_id", "default_user"))
+                if trustbuilders:
+                    saved_content = "\n".join([f"- {entry}" for entry in trustbuilders])
+                    with st.chat_message("assistant"):
+                        st.markdown(f"Here are your saved TrustBuilders:\n{saved_content}")
+                else:
+                    with st.chat_message("assistant"):
+                        st.markdown("You haven't saved any TrustBuilders yet.")
+                st.session_state["handled"] = True
 
-        # Introduce a flag to track if a specific flow is handled
-        st.session_state["handled"] = False
+            # 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
 
-        # Handle missing Trust Bucket if needed
-        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"]
+                    # 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 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.get("wix_user_id", "default_user"))
-            if trustbuilders:
-                saved_content = "\n".join([f"- {entry}" for entry in trustbuilders])
-                with st.chat_message("assistant"):
-                    st.markdown(f"Here are your saved TrustBuilders:\n{saved_content}")
-            else:
-                with st.chat_message("assistant"):
-                    st.markdown("You haven't saved any TrustBuilders yet.")
-            st.session_state["handled"] = True
-
-        # Handle save TrustBuilder command
-        if  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:
+                        # 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.")
+                    
+                    # Mark as handled and exit to prevent further processing
+                    st.session_state["handled"] = True
+                    return  # Exit here to avoid triggering normal AI response
+
+            # Handle other memory queries if any
+            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:
+                    with st.chat_message("assistant"):
+                        st.markdown(memory_response)
+                    st.session_state["handled"] = True
+
+            # If not handled yet, proceed to the existing AI response generation
+            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
+                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"]
+
+                    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("Please provide the content to save as a TrustBuilder.")
-                st.session_state["handled"] = True
-                return
+                            st.markdown(combined_text, unsafe_allow_html=True)
 
-        
-        # Handle other memory queries if any
-        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:
-                with st.chat_message("assistant"):
-                    st.markdown(memory_response)
-                st.session_state["handled"] = True
+                    update_message_usage(st.session_state.get("wix_user_id", "default_user"))
 
-        # If not handled yet, proceed to your existing AI response generation
-        if not st.session_state["handled"]:
-            #  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()
-            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"):
-                        st.markdown(combined_text, unsafe_allow_html=True)
+                except Exception as e:
+                    logging.error(f"Error generating response: {e}")
+                    st.error("An error occurred while generating the response. Please try again.")
 
-                update_message_usage(st.session_state.get("wix_user_id", "default_user"))
+                st.session_state.chat_history.append({"role": "assistant", "content": cleaned_text})
+                copy_to_clipboard(cleaned_text)
+                st.session_state["handled"] = True  # Mark as handled
 
-            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
+# Call the function to handle the prompt
+handle_prompt(prompt)