yangzhitao commited on
Commit
f10ff89
·
1 Parent(s): 61d4a7d

fix: refine user ID extraction logic in create_submit_tab function

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -596,8 +596,9 @@ def create_submit_tab(tab_id: int, demo: gr.Blocks):
596
  # Check if user is logged in
597
  if not current_user_display:
598
  return styled_error("Please log in before submitting.")
599
- reg = re.compile(r"\*\*(.+)\*\*")
600
- user_id_match = reg.match(current_user_display)
 
601
  if not user_id_match:
602
  msg = f'Unable to get user ID ("{current_user_display!s}"). Please try logging in again.'
603
  return styled_error(msg)
 
596
  # Check if user is logged in
597
  if not current_user_display:
598
  return styled_error("Please log in before submitting.")
599
+ # Extract user_id from "Submitter ID: **{user_id}**" format
600
+ reg = re.compile(r"\*\*(.+?)\*\*")
601
+ user_id_match = reg.search(current_user_display)
602
  if not user_id_match:
603
  msg = f'Unable to get user ID ("{current_user_display!s}"). Please try logging in again.'
604
  return styled_error(msg)