Integrate daily limit check into message display logic

pages/chatbot.py

@@ -4,7 +4,7 @@ from langchain.prompts import PromptTemplate
 from model import selector
 from util import getYamlConfig
 from st_copy_to_clipboard import st_copy_to_clipboard
-from subscription.index import is_usage_limit_reached
+from subscription.index import is_usage_limit_reached, daily_limit_enabled
 
 def display_messages():
 
@@ -124,10 +124,13 @@ def page():
     # Displaying messages
     display_messages()
 
-    reached = is_usage_limit_reached()
     message = ""
-    if reached:
-        message = "Vous avez atteint la limite quotidienne de messages."
+    reached = False
+
+    if daily_limit_enabled():
+        reached = is_usage_limit_reached()
+        if reached:
+            message = "Vous avez atteint la limite quotidienne de messages."
 
     user_query = st.chat_input(message, disabled=reached)