add counter limit

src/app.py

@@ -16,7 +16,7 @@ from streamlit_option_menu import option_menu
 import os
 import re
 from model import ask
-from handler import save_feedback, translate_answer, check_question_feedback, translate_text, is_overlimit
+from handler import save_feedback, translate_answer, check_question_feedback, translate_text, is_overlimit, refresh_web_counter
 from text import feedback_instr, no_affiliation, description, no_replacement_for_official_advice, source_of_answer_short, source_of_answer
 from langdetect import detect
 from deep_translator import GoogleTranslator
@@ -197,7 +197,10 @@ if selected == "Chatbot":
             "content": "Hello! I'm Instant. How can I help you today?"
         }]
 
-    is_limit_exceeded = is_overlimit()
+    is_limit_exceeded = False
+    refresh = refresh_web_counter()
+    if refresh:
+        is_limit_exceeded = is_overlimit()
 
     working_hours = (10 <= current_hour < 13) or (17 <= current_hour < 20)
     if working_hours and not is_limit_exceeded:

src/handler.py

@@ -145,6 +145,23 @@ def add_question_ticker():
         print("Error updating web counter:", e)
         return False
 
+def refresh_web_counter():
+    current_hour = datetime.now().hour
+    refresh_hour = current_hour == 1
+
+    if refresh_hour:
+        try:
+            web_counter_collection.update_one(
+                {"type": "web"},
+                {
+                    "$set": {"counter": 0, "timestamp": datetime.now(timezone.utc)}
+                }
+            )
+        except Exception as e:
+            print("Error refreshing web counter:", e)
+
+    return True
+
 def is_overlimit():
     try:
         counter = web_counter_collection.find_one({"type": "web"})