Update agent/image_symptom_tool.py

agent/image_symptom_tool.py

@@ -1,64 +1,26 @@
-# agent/image_symptom_tool.py
-
 from langchain_core.tools import tool
-import requests
-import base64
-import io
-from PIL import Image
+from gradio_client import Client
 
 @tool
-def analyze_symptom_image(image_url: str, image_description: str = "") -> str:
+def analyze_symptom_image(image_url: str) -> str:
     """
-    Analyze a medical image via Vision HF Space.
-
-    Args:
-        image_url (str): PUBLICLY accessible image URL
-        image_description (str): Optional user hint
-
-    Returns:
-        str: Structured vision analysis text
+    Calls Nivra Vision Diagnosis Gradio Space using image URL
     """
-
     try:
-        print(f"📥 Fetching image from URL: {image_url}")
-
-        resp = requests.get(image_url, timeout=20)
-        resp.raise_for_status()
-
-        image = Image.open(io.BytesIO(resp.content)).convert("RGB")
-
-        # Encode to base64 ONLY for vision backend
-        buf = io.BytesIO()
-        image.save(buf, format="PNG")
-        img_base64 = base64.b64encode(buf.getvalue()).decode("utf-8")
+        client = Client("datdevsteve/nivra-vision-diagnosis")
 
-        vision_api = "https://datdevsteve-nivra-vision-diagnosis.hf.space/run/predict"
-
-        payload = {
-            "data": [img_base64]
-        }
-
-        print("🔬 Calling Vision HF Space")
-        vision_resp = requests.post(vision_api, json=payload, timeout=60)
-        vision_resp.raise_for_status()
-
-        result = vision_resp.json()
-
-        if "data" not in result or not result["data"]:
-            return "[VISION ERROR] No prediction returned."
-
-        diagnosis = result["data"][0]
+        result = client.predict(
+            image=image_url,
+            api_name="/predict"
+        )
 
         return f"""
-[SYMPTOM IMAGE ANALYSIS]
-{diagnosis}
-
-[USER IMAGE DESCRIPTION]
-{image_description or "Not provided"}
-""".strip()
+[SYMPTOM IMAGE ANALYSIS - SUCCESS]
+{result}
+"""
 
     except Exception as e:
         return f"""
-[SYMPTOM IMAGE ANALYSIS ERROR]
+[SYMPTOM IMAGE ANALYSIS - ERROR]
 {str(e)}
-""".strip()
+"""