Update chartdiagnosischecker.py

chartdiagnosischecker.py

@@ -12,6 +12,7 @@ from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain_community.vectorstores import Chroma
 from embedding_manager import DirectoryEmbeddingManager
 
+# ---------- Chart Checker ----------
 class ChartDiagnosisChecker:
     def __init__(self, pdf_dir: str, model: str = "gpt-4o"):
         self.embed_manager = DirectoryEmbeddingManager(pdf_dir)
@@ -43,10 +44,12 @@ class ChartDiagnosisChecker:
                 f"Diagnosis to validate: {diagnosis} (ICD-10: {icd10})\n\n"
                 f"Reference: {ref}\n\n"
                 f"Patient chart excerpts:\n{context}\n\n"
-                f"Does the patient have {diagnosis} or anything related to this diagnosis"
+                f"Does the patient have {diagnosis} from the patient contexts provided"
                 "Answer Yes or No, with a short justification using ONLY the provided chart text.\n"
+                "Also check whether the diagnois can be inferred implicitly like by the medication taken by patient"
+                "or any other means. Compare the drugs and tests done by the patient with those required for the diagnosis"
                 "Output must be valid JSON in the form:\n"
-                "{'answer': 'yes/no', 'rationale': 'one-line rationale'}"
+                "{'answer_explicit': 'yes/no', 'rationale_explicit': 'one-line rationale', answer_implicit': 'yes/no', 'rationale_implicit': 'one-line rationale'}"
             ),
             expected_output="JSON with keys answer and rationale",
             agent=self.agent,
@@ -62,8 +65,10 @@ class ChartDiagnosisChecker:
             "diagnosis": diagnosis,
             "icd10": icd10,
             "reference": ref,
-            "answer": result.get("answer", "unknown"),
-            "rationale": result.get("rationale", ""),
+            "answer_explicit": result.get("answer_explicit", "unknown"),
+            "rationale_explicit": result.get("rationale_explicit", ""),
+            "answer_implicit": result.get("answer_implicit", "unknown"),
+            "rationale_implicit": result.get("rationale_implicit", ""),
             "context": context
         }
 
@@ -75,4 +80,6 @@ class ChartDiagnosisChecker:
             result = self.check_one(entry)
             results.append(result)
             print(f"[ANSWER] {result}")
-        return results
+        return results
+
+