Update app.py

app.py

@@ -1747,8 +1747,7 @@ def handle_save_trustbuilder(content, specified_bucket=None):
     # Update last processed content
     st.session_state["last_processed_content"] = content
  # Confirm saving to the user
-    with st.chat_message("assistant"):
-        st.markdown(f"TrustBuilder allocated to **{bucket}** and saved successfully!")
+    return f"TrustBuilder allocated to **{bucket}** and saved successfully!"
 
 
 
@@ -1878,8 +1877,7 @@ def handle_prompt(prompt):
                         response = handle_save_trustbuilder(content_to_save, specified_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("Please provide the content to save as a TrustBuilder.")