Update app.py

app.py

@@ -360,8 +360,9 @@ def edit_3d_model(edit_prompt: str, transparent_image_path: str) -> str:
 def generate_3d_ui(prompt, request: gr.Request, progress=gr.Progress()):
     """UI wrapper with progress updates"""
     
-    # Get client info
-    client_ip = request.client.host if request else "unknown"
+    client_ip = request.headers.get("x-forwarded-for", "").split(",")[0].strip()
+    if not client_ip:
+        client_ip = request.client.host if request else "unknown"
     location = get_location_from_ip(client_ip)
     user_agent = request.headers.get("user-agent", "unknown") if request else "unknown"