cyberandy commited on
Commit
5617f65
·
verified ·
1 Parent(s): e0fc021

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -34
app.py CHANGED
@@ -167,37 +167,10 @@ custom_css = """
167
  }
168
  """
169
 
170
- custom_js = """
171
- function getUserId() {
172
- let userId = localStorage.getItem('wordlift_user_id');
173
- if (!userId) {
174
- userId = generateUUID();
175
- localStorage.setItem('wordlift_user_id', userId);
176
- }
177
- return userId;
178
- }
179
-
180
- function generateUUID() {
181
- return 'xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx'.replace(/[xy]/g, function(c) {
182
- var r = Math.random() * 16 | 0, v = c == 'x' ? r : (r & 0x3 | 0x8);
183
- return v.toString(16);
184
- });
185
- }
186
-
187
- function setUserId(userId) {
188
- document.getElementById('user-id-input').value = userId;
189
- }
190
-
191
- // Set user ID when page loads
192
- document.addEventListener('DOMContentLoaded', (event) => {
193
- setUserId(getUserId());
194
- });
195
- """
196
-
197
  # Initialize user ID
198
  user_id = str(uuid.uuid4())
199
 
200
- with gr.Blocks(css=custom_css, js=custom_js) as demo:
201
  gr.Markdown("<h1>WordLift Product Description Generation - [FREE]</h1>")
202
  with gr.Tab(label="WordLift Product Description Generation"):
203
  with gr.Row():
@@ -248,7 +221,7 @@ with gr.Blocks(css=custom_css, js=custom_js) as demo:
248
  user_id_display = gr.Markdown(visible=True, label="User ID")
249
 
250
  # Hidden textbox to store user ID
251
- user_id_input = gr.Textbox(elem_id="user-id-input", visible=False)
252
 
253
 
254
  # Toggle visibility of custom instruction based on selected type
@@ -261,8 +234,11 @@ with gr.Blocks(css=custom_css, js=custom_js) as demo:
261
  outputs=[custom_instruction],
262
  )
263
 
 
264
  def handle_submit(image, description_type, custom_instruction, user_id):
265
- user_id = get_or_create_user_id(user_id)
 
 
266
  check_and_reset_user_counter(user_id) # Check and reset counter if it's a new day
267
  test_counter = increment_user_counter(user_id)
268
 
@@ -279,7 +255,7 @@ with gr.Blocks(css=custom_css, js=custom_js) as demo:
279
  gr.update(interactive=True),
280
  f"You have {remaining_tests} free test(s) remaining today.",
281
  f"User ID: {user_id}",
282
- user_id # Return user_id to update hidden textbox
283
  )
284
  else:
285
  redirect_text = f"""
@@ -306,7 +282,7 @@ with gr.Blocks(css=custom_css, js=custom_js) as demo:
306
  gr.update(interactive=False),
307
  "You've used all your free tests for today. We invite you to book a demo for full access!",
308
  f"User ID: {user_id}",
309
- user_id # Return user_id to update hidden textbox
310
  )
311
  else:
312
  redirect_text = f"""
@@ -333,11 +309,10 @@ with gr.Blocks(css=custom_css, js=custom_js) as demo:
333
  gr.update(interactive=False),
334
  "All free tests used. We invite you to book a demo for full access!",
335
  f"User ID: {user_id}",
336
- user_id # Return user_id to update hidden textbox
337
  )
338
 
339
 
340
-
341
  # Update the outputs of the submit button click event
342
  submit_btn.click(
343
  handle_submit,
 
167
  }
168
  """
169
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
170
  # Initialize user ID
171
  user_id = str(uuid.uuid4())
172
 
173
+ with gr.Blocks(css=custom_css) as demo:
174
  gr.Markdown("<h1>WordLift Product Description Generation - [FREE]</h1>")
175
  with gr.Tab(label="WordLift Product Description Generation"):
176
  with gr.Row():
 
221
  user_id_display = gr.Markdown(visible=True, label="User ID")
222
 
223
  # Hidden textbox to store user ID
224
+ user_id_input = gr.Textbox(value="", visible=False)
225
 
226
 
227
  # Toggle visibility of custom instruction based on selected type
 
234
  outputs=[custom_instruction],
235
  )
236
 
237
+ # Modify the handle_submit function
238
  def handle_submit(image, description_type, custom_instruction, user_id):
239
+ if not user_id:
240
+ user_id = str(uuid.uuid4())
241
+
242
  check_and_reset_user_counter(user_id) # Check and reset counter if it's a new day
243
  test_counter = increment_user_counter(user_id)
244
 
 
255
  gr.update(interactive=True),
256
  f"You have {remaining_tests} free test(s) remaining today.",
257
  f"User ID: {user_id}",
258
+ user_id
259
  )
260
  else:
261
  redirect_text = f"""
 
282
  gr.update(interactive=False),
283
  "You've used all your free tests for today. We invite you to book a demo for full access!",
284
  f"User ID: {user_id}",
285
+ user_id
286
  )
287
  else:
288
  redirect_text = f"""
 
309
  gr.update(interactive=False),
310
  "All free tests used. We invite you to book a demo for full access!",
311
  f"User ID: {user_id}",
312
+ user_id
313
  )
314
 
315
 
 
316
  # Update the outputs of the submit button click event
317
  submit_btn.click(
318
  handle_submit,