Update app.py

app.py

@@ -52,19 +52,10 @@ def increment_user_counter(user_id):
     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()), None
-    
-    cookies = SimpleCookie()
-    cookies.load(request.headers.get('cookie', ''))
-    user_id = cookies.get('user_id')
-    
-    if user_id is None:
-        user_id = str(uuid.uuid4())
-        return user_id, ('user_id', user_id, {'httponly': True, 'max_age': 365 * 24 * 60 * 60})
-    else:
-        return user_id.value, None
+def get_or_create_user_id(user_id):
+    if not user_id:
+        return str(uuid.uuid4())
+    return user_id
 
 # Function to encode the image
 def encode_image(image_array):
@@ -176,10 +167,37 @@ css = """
   }
 """
 
+custom_js = """
+function getUserId() {
+    let userId = localStorage.getItem('wordlift_user_id');
+    if (!userId) {
+        userId = generateUUID();
+        localStorage.setItem('wordlift_user_id', userId);
+    }
+    return userId;
+}
+
+function generateUUID() {
+    return 'xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx'.replace(/[xy]/g, function(c) {
+        var r = Math.random() * 16 | 0, v = c == 'x' ? r : (r & 0x3 | 0x8);
+        return v.toString(16);
+    });
+}
+
+function setUserId(userId) {
+    document.getElementById('user-id-input').value = userId;
+}
+
+// Set user ID when page loads
+document.addEventListener('DOMContentLoaded', (event) => {
+    setUserId(getUserId());
+});
+"""
+
 # Initialize user ID
 user_id = str(uuid.uuid4())
 
-with gr.Blocks(css=css) as demo:
+with gr.Blocks(css=custom_css, js=custom_js) as demo:
     gr.Markdown("<h1>WordLift Product Description Generation - [FREE]</h1>")
     with gr.Tab(label="WordLift Product Description Generation"):
         with gr.Row():
@@ -228,6 +246,10 @@ 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")
                 user_id_display = gr.Markdown(visible=True, label="User ID")
+                
+        # Hidden textbox to store user ID
+        user_id_input = gr.Textbox(elem_id="user-id-input", visible=False)
+
 
         # Toggle visibility of custom instruction based on selected type
         def toggle_custom_instruction(type_selection):
@@ -239,8 +261,8 @@ with gr.Blocks(css=css) as demo:
             outputs=[custom_instruction],
         )
 
-        def handle_submit(image, description_type, custom_instruction, request: gr.Request):
-            user_id, cookie = get_or_create_user_id(request)
+        def handle_submit(image, description_type, custom_instruction, user_id):
+            user_id = get_or_create_user_id(user_id)
             check_and_reset_user_counter(user_id)  # Check and reset counter if it's a new day
             test_counter = increment_user_counter(user_id)
             
@@ -256,7 +278,8 @@ 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.",
-                        f"User ID: {user_id}"
+                        f"User ID: {user_id}",
+                        user_id  # Return user_id to update hidden textbox
                     )
                 else:
                     redirect_text = f"""
@@ -282,7 +305,8 @@ 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!",
-                        f"User ID: {user_id}"
+                        f"User ID: {user_id}",
+                        user_id  # Return user_id to update hidden textbox
                     )
             else:
                 redirect_text = f"""
@@ -308,15 +332,17 @@ 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!",
-                    f"User ID: {user_id}"
+                    f"User ID: {user_id}",
+                    user_id  # Return user_id to update hidden textbox
                 )
 
 
+
         # Update the outputs of the submit button click event
         submit_btn.click(
             handle_submit,
-            inputs=[input_img, description_type, custom_instruction],
-            outputs=[output_text, redirect_message, submit_btn, test_counter_display, user_id_display]
+            inputs=[input_img, description_type, custom_instruction, user_id_input],
+            outputs=[output_text, redirect_message, submit_btn, test_counter_display, user_id_display, user_id_input]
         )
     
 # Launch Gradio app