token regen patch and bug fix

apps/ai_tutor/app.py

@@ -235,7 +235,7 @@ async def cooldown(request: Request):
     user_details = await get_user_details(user_info["email"])
     current_datetime = get_time()
     cooldown, cooldown_end_time = await check_user_cooldown(
-        user_details, current_datetime
+        user_details, current_datetime, COOLDOWN_TIME, TOKENS_LEFT, REGEN_TIME
     )
     print(f"User in cooldown: {cooldown}")
     print(f"Cooldown end time: {cooldown_end_time}")

apps/ai_tutor/helpers.py

@@ -68,7 +68,7 @@ async def reset_tokens_for_user(user_info, TOKENS_LEFT, REGEN_TIME):
     # Calculate how many tokens should have been regenerated proportionally
     if current_tokens < max_tokens:
         # Calculate the regeneration rate per second based on REGEN_TIME for full regeneration
-        regeneration_rate_per_second = max_tokens / REGEN_TIME
+        regeneration_rate_per_second = (max_tokens - current_tokens) / REGEN_TIME
 
         # Calculate how many tokens should have been regenerated based on the elapsed time
         tokens_to_regenerate = int(