Update nivra_agent.py

nivra_agent.py

@@ -3,6 +3,7 @@
 #=========================================
 
 import os
+import base64
 import requests
 from dotenv import load_dotenv
 
@@ -13,24 +14,22 @@ from agent.image_symptom_tool import analyze_symptom_image
 load_dotenv()
 
 # ==================================================
-# Lazy singleton for RAG (HF-safe)
+# Lazy singleton for RAG
 # ==================================================
 
 _rag = None
 
-
 def get_rag():
     global _rag
     if _rag is None:
         _rag = NivraRAGRetriever()
     return _rag
 
-
 # ==================================================
 # SYSTEM PROMPT
 # ==================================================
 
-SYSTEM_PROMPT = """You are Nivra, a smart and helpful AI Healthcare Assistant with multimodal capabilities.
+YSTEM_PROMPT = """You are Nivra, a smart and helpful AI Healthcare Assistant with multimodal capabilities.
 
 🧠 **INTELLIGENT ROUTING RULES** (CRITICAL - Read First):
 1. **IF USER DESCRIBES PERSONAL SYMPTOMS** → Use structured medical format
@@ -98,10 +97,8 @@ Tuberculosis (TB) is caused by Mycobacterium tuberculosis bacteria, spread throu
 
 **FINAL CHECK**: Does user describe PERSONAL symptoms? YES=Medical format with respective token wrapping, NO=Natural response with respective token wrapping."""
 
-
-
 # ==================================================
-# GROQ RAW HTTP (HF SAFE)
+# GROQ CALL
 # ==================================================
 
 def call_groq(prompt: str) -> str:
@@ -110,7 +107,7 @@ def call_groq(prompt: str) -> str:
         return "⚠️ GROQ_API_KEY not configured."
 
     session = requests.Session()
-    session.trust_env = False  # HF Spaces proxy killer
+    session.trust_env = False
 
     response = session.post(
         "https://api.groq.com/openai/v1/chat/completions",
@@ -130,54 +127,23 @@ def call_groq(prompt: str) -> str:
     )
 
     if response.status_code != 200:
-        raise RuntimeError(
-            f"Groq API error {response.status_code}: {response.text}"
-        )
+        raise RuntimeError(f"Groq error {response.status_code}: {response.text}")
 
     return response.json()["choices"][0]["message"]["content"].strip()
 
 # ==================================================
-# MAIN CHAT FUNCTION
+# TEXT CHAT (UNCHANGED)
 # ==================================================
 
 def nivra_chat(user_input, chat_history=None):
-
-    if isinstance(user_input, dict):
-        user_input = user_input.get("text") or user_input.get("message", "")
     user_input = str(user_input).strip()
-
-    input_lower = user_input.lower()
-    text_keywords = ["fever", "headache", "cough", "pain", "vomiting", "chills"]
-
     tool_results = []
 
-    # -------------------------
-    # Text symptom tool
-    # -------------------------
-    if any(k in input_lower for k in text_keywords):
-        try:
-            tool_results.append(analyze_symptom_text.invoke(user_input))
-        except Exception:
-            tool_results.append("TEXT TOOL FAILED")
-
-    # -------------------------
-    # RAG retrieval
-    # -------------------------
     try:
+        tool_results.append(analyze_symptom_text.invoke(user_input))
         tool_results.append(get_rag().getRelevantDocs(user_input))
     except Exception:
-        tool_results.append("RAG FAILED")
-
-    # Fallback
-    if any("FAILED" in str(r) for r in tool_results):
-        return (
-            "[TOOLS USED] Tools failed - Network issue\n"
-            f"[SYMPTOMS] {user_input}\n"
-            "[PRIMARY DIAGNOSIS] Possible viral fever/infection\n"
-            "[DIAGNOSIS DESCRIPTION] Fever+chills suggests infection.\n"
-            "[FIRST AID] Rest, hydrate, paracetamol.\n"
-            "[EMERGENCY] No"
-        )
+        tool_results.append("TOOL FAILURE")
 
     final_prompt = f"""
 TOOL RESULTS:
@@ -189,11 +155,40 @@ USER INPUT:
 Provide diagnosis:
 """
 
+    return call_groq(final_prompt)
+
+# ==================================================
+# 🆕 IMAGE DIAGNOSIS ENTRY POINT
+# ==================================================
+
+def nivra_vision(image_base64: str, hint_text: str = "") -> str:
+    """
+    image_base64: base64-encoded image string
+    hint_text: optional user description
+    """
+
     try:
-        return call_groq(final_prompt)
+        image_result = analyze_symptom_image.invoke(image_base64)
     except Exception as e:
-        print("❌ GROQ ERROR:", e)
-        return (
-            "⚠️ Temporary AI service issue.\n\n"
-            "Please try again in a moment."
-        )
+        return f"""
+[TOOLS USED] analyze_symptom_image [/TOOLS USED]
+[SYMPTOMS] Image could not be analyzed [/SYMPTOMS]
+[PRIMARY DIAGNOSIS] Unable to assess from image [/PRIMARY DIAGNOSIS]
+[DIAGNOSIS DESCRIPTION]
+Image analysis failed. Please try again with a clearer image.
+[/DIAGNOSIS DESCRIPTION]
+[FIRST AID] Consult a healthcare professional [/FIRST AID]
+[EMERGENCY CONSULTATION REQUIRED] Yes [/EMERGENCY CONSULTATION REQUIRED]
+"""
+
+    final_prompt = f"""
+IMAGE ANALYSIS:
+{image_result}
+
+USER CONTEXT:
+{hint_text}
+
+Provide medical image-based preliminary assessment:
+"""
+
+    return call_groq(final_prompt)