Enhance diagnose function to include TCP connection checks for ports 80 and 443

api.py

@@ -61,15 +61,27 @@ def _check_document(image, endpoint_key: str) -> dict:
 def diagnose() -> dict:
     # ponytail: one function shows where HF->server path breaks
     import socket
+    import time
     from urllib.parse import urlparse
     out = {"base_url": BASE_URL, "endpoints": ENDPOINTS}
+    try:
+        out["egress_ip"] = requests.get("https://api.ipify.org", timeout=10).text
+    except Exception as e:
+        out["egress_ip"] = f"FAIL: {type(e).__name__}: {e}"
     try:
         p = urlparse(BASE_URL)
         host, port = p.hostname, p.port or (443 if p.scheme == "https" else 80)
         out["dns"] = socket.gethostbyname(host)
-        s = socket.create_connection((host, port), timeout=10)
-        out["tcp_443"] = "ok"
-        s.close()
+        for probe_port, key in [(80, "tcp_80"), (443, "tcp_443")]:
+            try:
+                t0 = time.time()
+                s = socket.create_connection((host, probe_port), timeout=10)
+                out[key] = f"ok {time.time()-t0:.2f}s"
+                s.close()
+            except Exception as e:
+                out[key] = f"FAIL: {type(e).__name__}: {e}"
+        if out.get("tcp_443", "").startswith("FAIL"):
+            return out
     except Exception as e:
         out["tcp_443"] = f"FAIL: {type(e).__name__}: {e}"
         return out