Update app.py

app.py

@@ -1746,9 +1746,10 @@ 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!")
 
-    # Generate response
-    return f"TrustBuilder allocated to **{bucket}** and saved successfully!"
 
 
 
@@ -1858,8 +1859,8 @@ def handle_prompt(prompt):
                 st.session_state.chat_history.append({"role": "assistant", "content": assistant_response})
                 with st.chat_message("assistant"):
                     st.markdown(assistant_response)
-            st.session_state["handled"] = True
-            return  # Exit to prevent further processing
+                st.session_state["handled"] = True
+                return  # Exit to prevent further processing
 
             # Handle save TrustBuilder command
             if not st.session_state["handled"]: