siyuwang541 commited on
Commit
8b24e49
·
verified ·
1 Parent(s): 60c193e
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -3,7 +3,7 @@ from huggingface_hub import InferenceClient
3
 
4
  client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
5
 
6
- def get_client_ip(request: gr.Request):
7
  """获取客户端真实IP地址"""
8
  if request:
9
  # 从请求头中获取真实IP(考虑代理情况)
@@ -12,12 +12,14 @@ def get_client_ip(request: gr.Request):
12
  client_ip = x_forwarded_for.split(",")[0]
13
  else:
14
  client_ip = request.client.host
 
 
15
  return client_ip
16
  return "unknown"
17
 
18
  def process(audio, image, request: gr.Request):
19
  """处理语音和图片的示例函数"""
20
- client_ip = get_client_ip(request)
21
  print(f"Processing request from IP: {client_ip}")
22
 
23
  if audio is not None:
@@ -45,7 +47,7 @@ def respond(
45
  request: gr.Request # 添加请求对象
46
  ):
47
  # 获取客户端IP
48
- client_ip = get_client_ip(request)
49
  print(f"Chat request from IP: {client_ip}")
50
 
51
  # 如果有上传的音频或图片,添加到消息中
 
3
 
4
  client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
5
 
6
+ def get_client_ip(request: gr.Request, debug_mode=False):
7
  """获取客户端真实IP地址"""
8
  if request:
9
  # 从请求头中获取真实IP(考虑代理情况)
 
12
  client_ip = x_forwarded_for.split(",")[0]
13
  else:
14
  client_ip = request.client.host
15
+ if(debug_mode):
16
+ print(f"Debug: Client IP detected as {client_ip}")
17
  return client_ip
18
  return "unknown"
19
 
20
  def process(audio, image, request: gr.Request):
21
  """处理语音和图片的示例函数"""
22
+ client_ip = get_client_ip(request, True)
23
  print(f"Processing request from IP: {client_ip}")
24
 
25
  if audio is not None:
 
47
  request: gr.Request # 添加请求对象
48
  ):
49
  # 获取客户端IP
50
+ client_ip = get_client_ip(request, True)
51
  print(f"Chat request from IP: {client_ip}")
52
 
53
  # 如果有上传的音频或图片,添加到消息中