Update agent/image_symptom_tool.py

agent/image_symptom_tool.py

@@ -1,24 +1,25 @@
 from langchain_core.tools import tool
 from gradio_client import Client
+import json
 
 @tool
-def analyze_symptom_image(image_url: str) -> str:
+def analyze_symptom_image(img_url: str) -> str:
     """
     Calls Nivra Vision Diagnosis Gradio Space using image URL
     """
     try:
-        client = Client("datdevsteve/nivra-vision-diagnosis")
-
-        result = client.predict(
-            image=image_url,
-            api_name="/predict_gradio"
+        result = Client.predict(
+            image_upload=None,  # Explicitly None
+            image_url=img_url,
+            top_k=2,
+            api_name="/predict"
         )
-
+        md_output = result[0]
+        result_json = json.loads(result[1])
         return f"""
 [SYMPTOM IMAGE ANALYSIS - SUCCESS]
-{result}
+{result_json}
 """
-
     except Exception as e:
         return f"""
 [SYMPTOM IMAGE ANALYSIS - ERROR]