Update app.py
Browse files
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:
|