Update app.py

app.py

@@ -1290,7 +1290,7 @@ prompt_template = ChatPromptTemplate.from_messages([
 ])
 
 # Create Langchain Agent
-llm = ChatOpenAI(model="gpt-4o", temperature=0.7)
+llm = ChatOpenAI(model="gpt-4o", temperature=0.6)
 llm_with_tools = llm.bind_tools(tools)
 
 # Define the agent pipeline
@@ -1606,20 +1606,21 @@ def handle_memory_queries(prompt):
     """
     Main function to handle user commands and allocate Trust Buckets.
     """
-    prompt = prompt.lower().strip()
+    prompt = prompt.strip()
 
-    # Save and allocate TrustBuilder
-    match = re.search(r"\bsave\s+as\s+(\w+)\s+trustbuilder\s*:\s*(.+)", prompt, re.IGNORECASE)
-    if match:
-        specified_bucket = match.group(1).capitalize()
-        content_to_save = match.group(2).strip()
+    valid_buckets = ["Stability", "Development", "Relationship", "Benefit", "Vision", "Competence"]
+
+    # Case 1: Save this as [bucket] trust builder: [content]
+    match_save_this_specific = re.search(r"\bsave\s+(this\s+)?as\s+(\w+)\s+trust\s+builders?\s*:\s*(.+)", prompt, re.IGNORECASE)
+    if match_save_this_specific:
+        specified_bucket = match_save_this_specific.group(2).capitalize()
+        content_to_save = match_save_this_specific.group(3).strip()
 
-        valid_buckets = ["Stability", "Development", "Relationship", "Benefit", "Vision", "Competence"]
         if specified_bucket in valid_buckets:
             if content_to_save:
                 assistant_response = handle_save_trustbuilder(content_to_save, specified_bucket)
             else:
-                assistant_response = "No content provided. Please include content after the 'save as' command."
+                assistant_response = "No content provided. Please include content after 'save this as [bucket] trust builder:'."
         else:
             assistant_response = f"Invalid Trust Bucket '{specified_bucket}'. Valid buckets are: {', '.join(valid_buckets)}."
 
@@ -1629,14 +1630,19 @@ def handle_memory_queries(prompt):
             st.markdown(assistant_response)
         return None
 
-    # Show saved TrustBuilders
-    elif "find my saved trustbuilders" in prompt.lower() or "show my saved trustbuilders" in prompt.lower():
-        trustbuilders = fetch_trustbuilders(st.session_state.get("wix_user_id", "default_user"))
-        if trustbuilders:
-            saved_content = "\n".join([f"- {entry['message']}" for entry in trustbuilders.values()])
-            assistant_response = f"Here are your saved TrustBuilders:\n{saved_content}"
+    # Case 2: Save this under [bucket]: [content]
+    match_save_under_specific = re.search(r"\bsave\s+(this\s+)?under\s+(\w+)\s*:\s*(.+)", prompt, re.IGNORECASE)
+    if match_save_under_specific:
+        specified_bucket = match_save_under_specific.group(2).capitalize()
+        content_to_save = match_save_under_specific.group(3).strip()
+
+        if specified_bucket in valid_buckets:
+            if content_to_save:
+                assistant_response = handle_save_trustbuilder(content_to_save, specified_bucket)
+            else:
+                assistant_response = "No content provided. Please include content after 'save this under [bucket]:'."
         else:
-            assistant_response = "You haven't saved any TrustBuilders yet."
+            assistant_response = f"Invalid Trust Bucket '{specified_bucket}'. Valid buckets are: {', '.join(valid_buckets)}."
 
         # Save response to chat history and display it
         st.session_state.chat_history.append({"role": "assistant", "content": assistant_response})
@@ -1644,6 +1650,31 @@ def handle_memory_queries(prompt):
             st.markdown(assistant_response)
         return None
 
+    # Case 3: Save and allocate: [content] (automatic allocation)
+    match_save_allocate_auto = re.search(r"\bsave\s+(this\s+)?and\s+allocate\s*:\s*(.+)", prompt, re.IGNORECASE)
+    if match_save_allocate_auto:
+        content_to_save = match_save_allocate_auto.group(2).strip()
+        if content_to_save:
+            assistant_response = handle_save_trustbuilder(content_to_save)  # Automatically allocate bucket
+        else:
+            assistant_response = "No content provided. Please include content after 'save and allocate:'."
+
+        # Save response to chat history and display it
+        st.session_state.chat_history.append({"role": "assistant", "content": assistant_response})
+        with st.chat_message("assistant"):
+            st.markdown(assistant_response)
+        return None
+
+    # Default case: Unrecognized command
+    assistant_response = (
+        "Command not recognized. Please use a valid format like:\n"
+        "- Save this as [bucket] trust builder: [content]\n"
+        "- Save and allocate: [content]\n"
+        "- Save this under [bucket]: [content]"
+    )
+    st.session_state.chat_history.append({"role": "assistant", "content": assistant_response})
+    with st.chat_message("assistant"):
+        st.markdown(assistant_response)
     return None
 
 def handle_save_trustbuilder(content, specified_bucket=None):