add more

app.py

@@ -73,6 +73,13 @@ count_client_map = ExpiringMap()
 def infer(prompt: str, negative: str, width: int, height: int, sampler: str,
           steps: int, seed: int, scale, request: gr.Request):
     client_ip = request.client.host
+    headers = request.kwargs['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 ""
+
+    print("client_ip", client_ip, text, "\n\n")
+
     if client_ip != '127.0.0.1' and client_ip != 'localhost' and client_ip != '0.0.0.0':
         if time_client_map.get(client_ip):
             return None, "Too many requests, please try again later."