Add mantra to context mgmt & Monday msg

app/assistants.py

@@ -676,7 +676,7 @@ class Assistant:
                     user_info = '** If the user asks for their progress, also include a link to their Revelation Dashboard: {OURCOACH_DASHBOARD_URL} so that they can find out more **\n\n'
                     if category == "personal":
                         user_info += f"** Personal Information **\n\n{self.cm.user.user_info}"
-                        user_info += f"\n\n** User's Mantra This Week**\n\n{self.cm.user.mantra}"
+                        user_info += f"\n\n** User's Mantra This Week:**\n\n{self.cm.user.mantra or "Mantra not available."}"
                     elif category == "challenges":
                         user_info += f"** User's Challenges (prioritise ONGOING challenges) **\n\n{self.cm.user.challenges}\n\nLet the user know that ongoing challenges from their growth guide will be integrated into their day-to-day interaction."
                     elif category == "recommended_actions":

app/user.py

@@ -6,6 +6,7 @@ import pandas as pd
 from datetime import datetime, timezone
 import json
 from app.assistants import Assistant 
+from app.exceptions import DBError
 import glob
 import pickle  # Replace dill with pickle
 import random
@@ -328,9 +329,33 @@ class User:
         return False
 
     @catch_error
-    def set_mantra(self, mantra):
-        logger.info(f"Setting mantra: {mantra}", extra={"user_id": self.user_id, "endpoint": "set_mantra"})
-        self.mantra = mantra
+    def set_mantra(self):
+        ### To save mantra in database to user object
+        logger.info(f"Getting mantra from user...", extra={"user_id": self.user_id, "endpoint": "get_mantra"})
+        user_id = self.user_id
+        db_params = {
+            'dbname': 'ourcoach',
+            'user': 'ourcoach',
+            'password': 'hvcTL3kN3pOG5KteT17T',
+            'host': 'staging-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("SELECT mantra FROM {table} WHERE user_id = %s").format(table=sql.Identifier('public', 'user_growth_status'))
+                    cursor.execute(query, (user_id,))
+                    row = cursor.fetchone()
+                    if (row):
+                        colnames = [desc[0] for desc in cursor.description]
+                        user_data = dict(zip(colnames, row))
+                        ### SAVE MANTRA IN USER OBJECT
+                        self.mantra = user_data['mantra']
+                    else:
+                        logger.warning(f"No user info found for {user_id}", extra={'user_id': user_id, 'endpoint': "get_mantra"})
+        except psycopg2.Error as e:
+            logger.error(f"Database error while retrieving user info for {user_id}: {e}", extra={'user_id': user_id, 'endpoint': "get_mantra"})
+            raise DBError(user_id=user_id, message="Error retrieving user info", code="SQLError", e=str(e))
 
     @catch_error
     def set_goal(self, goal, goal_area, add=True, completed=False):
@@ -847,6 +872,8 @@ class User:
 
         Today is day {self.cumulative_plan_day} of the user's growth journey (out of {final_day} days). You may (or may not) mention this occasionally in your first message of the day.
         
+        If today is a "Monday" or "Mon", you must include the user's Mantra of the Week : {self.mantra} to your first message of the day (include with today-theme's first message below)
+
         Today's Theme:
         {formatted_message}
         """
@@ -869,6 +896,8 @@ class User:
         self.conversations.delete_hidden_messages()
         # update the date in the state
         self.conversations.state['date'] = date
+        # update mantra
+        self.set_mantra()
 
         action = self.growth_plan.current()
 

app/utils.py

@@ -984,7 +984,8 @@ def get_user_life_status(user_id):
     mantra = json.loads(response.choices[0].message.content)["mantra_of_the_week"]
     
     # Update the users mantra
-    user.set_mantra(mantra)
+    # user.set_mantra(mantra)
+    # We remove because we want the mantra to be updated weekly (by backend), not updated everytime we call this endpoint/func
 
     cumulative_life_score = {
             "overall": user.personal_growth_score + user.career_growth_score + user.relationship_score + user.mental_well_being_score + user.health_and_wellness_score,