chengzeyi commited on
Commit
d81f080
·
1 Parent(s): f361085

fix client_ip

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -532,6 +532,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
532
  current_session_id,
533
  request: gr.Request):
534
  client_ip = request.client.host
 
 
 
535
  print(f"Client IP: {client_ip}")
536
  client_generation_manager = ClientManager.get_manager(client_ip)
537
  client_generation_manager.update_activity()
 
532
  current_session_id,
533
  request: gr.Request):
534
  client_ip = request.client.host
535
+ x_forwarded_for = request.headers.get('x-forwarded-for')
536
+ if x_forwarded_for:
537
+ client_ip = x_forwarded_for
538
  print(f"Client IP: {client_ip}")
539
  client_generation_manager = ClientManager.get_manager(client_ip)
540
  client_generation_manager.update_activity()