Update nivra_agent.py

nivra_agent.py

@@ -162,36 +162,40 @@ Provide diagnosis:
 # ==================================================
 
 def nivra_vision(image_url: str, hint_text: str = "") -> str:
-    try:
-        tool_output = analyze_symptom_image.invoke({
-            "image_url": image_url,
-            "image_description": hint_text,
-        })
-    except Exception as e:
+    """
+    Image-based medical analysis using vision tool + Groq reasoning
+    """
+
+    print("🖼️ nivra_vision called")
+
+    tool_output = analyze_symptom_image.invoke({
+        "image_url": image_url,
+        "image_description": hint_text
+    })
+
+    # 🚨 If vision tool already failed, DO NOT call Groq
+    if "ERROR" in tool_output.upper():
         return f"""
 [TOOLS USED] analyze_symptom_image [/TOOLS USED]
-[SYMPTOMS] Image could not be analyzed [/SYMPTOMS]
+[SYMPTOMS] Unable to analyze image [/SYMPTOMS]
 [PRIMARY DIAGNOSIS] Inconclusive [/PRIMARY DIAGNOSIS]
 [DIAGNOSIS DESCRIPTION]
-Image analysis failed due to a technical issue.
+{tool_output}
 [/DIAGNOSIS DESCRIPTION]
-[FIRST AID] Consult a medical professional [/FIRST AID]
+[FIRST AID] Please consult a medical professional [/FIRST AID]
 [EMERGENCY CONSULTATION REQUIRED] No [/EMERGENCY CONSULTATION REQUIRED]
-"""
+""".strip()
 
+    # Keep Groq prompt SMALL
     final_prompt = f"""
-You are a medical AI assistant.
-
-Vision tool result:
+VISION TOOL OUTPUT:
 {tool_output}
 
-User context:
+USER CONTEXT:
 {hint_text}
 
-Respond STRICTLY in the medical format.
-Keep description under 3 sentences.
-"""
-
+Provide a cautious preliminary medical assessment.
+""".strip()
 
     return call_groq(final_prompt)