Update PatientInfoExtractionEngine.py

PatientInfoExtractionEngine.py

@@ -53,17 +53,23 @@ class PatientInfoExtractionEngine:
         )
 
         # Task 1: Demographics Extraction
+        
         self.demographics_task = Task(
             description=(
-                "Extract patient demographics from the patient chart including:\n"
+                "Extract patient demographics from the patient chart. "
+                "All patient demographic information is usually located together in one section "
+                "(for example, at the beginning or in a 'Patient Information' block). "
+                "Extract details only from that section — do NOT search or aggregate information "
+                "from multiple parts of the chart.\n\n"
+                "Fields to extract:\n"
                 "- name\n"
                 "- dob (date of birth)\n"
                 "- age\n"
                 "- gender\n"
-                "- address (patient address only if present in chart,don't add any hospital name or medical centre name)\n\n"
+                "- address (only patient’s residential address; exclude any hospital or medical centre names)\n"
                 "- phone\n"
                 "- patient_identifier (may appear as patient ID, MRN, or insurance ID in the chart)\n\n"
-                "Return only a JSON object in the following format:\n"
+                "Return the result strictly as a JSON object in the format:\n"
                 "{\n"
                 "  'name': '',\n"
                 "  'dob': '',\n"
@@ -74,14 +80,15 @@ class PatientInfoExtractionEngine:
                 "  'patient_identifier': ''\n"
                 "}\n\n"
                 "Rules:\n"
-                "- Only include values that are explicitly present in the chart.\n"
+                "- Extract data only from the section where all patient demographic information is grouped together.\n"
+                "- Do not collect or merge values scattered across different notes or sections.\n"
+                "- Only include values explicitly mentioned in the chart.\n"
                 "- If a value is missing or not mentioned, leave it as an empty string.\n"
-                "- Do not hallucinate or guess missing information."
+                "- Do not hallucinate, infer, or guess missing details."
             ),
-            expected_output="JSON object with demographics fields only, leaving missing ones empty.",
-            agent=self.patient_info_agent,
+            expected_output="A JSON object containing the extracted patient demographics.",
+            agent=self.patient_info_agent,        
         )
-
         # Combine agents and tasks in Crew
         self.crew = Crew(
             agents=[self.patient_info_agent],