Update app.py

app.py

@@ -4,15 +4,12 @@ import socket
 def predict(text, request: gr.Request):
     client_ip = request.client.host
     local_ip = socket.gethostbyname(socket.gethostbyname(""))
-
-    headers = request.headers
-    real_client_ip = ""
-    print(headers)
-    # print(dict(headers))
-    # if headers and 'x-forwarded-for' in headers:
-    #     x_forwarded_for = headers['x-forwarded-for']
-    #     real_client_ip = x_forwarded_for.split(
-    #         ' ')[0] if x_forwarded_for else ""
+    headers = request.kwargs['headers']
+    
+    if headers and 'x-forwarded-for' in headers:
+        x_forwarded_for = headers['x-forwarded-for']
+        real_client_ip = x_forwarded_for.split(
+            ' ')[0] if x_forwarded_for else ""
 
     return text, {"real_client_ip": real_client_ip,
                   "client_ip": client_ip,