track user ip as user_id

app.py

@@ -3,8 +3,23 @@ from huggingface_hub import InferenceClient
 
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-def process(audio, image):
+def get_client_ip(request: gr.Request):
+    """获取客户端真实IP地址"""
+    if request:
+        # 从请求头中获取真实IP（考虑代理情况）
+        x_forwarded_for = request.headers.get("x-forwarded-for", "")
+        if x_forwarded_for:
+            client_ip = x_forwarded_for.split(",")[0]
+        else:
+            client_ip = request.client.host
+        return client_ip
+    return "unknown"
+
+def process(audio, image, request: gr.Request):
     """处理语音和图片的示例函数"""
+    client_ip = get_client_ip(request)
+    print(f"Processing request from IP: {client_ip}")
+    
     if audio is not None:
         sample_rate, audio_data = audio
         audio_info = f"音频采样率: {sample_rate}Hz, 数据长度: {len(audio_data)}"
@@ -26,19 +41,25 @@ def respond(
     temperature,
     top_p,
     audio,
-    image
+    image,
+    request: gr.Request  # 添加请求对象
 ):
+    # 获取客户端IP
+    client_ip = get_client_ip(request)
+    print(f"Chat request from IP: {client_ip}")
+    
     # 如果有上传的音频或图片，添加到消息中
     if audio is not None:
-        # 这里可以添加音频处理逻辑
         audio_sample_rate, audio_data = audio
         message += f"\n[附加音频信息: 采样率 {audio_sample_rate}Hz, 时长 {len(audio_data)/audio_sample_rate:.2f}秒]"
     
     if image is not None:
-        # 这里可以添加图片处理逻辑
         message += f"\n[附加图片信息: 尺寸 {image.shape}]"
     
-    messages = [{"role": "system", "content": system_message}]
+    # 可选：将IP信息添加到系统消息中
+    annotated_system_message = f"{system_message}\n[用户IP: {client_ip}]"
+    
+    messages = [{"role": "system", "content": annotated_system_message}]
 
     for val in history:
         if val[0]:
@@ -92,7 +113,7 @@ with gr.Blocks() as app:
         def user(user_message, chat_history):
             return "", chat_history + [[user_message, None]]
         
-        def bot(chat_history, system_message, max_tokens, temperature, top_p, audio, image):
+        def bot(chat_history, system_message, max_tokens, temperature, top_p, audio, image, request: gr.Request):
             # 获取最后一条用户消息
             user_message = chat_history[-1][0]
             
@@ -106,19 +127,20 @@ with gr.Blocks() as app:
                 temperature,
                 top_p,
                 audio,
-                image
+                image,
+                request  # 传递请求对象
             ):
                 bot_response = response
                 chat_history[-1][1] = bot_response
                 yield chat_history
             
-        # 连接事件
+        # 连接事件 - 添加request参数
         msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-            bot, [chatbot, system_msg, max_tokens, temperature, top_p, audio_input, image_input], chatbot
+            bot, [chatbot, system_msg, max_tokens, temperature, top_p, audio_input, image_input, gr.Request()], chatbot
         )
         
         submit_btn.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-            bot, [chatbot, system_msg, max_tokens, temperature, top_p, audio_input, image_input], chatbot
+            bot, [chatbot, system_msg, max_tokens, temperature, top_p, audio_input, image_input, gr.Request()], chatbot
         )
         
         clear.click(lambda: None, None, chatbot, queue=False)
@@ -131,11 +153,12 @@ with gr.Blocks() as app:
         audio_output = gr.Textbox(label="音频信息")
         image_output = gr.Textbox(label="图片信息")
         
+        # 添加request参数到处理函数
         process_btn.click(
             process,
-            inputs=[audio_processor, image_processor],
+            inputs=[audio_processor, image_processor, gr.Request()],
             outputs=[audio_output, image_output]
         )
 
 if __name__ == "__main__":
-    app.launch()
+    app.launch()