Post Growth Guide feature

app/assistants.py

@@ -496,8 +496,9 @@ class Assistant:
             elif tool.function.name == "end_conversation":
                 day_n = json.loads(tool.function.arguments)['day_n']
                 completed_micro_action = json.loads(tool.function.arguments)['completed_micro_action']
-
-                self.cm.user.update_micro_action_status(completed_micro_action)
+                # NOTE: must get the current microaction also
+                if completed_micro_action:
+                    self.cm.user.update_micro_action_status("COMPLETE")
 
                 # NOTE: we will record whether the user has completed the theme for the day
                 # NOTE: if no, we will include a short followup message to encourage the user the next day
@@ -542,9 +543,20 @@ class Assistant:
                 logger.info(f"Successfully cancelled run", 
                     extra={"user_id": self.cm.user.user_id, "endpoint": "assistant_change_goal"})
                 return "change_goal", just_finish_intro 
-
-            elif tool.function.name == "change_goal":
-                pass
+            elif tool.function.name == "set_reminder":
+                reminder = json.loads(tool.function.arguments)["content"]
+                time_hour = json.loads(tool.function.arguments)["time_hour"]
+                time_minute = json.loads(tool.function.arguments)["time_minute"]
+                time_second = json.loads(tool.function.arguments)["time_second"]
+                #timestamp = datetime(time_hour:time_minute:time_second)
+                timestamp = datetime.now().replace(hour=time_hour, minute=time_minute, second=time_second)
+                logger.info(f"Setting reminder: {reminder} for: {timestamp}", 
+                        extra={"user_id": self.cm.user.user_id, "endpoint": "assistant_set_reminder"})
+                self.cm.user.set_reminder({"reminder": reminder, "timestamp": timestamp})
+                tool_outputs.append({
+                    "tool_call_id": tool.id,
+                    "output": f"** Reminder ({reminder}) set for: {timestamp} **"
+                })
             
         # Submit all tool outputs at once after collecting them in a list
         if tool_outputs:

app/flows.py

