cyberandy commited on
Commit
25e9b9f
·
verified ·
1 Parent(s): 5f975a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -1
app.py CHANGED
@@ -27,13 +27,30 @@ def check_and_reset_user_counter(user_id):
27
  save_user_session_data(all_user_data)
28
  return user_data["test_counter"]
29
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
30
  # Function to increment the counter for a specific user
31
  def increment_user_counter(user_id):
32
  all_user_data = load_user_session_data(user_id)
33
  all_user_data[user_id]["test_counter"] += 1
34
  save_user_session_data(all_user_data)
35
  return all_user_data[user_id]["test_counter"]
36
-
37
  # Function to get or create user ID from cookie
38
  def get_or_create_user_id(request: gr.Request):
39
  if request is None:
 
27
  save_user_session_data(all_user_data)
28
  return user_data["test_counter"]
29
 
30
+ def save_user_session_data(all_user_data):
31
+ with open(SESSION_FILE, 'w') as f:
32
+ json.dump(all_user_data, f)
33
+
34
+ # Function to load user session data
35
+ def load_user_session_data(user_id):
36
+ if os.path.exists(SESSION_FILE):
37
+ with open(SESSION_FILE, 'r') as f:
38
+ all_user_data = json.load(f)
39
+ else:
40
+ all_user_data = {}
41
+
42
+ if user_id not in all_user_data:
43
+ all_user_data[user_id] = {"last_reset_date": str(date.today()), "test_counter": 0}
44
+
45
+ return all_user_data
46
+
47
  # Function to increment the counter for a specific user
48
  def increment_user_counter(user_id):
49
  all_user_data = load_user_session_data(user_id)
50
  all_user_data[user_id]["test_counter"] += 1
51
  save_user_session_data(all_user_data)
52
  return all_user_data[user_id]["test_counter"]
53
+
54
  # Function to get or create user ID from cookie
55
  def get_or_create_user_id(request: gr.Request):
56
  if request is None: