Update master_key.py

master_key.py

@@ -1,6 +1,6 @@
 """
 Improved Master Key for NHVAS Audit extraction:
-- TABLE_SCHEMAS: Enhanced definitions with better matching criteria
+- TABLE_SCHEMAS: Enhanced definitions with better matching criteria for Summary vs Basic tables
 - HEADING_PATTERNS: Improved regex patterns for main/sub headings
 - PARAGRAPH_PATTERNS: Enhanced patterns for key narrative sections
 """
@@ -99,6 +99,8 @@ TABLE_SCHEMAS = {
         "labels": ["Title", "Abbreviation", "Description"],
         "priority": 80
     },
+    
+    # 🎯 BASIC MANAGEMENT SCHEMAS (Compliance Tables - Lower Priority)
     "Maintenance Management": {
         "headings": [
             {"level": 1, "text": "NHVAS AUDIT SUMMARY REPORT"}
@@ -115,7 +117,8 @@ TABLE_SCHEMAS = {
             "Std 8. Training and Education"
         ],
         "priority": 60,
-        "context_keywords": ["maintenance"]
+        "context_keywords": ["maintenance"],
+        "context_exclusions": ["summary", "details", "audit findings"]  # Exclude Summary tables
     },
     "Mass Management": {
         "headings": [
@@ -133,7 +136,8 @@ TABLE_SCHEMAS = {
             "Std 8. Maintenance of Suspension"
         ],
         "priority": 60,
-        "context_keywords": ["mass"]
+        "context_keywords": ["mass"],
+        "context_exclusions": ["summary", "details", "audit findings"]  # Exclude Summary tables
     },
     "Fatigue Management": {
         "headings": [
@@ -150,8 +154,11 @@ TABLE_SCHEMAS = {
             "Std 7. Workplace conditions"
         ],
         "priority": 60,
-        "context_keywords": ["fatigue"]
+        "context_keywords": ["fatigue"],
+        "context_exclusions": ["summary", "details", "audit findings"]  # Exclude Summary tables
     },
+    
+    # 🎯 SUMMARY MANAGEMENT SCHEMAS (Detailed Tables with DETAILS column - Higher Priority)
     "Maintenance Management Summary": {
         "headings": [
             {"level": 1, "text": "Audit Observations and Comments"},
@@ -169,7 +176,8 @@ TABLE_SCHEMAS = {
             "Std 7. Internal Review", 
             "Std 8. Training and Education"
         ],
-        "priority": 70
+        "priority": 85,  # Higher priority than basic Maintenance Management
+        "context_keywords": ["maintenance", "summary", "details", "audit findings"]
     },
     "Mass Management Summary": {
         "headings": [
@@ -179,15 +187,16 @@ TABLE_SCHEMAS = {
         "columns": ["MASS MANAGEMENT", "DETAILS"],
         "labels": [
             "Std 1. Responsibilities",
-            "Std 2. Vehicle Control",
+            "Std 2. Vehicle Control", 
             "Std 3. Vehicle Use",
-            "Std 4. Records and Documentation", 
+            "Std 4. Records and Documentation",
             "Std 5. Verification",
             "Std 6. Internal Review",
             "Std 7. Training and Education",
             "Std 8. Maintenance of Suspension"
         ],
-        "priority": 70
+        "priority": 85,  # Higher priority than basic Mass Management
+        "context_keywords": ["mass", "summary", "details", "audit findings"]
     },
     "Fatigue Management Summary": {
         "headings": [
@@ -204,8 +213,11 @@ TABLE_SCHEMAS = {
             "Std 6. Records and Documentation",
             "Std 7. Workplace conditions"
         ],
-        "priority": 70
+        "priority": 85,  # Higher priority than basic Fatigue Management
+        "context_keywords": ["fatigue", "summary", "details", "audit findings"]
     },
+    
+    # Vehicle Registration Tables
     "Vehicle Registration Numbers Maintenance": {
         "headings": [
             {"level": 1, "text": "Vehicle Registration Numbers of Records Examined"},
@@ -256,6 +268,8 @@ TABLE_SCHEMAS = {
         "priority": 80,
         "context_keywords": ["driver", "scheduler", "fatigue"]
     },
+    
+    # Other Tables
     "Operator's Name (legal entity)": {
         "headings": [
             {"level": 1, "text": "CORRECTIVE ACTION REQUEST (CAR)"}