Update app.py

app.py

@@ -167,37 +167,10 @@ custom_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=custom_css, js=custom_js) as demo:
+with gr.Blocks(css=custom_css) as demo:
     gr.Markdown("<h1>WordLift Product Description Generation - [FREE]</h1>")
     with gr.Tab(label="WordLift Product Description Generation"):
         with gr.Row():
@@ -248,7 +221,7 @@ with gr.Blocks(css=custom_css, js=custom_js) as demo:
                 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)
+        user_id_input = gr.Textbox(value="", visible=False)
 
 
         # Toggle visibility of custom instruction based on selected type
@@ -261,8 +234,11 @@ with gr.Blocks(css=custom_css, js=custom_js) as demo:
             outputs=[custom_instruction],
         )
 
+        # Modify the handle_submit function
         def handle_submit(image, description_type, custom_instruction, user_id):
-            user_id = get_or_create_user_id(user_id)
+            if not user_id:
+                user_id = str(uuid.uuid4())
+            
             check_and_reset_user_counter(user_id)  # Check and reset counter if it's a new day
             test_counter = increment_user_counter(user_id)
             
@@ -279,7 +255,7 @@ with gr.Blocks(css=custom_css, js=custom_js) as demo:
                         gr.update(interactive=True),
                         f"You have {remaining_tests} free test(s) remaining today.",
                         f"User ID: {user_id}",
-                        user_id  # Return user_id to update hidden textbox
+                        user_id
                     )
                 else:
                     redirect_text = f"""
@@ -306,7 +282,7 @@ with gr.Blocks(css=custom_css, js=custom_js) as demo:
                         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}",
-                        user_id  # Return user_id to update hidden textbox
+                        user_id
                     )
             else:
                 redirect_text = f"""
@@ -333,11 +309,10 @@ with gr.Blocks(css=custom_css, js=custom_js) as demo:
                     gr.update(interactive=False),
                     "All free tests used. We invite you to book a demo for full access!",
                     f"User ID: {user_id}",
-                    user_id  # Return user_id to update hidden textbox
+                    user_id
                 )
 
 
-
         # Update the outputs of the submit button click event
         submit_btn.click(
             handle_submit,