Add connection diagnostics feature and improve error handling in API calls

api.py

@@ -6,7 +6,10 @@ import numpy as np
 BASE_URL = os.environ.get(
     "ID_API_URL",
     "http://127.0.0.1:8082"
-)
+).rstrip("/")
+# ponytail: tolerate /api suffix (website env has it, this code adds it)
+if BASE_URL.endswith("/api"):
+    BASE_URL = BASE_URL[:-len("/api")]
 
 ENDPOINTS = {
     "id": f"{BASE_URL}/api/check_id",
@@ -47,14 +50,42 @@ def _check_document(image, endpoint_key: str) -> dict:
         )
         response.raise_for_status()
         return response.json()
-    except requests.exceptions.ConnectionError:
-        return {"error": f"Cannot connect to server at {BASE_URL}. Make sure the SDK server is running."}
+    except requests.exceptions.ConnectionError as e:
+        return {"error": f"Cannot connect to server at {BASE_URL}: {e}. Make sure the SDK server is running."}
     except requests.exceptions.Timeout:
         return {"error": "Request timed out"}
     except requests.exceptions.RequestException as e:
         return {"error": f"API request failed: {str(e)}"}
 
 
+def diagnose() -> dict:
+    # ponytail: one function shows where HF->server path breaks
+    import socket
+    from urllib.parse import urlparse
+    out = {"base_url": BASE_URL, "endpoints": ENDPOINTS}
+    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()
+    except Exception as e:
+        out["tcp_443"] = f"FAIL: {type(e).__name__}: {e}"
+        return out
+    try:
+        r = requests.get(BASE_URL, timeout=10)
+        out["https_get_root"] = f"{r.status_code}: {r.text[:120]}"
+    except Exception as e:
+        out["https_get_root"] = f"FAIL: {type(e).__name__}: {e}"
+    try:
+        r = requests.post(ENDPOINTS["id"], files={"image": ("t.jpg", b"not-an-image", "image/jpeg")}, timeout=15)
+        out["https_post_check_id"] = f"{r.status_code}: {r.text[:200]}"
+    except Exception as e:
+        out["https_post_check_id"] = f"FAIL: {type(e).__name__}: {e}"
+    return out
+
+
 def check_id(image) -> dict:
     return _check_document(image, "id")
 

ui.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from api import check_id, check_credit, check_mrz
+from api import check_id, check_credit, check_mrz, diagnose
 
 ID_EXAMPLES = [[f"images/id/{i}.jpg"] for i in range(1, 8)]
 CREDIT_EXAMPLES = [["images/bank/demo1.jpg"], ["images/bank/demo2.png"], ["images/bank/demo3.png"]]
@@ -127,4 +127,9 @@ def create_interface():
             outputs=[result_html, raw_json],
         )
 
+        with gr.Accordion("Connection diagnostics", open=False):
+            diag_btn = gr.Button("Test connection")
+            diag_json = gr.JSON(label="Diagnostics")
+            diag_btn.click(fn=diagnose, outputs=diag_json)
+
     return demo