Update app.py

app.py

@@ -1701,10 +1701,7 @@ initialize_user_session()
 retrieve_user_data(st.session_state["wix_user_id"])  # Fetch and display saved data for the user
 
 user_name = extract_name(st.session_state["email"])
-welcome_placeholder = st.empty()
-welcome_placeholder.info(f"**Hello User!**")
-time.sleep(3)
-welcome_placeholder.empty()
+
 
 
 
@@ -1715,49 +1712,32 @@ global combined_text
 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 prevent further processing if a specific flow is handled
+        # Introduce a flag to track if a specific flow is handled
         st.session_state["handled"] = False
 
-        # Check if a missing Trust Bucket needs to be handled
+        # 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)
-                with st.chat_message("assistant"):
-                    st.markdown(f"Got it! I've saved this TrustBuilder under '{bucket}' Trust. Let me know if there's anything else you'd like to do!")
-            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.")
+            # [Your existing missing bucket handling code]
+            # ...
             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", {})
+        # 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"])
             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})
-
-                # Inform the user that saved content is being used
+                saved_content = "\n".join([f"- {entry}" for entry in trustbuilders])
                 with st.chat_message("assistant"):
-                    st.markdown("Using your saved TrustBuilders to generate a response...")
-                st.session_state["handled"] = True
+                    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. Please save some to use them in responses.")
-                st.session_state["handled"] = True
+                    st.markdown("You haven't saved any TrustBuilders yet.")
+            st.session_state["handled"] = True
 
-        # Handle save commands or memory queries
+        # Handle other memory queries if any
         if not st.session_state["handled"]:
             memory_response = handle_memory_queries(prompt)
             if memory_response:
@@ -1765,46 +1745,44 @@ if st.session_state["used_messages"] < st.session_state["message_limit"]:
                     st.markdown(memory_response)
                 st.session_state["handled"] = True
 
-        # Handle general AI queries
+        # If not handled yet, proceed to your existing AI response generation
         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"):
-                    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."
-
-                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}"
-
+            # Your existing code block (kept unchanged)
+            # -----------------------------------------------------------
+            else:
+                # 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"):
-                        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.")
+                        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)
 
-            # Update chat history with AI response
-            st.session_state.chat_history.append({"role": "assistant", "content": cleaned_text})
-            copy_to_clipboard(cleaned_text)
+                    update_message_usage(st.session_state["wix_user_id"])
 
-        # Reset the handled flag for the next prompt
-        st.session_state["handled"] = False
+                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) 
+            # -----------------------------------------------------------
+            st.session_state["handled"] = True  # Mark as handled