Alignment

app/user.py

@@ -11,8 +11,9 @@ import random
 import logging
 import psycopg2
 from psycopg2 import sql
+import random
 
-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, EDUCATION_STATE, FOLLUP_ACTION_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, EDUCATION_STATE, FOLLUP_ACTION_STATE, FUNFACT_STATE
 from pydantic import BaseModel
 from datetime import datetime
 
@@ -400,18 +401,10 @@ class User:
                     },
                     {
                     "day": 4,
-                    "coachingTheme": "MICRO_ACTION_STATE"
+                    "coachingTheme": "FUNFACT_STATE"
                     },
                     {
                     "day": 5,
-                    "coachingTheme": "FOLLUP_ACTION_STATE"
-                    },
-                    {
-                    "day": 6,
-                    "coachingTheme": "OPEN_DISCUSSION_STATE"
-                    },
-                    {
-                    "day": 7,
                     "coachingTheme": "FINAL_SUMMARY_STATE"
                     }
                 ]
@@ -841,7 +834,11 @@ class User:
         elif theme == "FOLLUP_ACTION_STATE":
             reminder_message = "\n".join([f"{i+1}. {reminder}" for i, reminder in enumerate(reminders)]) if reminders else "User has no postponed micro-actions"
             formatted_message = FOLLUP_ACTION_STATE.format(self.get_current_goal(), day, len(self.growth_plan.array), reminder_message)
-        
+        elif theme == "FUNFACT_STATE":
+            topics = ["Fun Fact about the User's Goal", "How Personality Type is affecting/shaping their behaviour towards the goal", "How Love Language may impact and be relevant toward the goal"]
+            randomized_topic = random.choice(topics)
+            formatted_message = FUNFACT_STATE.format(self.get_current_goal(), day, len(self.growth_plan.array), randomized_topic)
+
         # prompt = f"""** It is a new day: {date} ** 
         # Additional System Instruction: 
         # - Remember all of the user's personal information. Use this information to be as personalised as possible when conversing by including it in your responses where relevant.