abidlabs HF Staff commited on
Commit
16e662f
·
1 Parent(s): 1d79dcc
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -36,12 +36,13 @@ def get_user(request: Request):
36
 
37
  @app.get('/')
38
  def public(request: Request, user: dict = Depends(get_user)):
39
- host = str(request.url).rstrip("/").replace("http://", "https://")
40
- print(">>>>>>>", host)
 
41
  if user:
42
- return RedirectResponse(url=f"{host}/gradio")
43
  else:
44
- return RedirectResponse(url=f"{host}/login-demo")
45
 
46
  @app.route('/logout')
47
  async def logout(request: Request):
 
36
 
37
  @app.get('/')
38
  def public(request: Request, user: dict = Depends(get_user)):
39
+ root_url = gr.route_utils.get_root_url(request, "/", None)
40
+ root_url = root_url.replace("http://", "https://")
41
+ print(">>>>>>>", root_url)
42
  if user:
43
+ return RedirectResponse(url=f"{root_url}/gradio")
44
  else:
45
+ return RedirectResponse(url=f"{root_url}/login-demo")
46
 
47
  @app.route('/logout')
48
  async def logout(request: Request):