Update app.py

app.py

@@ -3,23 +3,28 @@ import socket
 
 def predict(text, request: gr.Request):
     client_ip = request.client.host
-    local_ip = socket.gethostbyname(socket.gethostname())
+    local_ip = socket.gethostbyname(socket.gethostbyname(""))
+
     headers = request.headers
-    x_forwarded_for = headers['x-forwarded-for'] 
-    real_client_ip = x_forwarded_for.split(" ")[0]
+    real_client_ip = ""
+    if '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, 
-                  "local_ip": local_ip, 
+    return text, {"real_client_ip": real_client_ip,
+                  "client_ip": client_ip,
+                  "local_ip": local_ip,
                   "headers": headers}
 
+
 with gr.Blocks() as block:
     gr.Markdown("## Gradio get client IP")
     text = gr.Textbox(label="dummy input")
     output = gr.JSON({})
     btn = gr.Button("Test")
 
-    btn.click(predict, inputs=[text],outputs=[text, output])
+    btn.click(predict, inputs=[text], outputs=[text, output])
 
 
-block.launch(share=False, server_name='0.0.0.0',show_api=True)
+block.launch(share=False, server_name='0.0.0.0', show_api=True)