Update app.py

app.py

@@ -1712,29 +1712,37 @@ welcome_placeholder.empty()
 # Input field for chatbot interaction
 prompt = st.chat_input("")
 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
+
+        # 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 save commands based on the user prompt
-        memory_response = handle_memory_queries(prompt)
-        if memory_response:
-            with st.chat_message("assistant"):
-                st.markdown(memory_response)
-        elif st.session_state.get("missing_trustbucket_content"):
-            bucket = prompt.strip()
+        # Check if a missing bucket needs to be handled
+        if st.session_state.get("missing_trustbucket_content"):
+            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.")
+
         else:
-                # Generate a response with AI for general queries
+            # Handle save commands or general queries
+            memory_response = handle_memory_queries(prompt)
+            if memory_response:
+                with st.chat_message("assistant"):
+                    st.markdown(memory_response)
+            else:
+                # Generate a response for general queries using AI
                 with st.chat_message("user"):
                     st.markdown(prompt)
                 response_placeholder = st.empty()
@@ -1743,15 +1751,17 @@ if st.session_state["used_messages"] < st.session_state["message_limit"]:
                         add_dot_typing_animation()
                         display_typing_indicator()
                 cleaned_text = ""
+
                 try:
                     output = agent_executor.invoke({
-                        "input": f"{prompt} Be in natural tone,doesn’t use flowery language and typical ai words .Use knowledge when user use 'our' or 'Volkswagen' pull only directly information from knowledgebase and display as it is if not found use google then provide relevant sources.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 .",
+                        "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}"
+
                     with response_placeholder:
                         with st.chat_message("assistant"):
                             st.markdown(combined_text, unsafe_allow_html=True)
@@ -1761,6 +1771,6 @@ if st.session_state["used_messages"] < st.session_state["message_limit"]:
                 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)
+                copy_to_clipboard(cleaned_text)