Merge pull request #6 from tools4ds/retargeting

apps/ai_tutor/helpers.py

@@ -19,9 +19,13 @@ async def check_user_cooldown(
     last_message_time_str = user_info["metadata"].get("last_message_time")
 
     # Convert from ISO format string to datetime object and ensure UTC timezone
-    last_message_time = datetime.fromisoformat(last_message_time_str).replace(
-        tzinfo=timezone.utc
-    )
+    try:
+        last_message_time = datetime.fromisoformat(last_message_time_str).replace(
+            tzinfo=timezone.utc
+        )
+    except Exception:  # this probably means the user has never sent a message before
+        return False, None
+
     current_time = datetime.fromisoformat(current_time).replace(tzinfo=timezone.utc)
 
     # Calculate the elapsed time
@@ -47,9 +51,13 @@ async def reset_tokens_for_user(user_info, TOKENS_LEFT, REGEN_TIME):
     user_info = convert_to_dict(user_info)
     last_message_time_str = user_info["metadata"].get("last_message_time")
 
-    last_message_time = datetime.fromisoformat(last_message_time_str).replace(
-        tzinfo=timezone.utc
-    )
+    try:
+        last_message_time = datetime.fromisoformat(last_message_time_str).replace(
+            tzinfo=timezone.utc
+        )
+    except Exception:  # this probably means the user has never sent a message before
+        last_message_time = datetime.min.replace(tzinfo=timezone.utc)
+
     current_time = datetime.fromisoformat(get_time()).replace(tzinfo=timezone.utc)
 
     # Calculate the elapsed time since the last message

apps/ai_tutor/templates/login.html

@@ -114,7 +114,7 @@
         class="avatar"
       />
       <h1>Terrier Tutor</h1>
-      <p>Welcome to the DS598 AI Tutor. Please sign in to continue.</p>
+      <p>Welcome to the DS701 AI Tutor. Please sign in to continue.</p>
       <form action="/login/google" method="get">
         <button type="submit" class="button">Sign in with Google</button>
       </form>