Upload agents/priority_router.py with huggingface_hub

agents/priority_router.py

@@ -315,19 +315,27 @@ Be conservative - err on the side of higher priority for concerning findings."""
         """Check for critical findings that require immediate communication."""
         detected_critical = []
         
-        report_lower = report_text.lower()
-        for critical in self.CRITICAL_FINDINGS:
-            if critical in report_lower:
-                detected_critical.append(critical.replace("_", " ").title())
-        
-        # Also check finding types
+        # Only check actual findings from the analysis, not report text
+        # (Report text may contain "no pneumothorax" which would false-positive)
         for finding in findings:
             finding_type = finding.get("type", "").lower()
-            if finding_type in self.CRITICAL_FINDINGS:
+            severity = finding.get("severity", "").lower()
+            
+            # Only flag as critical if it's actually a critical finding type
+            # AND has high/critical severity
+            if finding_type in self.CRITICAL_FINDINGS and severity in ["critical", "high", "moderate"]:
                 name = finding_type.replace("_", " ").title()
                 if name not in detected_critical:
                     detected_critical.append(name)
         
+        # Also check for specific high-severity findings
+        for finding in findings:
+            severity = finding.get("severity", "").lower()
+            if severity == "critical":
+                finding_type = finding.get("type", "Unknown").replace("_", " ").title()
+                if finding_type not in detected_critical:
+                    detected_critical.append(f"{finding_type} (Critical)")
+        
         return detected_critical
     
     def _determine_routing(self, priority_level: str, critical_findings: List[str]) -> Dict: