added get_daily messages

app/main.py

@@ -384,6 +384,13 @@ def process_gg_session(request: GGItem, api_key: str = Security(get_api_key)):
     logger.info(f"GG session processed: {session_id}, response: {response}", extra={"user_id": user_id, "endpoint": "/process_gg_session"})
     return {"response": response}
 
+@app.get("/user_daily_messages")
+def get_daily_message(user_id: str,  api_key: str = Security(get_api_key)):
+    logger.info("Getting daily messages", extra={"user_id": user_id, "endpoint": "/user_daily_messages"})
+    user = get_user(user_id)
+    daily_messages = user.get_daily_messages()
+    return {"response": daily_messages}
+
 @app.post("/batch_refresh_users")
 def refresh_multiple_users(user_ids: List[str], api_key: str = Security(get_api_key)):
     logger.info("Refreshing multiple users", extra={"endpoint": "/batch_refresh_users"})

app/user.py

@@ -40,6 +40,7 @@ class ConversationManager:
         self.state = {'date': pd.Timestamp.now().strftime("%d-%m-%Y %a %H:%M:%S")}
 
         self.current_thread = self.create_thread()
+        self.daily_thread = None
 
         logger.info("Initializing conversation state", extra={"user_id": self.user.user_id, "endpoint": "conversation_init"})
     
@@ -71,13 +72,13 @@ class ConversationManager:
                                 """)
         return thread
 
-    def _get_current_thread_history(self, remove_system_message=True, msg=None, thread=None):
+    def _get_current_thread_history(self, remove_system_message=True, _msg=None, thread=None):
         if thread is None:
             thread = self.current_thread
         if not remove_system_message:
             return [{"role": msg.role, "content": msg.content[0].text.value} for msg in self.client.beta.threads.messages.list(thread.id, order="asc")]
-        if msg:
-            return [{"role": msg.role, "content": msg.content[0].text.value} for msg in self.client.beta.threads.messages.list(thread.id, order="asc", after=msg.id)][1:]
+        if _msg:
+            return [{"role": msg.role, "content": msg.content[0].text.value} for msg in self.client.beta.threads.messages.list(thread.id, order="asc", after=_msg.id)][1:]
         return [{"role": msg.role, "content": msg.content[0].text.value} for msg in self.client.beta.threads.messages.list(thread.id, order="asc")][1:] # remove the system message
 
     def add_message_to_thread(self, thread_id, role, content):
@@ -151,6 +152,32 @@ class ConversationManager:
     def _add_ai_message(self, text):
         return self.add_message_to_thread(self.current_thread.id, "assistant", text)
 
+    def get_daily_thread(self):
+        if self.daily_thread is None:
+            messages = self._get_current_thread_history(remove_system_message=False)
+
+            self.daily_thread = self.client.beta.threads.create(
+            messages=messages[:30]
+            )
+
+            # Add remaining messages one by one if there are more than 30
+            for msg in messages[30:]:
+                self.add_message_to_thread(
+                    self.daily_thread.id,
+                    msg['role'],
+                    msg['content']
+                )
+            self.last_daily_message = list(self.client.beta.threads.messages.list(self.daily_thread.id, order="asc"))[-1]
+        else:
+            messages = self._get_current_thread_history(remove_system_message=False, _msg=self.last_daily_message)
+            self.client.beta.threads.delete(self.daily_thread.id)
+            self.daily_thread = self.client.beta.threads.create(messages=messages)
+            self.last_daily_message = list(self.client.beta.threads.messages.list(self.daily_thread.id, order="asc"))[-1]
+        logger.info(f"Daily Thread: {self._get_current_thread_history(thread=self.daily_thread)}", extra={"user_id": self.user.user_id, "endpoint": "send_morning_message"})
+        logger.info(f"Last Daily Message: {self.last_daily_message}", extra={"user_id": self.user.user_id, "endpoint": "send_morning_message"})
+        return self._get_current_thread_history(thread=self.daily_thread)
+        # [{"role":, "content":}, ....]
+
     def _send_morning_message(self, text):
         # create a new thread
         # OPENAI LIMITATION: Can only attach a maximum of 32 messages when creating a new thread
@@ -735,6 +762,9 @@ Use bullet points or numbered lists where appropriate to enhance readability."""
         except Exception as e:
             return False
     
+    def get_daily_messages(self):
+        return self.conversations.get_daily_thread()
+
     def change_assistant(self, asst_id):
         self.asst_id = asst_id
         self.conversations.assistants['general'] = Assistant(self.asst_id, self.conversations)