Update app.py

app.py

@@ -1677,10 +1677,7 @@ def handle_save_trustbuilder(content, specified_bucket=None):
     """
     Handles saving TrustBuilders by detecting or automatically allocating the Trust Bucket.
     """
-    # Avoid reprocessing the same content
-    if "last_processed_content" in st.session_state and st.session_state["last_processed_content"] == content:
-        return None  # Exit if the content was already processed
-
+  
     trust_buckets = {
         "Stability": [
             "track record", "longevity", "size", "stability", "experience", "established", "heritage",
@@ -1746,13 +1743,11 @@ def handle_save_trustbuilder(content, specified_bucket=None):
     save_content(st.session_state["wix_user_id"], content_to_save)
 
     # Update last processed content
-    st.session_state["last_processed_content"] = content
     return f"TrustBuilder allocated to **{bucket}** and saved successfully!"
 
 
     # Generate response
 
-
 # Function to update the message counter in a static location
 
 if "email" not in st.session_state:
@@ -1870,15 +1865,16 @@ def handle_prompt(prompt):
                         specified_bucket = bucket_match.group(1).capitalize()
 
                     if content_to_save:
-                    
-                        # 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:
-                            handle_save_trustbuilder(content_to_save, specified_bucket)
-                        else:
+                        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("Please provide the content to save as a TrustBuilder.")
-                    
+                                st.markdown(response)
+                    else:
+                        # Prompt user to provide content if missing
+                        with st.chat_message("assistant"):
+                            st.markdown("Please provide the content to save as a TrustBuilder.")
+
                     # Mark as handled and exit to prevent further processing
                     st.session_state["handled"] = True
                     return  # Exit here to avoid triggering normal AI response