Update app.py

app.py

@@ -148,9 +148,22 @@ async def shutdown():
     if http_client:
         await http_client.aclose()
 
-def get_domain() -> str:
+def get_domain_old() -> str:
     return os.environ.get("SPACE_HOST", "localhost").replace("https://", "").replace("http://", "")
 
+
+def get_domain() -> str:
+    import urllib.request
+    try:
+        # Attempt to get the public IPv4 address first (best for VPN/Proxy links)
+        public_ip = urllib.request.urlopen('https://api.ipify.org', timeout=3).read().decode('utf8')
+        return public_ip
+    except Exception:
+        # Fallback to the local interface IPv4 using your existing function
+        ipv4, _ = get_real_ips()
+        return ipv4 if ipv4 else "127.0.0.1"
+
+
 def generate_uuid(seed: str | None = None) -> str:
     # Real, standard RFC-4122 UUID
     if seed is None: