获取client real ip。

app.py

@@ -3,6 +3,7 @@
 import openai
 import html
 import os
+import socket
 
 import gradio as gr
 import random
@@ -21,6 +22,20 @@ MODELS = [
 
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
+
+def parse_request(request: gr.Request):
+    client_ip = request.client.host
+    local_ip = socket.gethostbyname(socket.gethostbyname(""))
+    headers = request.headers
+    if headers and 'x-forwarded-for' in headers:
+        x_forwarded_for = headers['x-forwarded-for']
+        client_ip = x_forwarded_for.split(' ')[0] if x_forwarded_for else ""
+
+    return {"client_ip": client_ip,
+            "local_ip": local_ip,
+            "headers": headers}
+
+
 def reduce_history(history):
     mid = len(history) // 2
     logging.info('history reduced')
@@ -28,7 +43,7 @@ def reduce_history(history):
 
 
 def chat_with_model(request, role, content, history=[], model=MODELS[0]):
-    client = request.client.host
+    client_ip = parse_request(request)["client_ip"]
     if role == 'system':
         history = []
     history.append({'role': role, 'content': content})
@@ -37,10 +52,11 @@ def chat_with_model(request, role, content, history=[], model=MODELS[0]):
         messages=history
     )
 
-    content=html.escape(response['choices'][0]['message']['content'])
+    content = html.escape(response['choices'][0]['message']['content'])
     total_tokens = response['usage']['total_tokens']
     history.append({'role': 'assistant', 'content': content})
-    logging.info('client={}, history count={}'.format(client, str(len(history))))
+    logging.info('client={}, history count={}'.format(
+        client_ip, str(len(history))))
     if 4096 - total_tokens < 500:
         history = reduce_history(history)
     return content, history
@@ -67,25 +83,30 @@ def launch():
             return user_message, chatbot + [[user_message, None]]
 
         def bot_resp(msg, chatbot, history, model, request: gr.Request):
-            bot_message, history = chat_with_model(request, 'user', msg, history, model)
+            bot_message, history = chat_with_model(
+                request, 'user', msg, history, model)
             chatbot[-1][1] = bot_message
             return '', chatbot, history
 
         def bot_change(selected_bot, history, model, request: gr.Request):
-            logging.info(f'client={request.client.host} Bot changed={selected_bot}')
+            client_ip = parse_request(request)["client_ip"]
+            logging.info(
+                f'client={client_ip} Bot changed={selected_bot}')
             if BOT_INIT_PROMPTS.get(selected_bot):
                 content, history = chat_with_model(request,
-                    'system', BOT_INIT_PROMPTS[selected_bot], history, model)
+                                                   'system', BOT_INIT_PROMPTS[selected_bot], history, model)
                 return [[None, content]], '', history
             else:
                 return [], '', []
 
         # Events
         msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-            bot_resp, [msg, chatbot, history, model_selector], [msg, chatbot, history]
+            bot_resp, [msg, chatbot, history, model_selector], [
+                msg, chatbot, history]
         )
         submit.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-            bot_resp, [msg, chatbot, history, model_selector], [msg, chatbot, history]
+            bot_resp, [msg, chatbot, history, model_selector], [
+                msg, chatbot, history]
         )
         clear.click(lambda: None, None, chatbot, queue=False).then(
             bot_change, [bot_selector, history, model_selector], [chatbot, msg, history])