@@ -1,3 +1,44 @@
+POST_GG_STATE = f"""
+* Post-Growth Guide Message *
+Objective: Provide a space for reflection on the user's growth guide session while providing encouragement and inspiration to the user to help them stay motivated on their growth journey.
+
+The user:
+{{}}
+
+Summary of the Growth Guide Session: 
+{{}}
+
+Growth Guide Session Notes: 
+{{}}
+
+Guidelines:
+** PRIORITIZE THESE INSTRUCTIONS BELOW AND IGNORE OVERLAPPING INSTRUCTIONS FROM THE SYSTEM PROMPT **
+
+** IMPORTANT **: When you feel that the user is replying with short messages and not in the mood to chat, end the conversation immediately. Always try to make the conversation quick & not asking too many questions
+
+** IMPORTANT **: Only send maximum 2 questions in this day. Do **not** overwhelm the user with too many questions! And there should only be one question mark in one message! (max one question in a single message)
+
+** IMPORTANT **: Don't forget the message formatting rule where it applies: Separate statements and questions with a single line break and Encapsulate the question with asterisk, like this: *question*! 
+
+** IMPORTANT **: Do not explicitly state the function name that you are calling in the response
+
+** IMPORTANT **: You must end the interaction gracefully. End with a valueable and strong statement to encourage the user.
+
+** IMPORTANT **: Although asking the user for their feedback and views is good, ensure not to pose too many questions to the user. Maintain a healthy coaching conversation flow.
+
+** IMPORTANT **: Don't ask the user for ideas on what to do. Remember that you are the coach and you should be assert ideas/actions to the user. You will be punished if you don't adhere to this.
+
+** IMPORTANT **: If the user asks anything that results into a list, you only give maximum 3 items !
+
+** IMPORTANT **: Keep the conversation creative, but make sure your responses short and succint like you are interacting via text messages as a modern and adept texter.
+
+Keep the message short (only 2-3 lines maximum), please.
+
+Task:
+The user has just completed their growth guide session. Generate an appropriate and encouraging message that reflects on the user's growth guide session and provides inspiration for their continued growth journey. Use the user's profile information and the summary of the growth guide session to craft a personalized and motivational message. Remember to keep the message concise and engaging to encourage the user to continue their growth journey with enthusiasm.
+
+"""
+
 MICRO_ACTION_STATE = f"""
 ** Micro-Action ** (PLEASE READ CAREFULLY)
 Objective: Build momentum for acheiving the user's goal through bite-sized, actionable tasks that feel achievable quickly. Do **not** say "Today's micro-action is:"! Just state the micro-action directly!

app/main.py

@@ -380,7 +380,7 @@ def process_gg_session(request: GGItem, api_key: str = Security(get_api_key)):
     session_data = get_growth_guide_session(user_id, session_id)
     
     # update user
-    response = user.process_growth_guide_session(session_data)
+    response = user.process_growth_guide_session(session_data, session_id)
     logger.info(f"GG session processed: {session_id}, response: {response}", extra={"user_id": user_id, "endpoint": "/process_gg_session"})
     return {"response": response}
 

app/user.py

@@ -10,10 +10,12 @@ import pickle  # Replace dill with pickle
 import random
 import logging
 
-from app.flows import FINAL_SUMMARY_STATE, FINAL_SUMMARY_STATE, MICRO_ACTION_STATE, MOTIVATION_INSPIRATION_STATE, OPEN_DISCUSSION_STATE, PROGRESS_REFLECTION_STATE, PROGRESS_SUMMARY_STATE
+from app.flows import FINAL_SUMMARY_STATE, FINAL_SUMMARY_STATE, MICRO_ACTION_STATE, MOTIVATION_INSPIRATION_STATE, OPEN_DISCUSSION_STATE, POST_GG_STATE, PROGRESS_REFLECTION_STATE, PROGRESS_SUMMARY_STATE
 from pydantic import BaseModel
 from datetime import datetime
 
+from app.utils import update_growth_guide_summary
+
 class UserDataItem(BaseModel):
         role: str 
         content: str
@@ -271,8 +273,10 @@ class User:
         self.done_first_reflection = None
         self.goal = []
         self.micro_actions = []
+        self.recommended_micro_actions = []
         self.challenges = []
         self.other_focusses = []
+        self.reminders = None
 
         # Read growth_plan.json and store it
         growth_plan = {"growthPlan": [
@@ -345,16 +349,50 @@ class User:
         self.conversations = ConversationManager(client, self, asst_id)
 
     def get_current_goal(self):
-        return self.goal[-1] if self.goal else None
-
-    def set_goal(self, goal):
-        self.goal.append({"goal": goal, "created_at": pd.Timestamp.now().strftime("%d-%m-%Y %a %H:%M:%S")})
-
-    def update_micro_action_status(self, completed_micro_action):
-        if completed_micro_action:
-            self.micro_actions[-1]["status"] = "COMPLETE"
-            self.micro_actions[-1]["updated_at"] = pd.Timestamp.now().strftime("%d-%m-%Y %a %H:%M:%S")
-
+        # look for the goal with status = ONGOING
+        for goal in self.goal:
+            if goal.status == "ONGOING":
+                return goal.content
+        else:
+            return self.goal[-1].content if self.goal else None
+
+    def update_goal(self, goal, status, content=None):
+        for g in self.goal:
+            if g.content == goal:
+                g.status = status
+                g.updated_at = pd.Timestamp.now().strftime("%d-%m-%Y %a %H:%M:%S")
+                if content:
+                    g.content = content
+                return True
+        return False
+
+    def set_goal(self, goal, add=True, completed=False):
+        current_goal = self.get_current_goal()
+        
+        if completed:
+            self.update_goal(current_goal, "COMPLETED")
+
+        if current_goal is None:
+            new_goal = UserDataItem(role="assistant", content=goal, user_id=self.user_id, status="ONGOING", created_at=pd.Timestamp.now().strftime("%d-%m-%Y %a %H:%M:%S"), updated_at=pd.Timestamp.now().strftime("%d-%m-%Y %a %H:%M:%S"))
+            self.goal.append(new_goal)
+
+        if add:
+            if current_goal:
+                # update current_goal status to "IDLE"
+                self.update_goal(current_goal, "IDLE")
+            new_goal = UserDataItem(role="assistant", content=goal, user_id=self.user_id, status="ONGOING", created_at=pd.Timestamp.now().strftime("%d-%m-%Y %a %H:%M:%S"), updated_at=pd.Timestamp.now().strftime("%d-%m-%Y %a %H:%M:%S"))
+            self.goal.append(new_goal)
+        else:
+            self.update_goal(current_goal, "ONGOING", content=goal)
+
+    def update_recommended_micro_action_status(self, micro_action, status):
+        for ma in self.recommended_micro_actions:
+            if ma.content == micro_action:
+                ma.status = status
+                ma.updated_at = pd.Timestamp.now().strftime("%d-%m-%Y %a %H:%M:%S")
+                return True
+        return False
+    
     def add_ai_message(self, text):
         self.conversations._add_ai_message(text)
         return text
@@ -364,6 +402,7 @@ class User:
         self.growth_plan.reset()
         self.goal = []
         self.micro_actions = []
+        self.recommended_micro_actions = []
         self.challenges = []
         self.other_focusses = []
 
@@ -452,9 +491,21 @@ class User:
             # reset the growth_plan
             self.growth_plan.reset()
 
+        if self.reminders is not None:
+            logger.info(f"Adding reminders: {self.reminders}", extra={"user_id": self.user_id, "endpoint": "user_send_message"})
+            response['reminders'] = self.reminders
+            self.reminders = None
+        else:
+            response['reminders'] = None
+        
         logger.info(f"Response: {response}", extra={"user_id": self.user_id, "endpoint": "user_send_message"})
         return response
 
+    def set_reminder(self, reminder):
+        self.reminders = []
+        self.reminders.append(reminder)
+        logger.info(f"Reminders: {self.reminders}", extra={"user_id": self.user_id, "endpoint": "set_reminder"})
+
     def get_messages(self, exclude_system_msg=True):
         if not exclude_system_msg:
             return self.conversations._get_current_thread_history(False)
@@ -471,10 +522,19 @@ class User:
             formatted_message = MOTIVATION_INSPIRATION_STATE.format(self.get_current_goal(), day, len(self.growth_plan.array))
         elif theme == "PROGRESS_REFLECTION_STATE":
             formatted_message = PROGRESS_REFLECTION_STATE.format(self.get_current_goal(), day, len(self.growth_plan.array))
+            if len(self.challenges):
+                challenge = self.challenges.pop(0)
+                formatted_message += f"\n\n** Today, reflect on the users' challenge of: {challenge.content} **"
         elif theme == "MICRO_ACTION_STATE":
             formatted_message = MICRO_ACTION_STATE.format(self.get_current_goal(), day, len(self.growth_plan.array))
+            if len(self.recommended_micro_actions):
+                todays_micro_action = self.recommended_micro_actions.pop(0)
+                formatted_message += f"\n\n** Today's Micro Action is: {todays_micro_action.content} **"
         elif theme == "OPEN_DISCUSSION_STATE":
             formatted_message = OPEN_DISCUSSION_STATE.format(self.get_current_goal(), day, len(self.growth_plan.array))
+            if len(self.other_focusses):
+                focus = self.other_focusses.pop(0)
+                formatted_message += f"\n\n** Today, focus the discussion on: {focus.content} **"
         elif theme == "PROGRESS_SUMMARY_STATE":
             formatted_message = PROGRESS_SUMMARY_STATE.format(self.get_current_goal(), day, len(self.growth_plan.array))
         elif theme == "FINAL_SUMMARY_STATE":
@@ -495,7 +555,9 @@ class User:
         response = self.conversations._send_morning_message(prompt)
 
         if theme == "MICRO_ACTION_STATE":
-            self.micro_actions.append({'micro_action': response, 'status': 'PENDING', 'created_at': pd.Timestamp.now().strftime("%d-%m-%Y %a %H:%M:%S"), 'updated_at': pd.Timestamp.now().strftime("%d-%m-%Y %a %H:%M:%S")})
+            # check for any recommended microactions
+            micro_action = UserDataItem(role="assistant", content=response['content'], user_id=self.user_id, status="PENDING", created_at=pd.Timestamp.now().strftime("%d-%m-%Y %a %H:%M:%S"), updated_at=pd.Timestamp.now().strftime("%d-%m-%Y %a %H:%M:%S"))
+            self.micro_actions.append(micro_action)
 
         return response
 
@@ -564,14 +626,14 @@ class User:
             ],
             temperature=0.2
         )
-        return response.choices[0].message.content
+        return {'overview': response.choices[0].message.content}
 
     def _update_user_data(self, data_type, text_input, extra_text=""):
         data_mapping = {
             'micro_actions': {
                 'prompt_description': 'micro actions',
                 'status': 'RECOMMENDED',
-                'attribute': 'micro_actions',
+                'attribute': 'recommended_micro_actions',
                 'endpoint': f'update_{data_type}',
             },
             'challenges': {
@@ -629,7 +691,7 @@ class User:
             logger.error(f"Failed to update {data_type}: {e}", extra={"user_id": self.user_id})
 
     def update_user_data(self, gg_report):
-        self.micro_actions = self._update_user_data('micro_actions', gg_report[0]['answer'])
+        self.recommended_micro_actions = self._update_user_data('micro_actions', gg_report[0]['answer'])
 
         extra_text = f"User has new challenge:\n{gg_report[1]['answer']}\n\n"
         self._update_user_data('challenges', gg_report[2]['answer'], extra_text=extra_text)
@@ -659,7 +721,7 @@ class User:
         else:
             logger.info(f"User goal remains unchanged.", extra={"user_id": self.user_id, "endpoint": "_update_goal"})
 
-    def process_growth_guide_session(self, session_data):
+    def process_growth_guide_session(self, session_data, booking_id):
         logger.info(f"Processing growth guide session data: {session_data}", extra={"user_id": self.user_id, "endpoint": "process_growth_guide_session"})
 
         # Generate the ourcoach_report (summary)
@@ -667,14 +729,23 @@ class User:
         gg_report = session_data["gg_report"]
 
         ourcoach_report = self._summarize_zoom(zoom_ai_summary)
+        # add this report to the db
+        update_growth_guide_summary(self.user_id, booking_id, ourcoach_report)
 
         # Update user data based on growth guide answers
         self.update_user_data(gg_report)
-        self.update_user_info(gg_report[5]['answer'] + "\n\n" + ourcoach_report)
+        self.update_user_info(gg_report[5]['answer'] + "\n\n" + ourcoach_report['overview'])
 
         # Send hidden message to AI to generate the response
         logger.info(f"Sending hidden message to AI to generate response", extra={"user_id": self.user_id, "endpoint": "process_growth_guide_session"})
-        response = self.conversations._send_morning_message("I have completed the growth guide session. Please generate the response.")
+        
+        # post_gg_prompt = POST_GG_STATE.format(self.user_info, ourcoach_report, gg_report)
+        post_gg_promt = f"""The user: {self.user_info} has completed their growth guide session. 
+        Send them a warm and congratulatory message acknoledging this and a link to their web dasboard.
+        Example:
+        Hey <user>! [...], you can find a details of your session on your revelation dashboard: <link>"""
+        
+        response = self.conversations._send_morning_message(post_gg_promt)
         logger.info(f"Response: {response}", extra={"user_id": self.user_id, "endpoint": "process_growth_guide_session"})
         return response
 
@@ -783,6 +854,7 @@ Use bullet points or numbered lists where appropriate to enhance readability."""
     def __str__(self):
         return f"""User(user_id={self.user_id}
                 micro_actions={self.micro_actions}
+                recommended_actions={self.recommended_micro_actions}
                 challenge={self.challenges}
                 other_focusses={self.other_focusses}
                 goals={self.goal}
@@ -793,6 +865,7 @@ Use bullet points or numbered lists where appropriate to enhance readability."""
     def __repr__(self):
         return f"""User(user_id={self.user_id}
                 micro_actions={self.micro_actions}
+                recommended_actions={self.recommended_micro_actions}
                 challenge={self.challenges}
                 other_focusses={self.other_focusses}
                 goals={self.goal}

app/utils.py

@@ -941,6 +941,33 @@ def get_user_info(user_id):
         logger.error(f"Database error while retrieving user info for {user_id}: {e}", extra={'user_id': user_id, 'endpoint': function_name})
         return None
 
+def update_growth_guide_summary(user_id, session_id, ourcoach_summary):
+    function_name = update_growth_guide_summary.__name__
+    logger.info(f"Updating growth guide summary for user {user_id} and session {session_id}", extra={'user_id': user_id, 'endpoint': function_name})
+
+    db_params = {
+        'dbname': 'ourcoach',
+        'user': 'ourcoach',
+        'password': 'tfPMEnYFWKGSQqZWVhQo',
+        'host': 'production-ourcoach.cx8se8o0iaiy.ap-southeast-1.rds.amazonaws.com',
+        'port': '5432'
+    }
+
+    try:
+        with psycopg2.connect(**db_params) as conn:
+            with conn.cursor() as cursor:
+                query = sql.SQL("""
+                    UPDATE {table}
+                    SET ourcoach_summary = %s
+                    WHERE user_id = %s AND booking_id = %s
+                """).format(table=sql.Identifier('public', 'user_notes'))
+                cursor.execute(query, (json.dumps(ourcoach_summary), user_id, session_id))
+                conn.commit()
+                logger.info(f"Growth guide summary updated successfully for user {user_id} and session {session_id}", extra={'user_id': user_id, 'endpoint': function_name})
+    except psycopg2.Error as e:
+        logger.error(f"Database error while updating growth guide summary: {e}", extra={'user_id': user_id, 'endpoint': function_name})
+        raise e
+
 def add_growth_guide_session(user_id, session_id, coach_id, session_started_at, zoom_ai_summary, gg_report, ourcoach_summary):
     function_name = add_growth_guide_session.__name__
     logger.info(f"Adding growth guide session for user {user_id} and session {session_id}", extra={'user_id': user_id, 'endpoint': function_name})