Update app.py

app.py

@@ -7,6 +7,7 @@ from datetime import datetime, date
 import os
 import json
 import uuid
+from http.cookies import SimpleCookie
 
 # File to store session data
 SESSION_FILE = "user_session_data.json"
@@ -15,24 +16,6 @@ SESSION_FILE = "user_session_data.json"
 api_key = os.getenv("OPENAI_API_KEY")
 
 
-# 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 save user session data
-def save_user_session_data(all_user_data):
-    with open(SESSION_FILE, 'w') as f:
-        json.dump(all_user_data, f)
-
 # Function to check and reset the counter if necessary for a specific user
 def check_and_reset_user_counter(user_id):
     all_user_data = load_user_session_data(user_id)
@@ -51,6 +34,25 @@ def increment_user_counter(user_id):
     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:
+        return str(uuid.uuid4())
+    
+    cookies = SimpleCookie()
+    cookies.load(request.headers.get('cookie', ''))
+    user_id = cookies.get('user_id')
+    
+    if user_id is None:
+        user_id = str(uuid.uuid4())
+        cookies['user_id'] = user_id
+        cookies['user_id']['httponly'] = True
+        cookies['user_id']['max-age'] = 365 * 24 * 60 * 60  # 1 year
+    else:
+        user_id = user_id.value
+    
+    return user_id, cookies.output(header='', sep=';')
+
 # Function to encode the image
 def encode_image(image_array):
     # Convert numpy array to an image file and encode it in base64
@@ -213,7 +215,6 @@ with gr.Blocks(css=css) as demo:
                 redirect_message = gr.Markdown(visible=False, elem_classes="redirect-message")
                 test_counter_display = gr.Markdown(visible=True, label="Test Counter")
 
-
         # Toggle visibility of custom instruction based on selected type
         def toggle_custom_instruction(type_selection):
             return gr.update(visible=(type_selection == "Other"))
@@ -224,9 +225,9 @@ with gr.Blocks(css=css) as demo:
             outputs=[custom_instruction],
         )
 
-        # Modified function to handle test limit with improved messaging
-        def handle_submit(image, description_type, custom_instruction):
-            global user_id
+        # Modified function to handle test limit with improved messaging and cookie-based user sessions
+        def handle_submit(image, description_type, custom_instruction, request: gr.Request):
+            user_id, cookie = get_or_create_user_id(request)
             check_and_reset_user_counter(user_id)  # Check and reset counter if it's a new day
             test_counter = increment_user_counter(user_id)
             if test_counter <= 4:
@@ -238,6 +239,7 @@ with gr.Blocks(css=css) as demo:
                         gr.update(visible=False),
                         gr.update(interactive=True),
                         f"You have {remaining_tests} free test(s) remaining today.",
+                        cookie,
                     )
                 else:
                     redirect_text = """
@@ -263,6 +265,7 @@ with gr.Blocks(css=css) as demo:
                         gr.update(visible=True, value=redirect_text),
                         gr.update(interactive=False),
                         "You've used all your free tests for today. We invite you to book a demo for full access!",
+                        cookie,
                     )
             else:
                 redirect_text = """
@@ -288,6 +291,7 @@ with gr.Blocks(css=css) as demo:
                     gr.update(visible=True, value=redirect_text),
                     gr.update(interactive=False),
                     "All free tests used. We invite you to book a demo for full access!",
+                    cookie,
                 )
 
         submit_btn.click(