Update app.py

app.py

@@ -27,13 +27,30 @@ def check_and_reset_user_counter(user_id):
     save_user_session_data(all_user_data)
     return user_data["test_counter"]
 
+def save_user_session_data(all_user_data):
+    with open(SESSION_FILE, 'w') as f:
+        json.dump(all_user_data, f)
+
+# Function to load user session data
+def load_user_session_data(user_id):
+    if os.path.exists(SESSION_FILE):
+        with open(SESSION_FILE, 'r') as f:
+            all_user_data = json.load(f)
+    else:
+        all_user_data = {}
+    
+    if user_id not in all_user_data:
+        all_user_data[user_id] = {"last_reset_date": str(date.today()), "test_counter": 0}
+    
+    return all_user_data
+
 # Function to increment the counter for a specific user
 def increment_user_counter(user_id):
     all_user_data = load_user_session_data(user_id)
     all_user_data[user_id]["test_counter"] += 1
     save_user_session_data(all_user_data)
     return all_user_data[user_id]["test_counter"]
-
+    
 # Function to get or create user ID from cookie
 def get_or_create_user_id(request: gr.Request):
     if request is None: