lunarflu HF staff commited on
Commit
ac9050d
·
verified ·
1 Parent(s): bf6a2a7
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -37,7 +37,8 @@ threading.Thread(target=run_bot).start()
37
 
38
  # Gradio ------------------------------------------------------------------------------------------------------------
39
  def validate_link(request: gr.Request) -> bool:
40
- url_str = str(request.url)
 
41
  query_params = parse_qs(urlparse(url_str).query)
42
  user_id = query_params.get('user_id', [None])[0]
43
  token = query_params.get('token', [None])[0]
@@ -54,7 +55,8 @@ def hello(profile: gr.OAuthProfile | None, request: gr.Request) -> str:
54
  if not validate_link(request):
55
  return "❌ Invalid link. Please use the link provided by the bot."
56
 
57
- url_str = str(request.url)
 
58
  query_params = parse_qs(urlparse(url_str).query)
59
  user_id = query_params.get('user_id', [None])[0]
60
 
 
37
 
38
  # Gradio ------------------------------------------------------------------------------------------------------------
39
  def validate_link(request: gr.Request) -> bool:
40
+ referer = request.headers.get('Referer', '')
41
+ url_str = str(referer)
42
  query_params = parse_qs(urlparse(url_str).query)
43
  user_id = query_params.get('user_id', [None])[0]
44
  token = query_params.get('token', [None])[0]
 
55
  if not validate_link(request):
56
  return "❌ Invalid link. Please use the link provided by the bot."
57
 
58
+ referer = request.headers.get('Referer', '')
59
+ url_str = str(referer)
60
  query_params = parse_qs(urlparse(url_str).query)
61
  user_id = query_params.get('user_id', [None])[0]
62