Update app.py

app.py

@@ -231,11 +231,11 @@ def clean_response(text: str) -> str:
     # Pre-compiled regex patterns for cleaning
     patterns = [
         (re.compile(r"\[.*?\]|\bNone\b", re.IGNORECASE), ""),
-        (re.compile(r"To analyze the patient record excerpt.*?medications\.", re.IGNORECASE), ""),
+        (re.compile(r"To (analyze|proceed).*?medications\.", re.IGNORECASE), ""),
         (re.compile(r"Since the previous attempts.*?\.", re.IGNORECASE), ""),
-        (re.compile(r"I need to.*?medications\.", re.IGNORECASE), ""),
-        (re.compile(r"Retrieving tools.*?\.", re.IGNORECASE), ""),
-        (re.compile(r"I will start by retrieving.*?\.", re.IGNORECASE), ""),
+        (re.compile(r"I need to.*?results\.", re.IGNORECASE), ""),
+        (re.compile(r"(Therefore, )?I will start by retrieving.*?\.", re.IGNORECASE), ""),
+        (re.compile(r"(Therefore, )?Retrieving tools.*?\.", re.IGNORECASE), ""),
         (re.compile(r"This requires reviewing.*?\.", re.IGNORECASE), ""),
         (re.compile(r"Given the context, it is important to review.*?\.", re.IGNORECASE), ""),
         (re.compile(r"\s+"), " "),
@@ -286,8 +286,8 @@ def summarize_findings(combined_response: str) -> str:
                 if diagnosis:
                     diagnoses.append(diagnosis)
     
-    # Extract findings from non-sectioned text (e.g., psychiatric evaluation, medications)
-    medication_pattern = re.compile(r"medications including ([^\.]+)", re.IGNORECASE)
+    # Extract findings from non-sectioned text
+    medication_pattern = re.compile(r"medications includ(?:e|ing|ed) ([^\.]+)", re.IGNORECASE)
     evaluation_pattern = re.compile(r"psychiatric evaluation.*?mention of ([^\.]+)", re.IGNORECASE)
     
     for line in combined_response.splitlines():
@@ -298,12 +298,12 @@ def summarize_findings(combined_response: str) -> str:
         med_match = medication_pattern.search(line)
         if med_match:
             meds = med_match.group(1).strip()
-            diagnoses.append(f"use of medications ({meds}), which may indicate an undiagnosed psychiatric condition requiring urgent review")
+            diagnoses.append(f"use of medications ({meds}), suggesting an undiagnosed psychiatric condition requiring urgent review")
         
         eval_match = evaluation_pattern.search(line)
         if eval_match:
             details = eval_match.group(1).strip()
-            diagnoses.append(f"psychiatric evaluation noting {details}, suggesting a potential missed psychiatric diagnosis requiring urgent review")
+            diagnoses.append(f"psychiatric evaluation noting {details}, indicating a potential missed psychiatric diagnosis requiring urgent review")
     
     if not diagnoses:
         return "No missed diagnoses were identified in the provided records."
@@ -316,7 +316,7 @@ def summarize_findings(combined_response: str) -> str:
     summary = "The patient record indicates missed diagnoses including "
     summary += ", ".join(unique_diagnoses[:-1])
     summary += f", and {unique_diagnoses[-1]}" if len(unique_diagnoses) > 1 else unique_diagnoses[0]
-    summary += ". These findings, derived from the provided clinical data, suggest potential oversights in the patient's medical evaluation and require urgent clinical review to prevent adverse outcomes."
+    summary += ". These findings suggest potential oversights in the patient's medical evaluation and require urgent clinical review to prevent adverse outcomes."
     
     return summary
 
@@ -352,7 +352,7 @@ def init_agent():
 def create_ui(agent):
     """Optimized UI creation with pre-compiled templates"""
     PROMPT_TEMPLATE = """
-Analyze the patient record excerpt for missed diagnoses only, focusing on clinical findings such as symptoms, medications, or evaluation results. Provide a concise, evidence-based summary as a single paragraph without headings or bullet points. Include specific clinical findings (e.g., 'elevated blood pressure (160/95) on page 10'), their potential implications (e.g., 'may indicate untreated hypertension'), and a recommendation for urgent review. Do not use external tools unless explicitly required by the excerpt, and avoid mentioning other oversight categories like medication conflicts. If no missed diagnoses are found, state 'No missed diagnoses identified' in a single sentence. Use only the information provided in the excerpt below.
+Analyze the patient record excerpt for missed diagnoses, focusing only on clinical findings such as symptoms, medications, or evaluation results provided in the excerpt. Provide a concise, evidence-based summary in one paragraph without headings or bullet points. Include specific findings (e.g., 'elevated blood pressure (160/95) on page 10'), their implications (e.g., 'may indicate untreated hypertension'), and recommend urgent review. Do NOT use external tools or retrieve additional data. If no missed diagnoses are found, state 'No missed diagnoses identified' in one sentence. Ignore non-clinical data (e.g., name, date of birth) and other oversight categories (e.g., medication conflicts).
 Patient Record Excerpt (Chunk {0} of {1}):
 {chunk}
 """