Update app.py

app.py

@@ -47,7 +47,7 @@ def file_hash(path: str) -> str:
     with open(path, "rb") as f:
         return hashlib.md5(f.read()).hexdigest()
 
-def extract_priority_pages(file_path: str, max_pages: int = 20) -> str:
+def extract_priority_pages(file_path: str, max_pages: int = 10) -> str:
     try:
         text_chunks = []
         with pdfplumber.open(file_path) as pdf:
@@ -128,7 +128,7 @@ def init_agent():
         tool_files_dict={"new_tool": target_tool_path},
         force_finish=True,
         enable_checker=True,
-        step_rag_num=8,
+        step_rag_num=4,
         seed=100,
         additional_default_tools=[],
     )
@@ -154,44 +154,36 @@ def create_ui(agent):
             extracted = ""
             file_hash_value = ""
             if files:
-                with ThreadPoolExecutor(max_workers=4) as executor:
+                with ThreadPoolExecutor(max_workers=6) as executor:
                     futures = [executor.submit(convert_file_to_json, f.name, f.name.split(".")[-1].lower()) for f in files]
                     results = [sanitize_utf8(f.result()) for f in as_completed(futures)]
                     extracted = "\n".join(results)
                     file_hash_value = file_hash(files[0].name) if files else ""
 
             prompt = f"""
-You are a clinical reasoning assistant. Analyze the following medical records with a focus on identifying potential clinical oversights. Your task is to provide a comprehensive and structured summary addressing the following:
+Analyze the medical records for clinical oversights. Provide a concise, evidence-based summary under these headings:
 
-1. **Potential Missed Diagnoses**:
-   - Consider any contradictory or missing evidence regarding the patient’s history, symptoms, or test results.
-   - Take into account psychiatric, neurological, infectious, autoimmune, and genetic conditions.
-   - Consider family history, trauma history, and childhood developmental information.
+1. **Missed Diagnoses**:
+   - Identify inconsistencies in history, symptoms, or tests.
+   - Consider psychiatric, neurological, infectious, autoimmune, genetic conditions, family history, trauma, and developmental factors.
 
-2. **Flagged Medication Conflicts**:
-   - Cross-check all prescribed medications for contraindications, interactions, or off-label use without justification.
-   - Verify if any medications may worsen known diagnoses or contribute to adverse effects.
+2. **Medication Conflicts**:
+   - Check for contraindications, interactions, or unjustified off-label use.
+   - Assess if medications worsen diagnoses or cause adverse effects.
 
-3. **Incomplete or Missing Assessments**:
-   - Identify any domains of assessment that are completely missing or only superficially addressed (e.g., cognitive, psychiatric, social, family).
-   - Point out gaps in documentation of prior medical history, substance use, or lab/imaging results.
+3. **Incomplete Assessments**:
+   - Note missing or superficial cognitive, psychiatric, social, or family assessments.
+   - Highlight gaps in medical history, substance use, or lab/imaging documentation.
 
-4. **Abnormal Results Needing Urgent Follow-up**:
-   - Extract and highlight lab results, imaging, behavioral observations, or legal history that warrant immediate reassessment or specialist referral.
+4. **Urgent Follow-up**:
+   - Flag abnormal lab results, imaging, behaviors, or legal history needing immediate reassessment or referral.
 
-Make the answer precise, evidence-based, and structured clearly under each heading. Avoid repeating input and remove tool-call formatting.
+Medical Records (Truncated to 8k chars):
+{extracted[:8000]}
 
----
-
-Medical Records Input (Truncated to 12k chars):
-
-{extracted[:12000]}
-
----
-Begin your analysis below:
+Begin analysis:
 """
 
-
             try:
                 if history and history[-1]["content"].startswith("⏳"):
                     history.pop()
@@ -200,7 +192,7 @@ Begin your analysis below:
                     message=prompt,
                     history=[],
                     temperature=0.2,
-                    max_new_tokens=2048,
+                    max_new_tokens=1024,
                     max_token=4096,
                     call_agent=False,
                     conversation=[],
@@ -246,4 +238,4 @@ if __name__ == "__main__":
         show_error=True,
         allowed_paths=[report_dir],
         share=False
-    )
+    )