Update updated_word.py

updated_word.py

@@ -1490,6 +1490,82 @@ def process_red_text_in_paragraph(paragraph, context_text, flat_json):
 
     return replacements_made
 
+def process_red_text_in_context_paragraph(paragraph, heading_text, flat_json, operator_name):
+    """Process red text found in paragraphs following headings"""
+    replacements_made = 0
+    red_text_segments = []
+    
+    for run in paragraph.runs:
+        if is_red(run) and run.text.strip():
+            red_text_segments.append(run.text.strip())
+    
+    if not red_text_segments:
+        return 0
+    
+    combined_red_text = " ".join(red_text_segments).strip()
+    print(f"      🔍 Red text found: '{combined_red_text}'")
+    
+    replacement_value = None
+    
+    # Determine what to replace based on heading context
+    if any(mgmt_type in heading_text.upper() for mgmt_type in ["MAINTENANCE MANAGEMENT", "MASS MANAGEMENT", "FATIGUE MANAGEMENT"]):
+        # For management section headings, replace with operator name
+        if operator_name:
+            replacement_value = operator_name
+            print(f"      ✅ Using operator name for management section: '{operator_name}'")
+    
+    elif "NHVAS APPROVED AUDITOR DECLARATION" in heading_text.upper():
+        # For auditor declarations, look for auditor name
+        auditor_name = None
+        for key, value in flat_json.items():
+            if "auditor" in key.lower() and "name" in key.lower():
+                if isinstance(value, list) and value:
+                    auditor_name = str(value[0]).strip()
+                elif value:
+                    auditor_name = str(value).strip()
+                break
+        
+        if auditor_name:
+            replacement_value = auditor_name
+            print(f"      ✅ Using auditor name: '{auditor_name}'")
+    
+    elif "OPERATOR DECLARATION" in heading_text.upper():
+        # For operator declarations, use operator name
+        if operator_name:
+            replacement_value = operator_name
+            print(f"      ✅ Using operator name for operator declaration: '{operator_name}'")
+    
+    else:
+        # For other headings, try to find a relevant match
+        # First try direct match
+        kv = find_matching_json_key_and_value(combined_red_text, flat_json)
+        if kv:
+            replacement_value = get_value_as_string(kv[1], combined_red_text)
+        else:
+            # Try contextual search with heading
+            context_queries = [f"{heading_text} {combined_red_text}", combined_red_text, heading_text]
+            for query in context_queries:
+                kv = find_matching_json_key_and_value(query, flat_json)
+                if kv:
+                    replacement_value = get_value_as_string(kv[1], combined_red_text)
+                    print(f"      ✅ Found match with combined query: {kv[0]}")
+                    break
+    
+    # Apply the replacement if we found a suitable value
+    if replacement_value:
+        red_runs = [run for run in paragraph.runs if is_red(run) and run.text.strip()]
+        if red_runs:
+            red_runs[0].text = replacement_value
+            red_runs[0].font.color.rgb = RGBColor(0, 0, 0)
+            for run in red_runs[1:]:
+                run.text = ''
+            replacements_made = 1
+            print(f"      ✅ Replaced with: '{replacement_value}'")
+    else:
+        print(f"      ❌ No suitable replacement found for: '{combined_red_text}'")
+    
+    return replacements_made
+
 # ============================================================================
 # Orchestrator
 # ============================================================================