Refactor vision client initialization and update Gradio launch configuration

agent/image_symptom_tool.py

@@ -1,10 +1,34 @@
-from langchain_core.tools import tool
-from gradio_client import Client
 import json
+import os
+from threading import Lock
+
+from gradio_client import Client
+from langchain_core.tools import tool
+
+VISION_SPACE_URL = os.getenv(
+    "VISION_SPACE_URL",
+    "https://datdevsteve-nivra-vision-diagnosis-v2.hf.space/",
+)
+
+_vision_client = None
+_vision_client_lock = Lock()
 
-# Instantiate client once (efficient & stable)
-VISION_SPACE_URL = "https://datdevsteve-nivra-vision-diagnosis-v2.hf.space/"
-vision_client = Client(VISION_SPACE_URL)
+
+def get_vision_client() -> Client:
+    """
+    Lazily initializes the remote Gradio client so import-time network
+    failures do not prevent the app from starting.
+    """
+    global _vision_client
+
+    if _vision_client is not None:
+        return _vision_client
+
+    with _vision_client_lock:
+        if _vision_client is None:
+            _vision_client = Client(VISION_SPACE_URL)
+
+    return _vision_client
 
 
 @tool
@@ -14,6 +38,8 @@ def analyze_symptom_image(img_url: str) -> str:
     Returns structured JSON string.
     """
     try:
+        vision_client = get_vision_client()
+
         result = vision_client.predict(
             image_upload=None,
             image_url=img_url,

app.py

@@ -93,9 +93,14 @@ with gr.Blocks(title="🩺 Nivra AI Agent") as demo:
 # Launch (HF-safe)
 # ==================================================
 
-demo.launch(
-    server_name="0.0.0.0",
-    server_port=7860,
-    ssr_mode=False,
-    show_error=True
-)
+launch_kwargs = {
+    "server_name": "0.0.0.0",
+    "ssr_mode": False,
+    "show_error": True,
+}
+
+configured_port = os.environ.get("GRADIO_SERVER_PORT") or os.environ.get("PORT")
+if configured_port:
+    launch_kwargs["server_port"] = int(configured_port)
+
+demo.launch(**launch_kwargs)