Update app.py

app.py

@@ -1752,6 +1752,8 @@ def handle_save_trustbuilder(content, specified_bucket=None):
 
     # Update last processed content
     st.session_state["last_processed_content"] = content
+    return f"TrustBuilder allocated to **{bucket}** and saved successfully!"
+
 
     # Generate response
 
@@ -1873,8 +1875,7 @@ def handle_prompt(prompt):
                         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: