#!/usr/bin/env python3
"""Bijlage A fix: Rebuild document with professional organogram."""

from docx import Document
from docx.shared import Inches, Pt, RGBColor
from docx.enum.text import WD_ALIGN_PARAGRAPH
from docx.oxml.ns import qn, nsdecls
import shutil
from pathlib import Path

# Read original
orig_doc = Document("/root/projects/jg/2026-PM-VBS-Element1/deliverables/PM_VBS01_01_PBZO_Beleidsdocument_v1.1.docx")
new_doc = Document()

# Copy up to "9. Bijlagen"
for i, p in enumerate(orig_doc.paragraphs):
    if p.text.strip() == '9. Bijlagen' and 'Heading 1' in p.style.name:
        # Copy up to and including this
        for j in range(i + 1):
            p_orig = orig_doc.paragraphs[j]
            new_para = new_doc.add_paragraph(p_orig.text)
            new_para.style = p_orig.style
        break

# Add professional Bijlage A
bijA_title = new_doc.add_paragraph("Bijlage A: Organigram")
bijA_title.style = new_doc.styles['Heading 2']

intro_para = new_doc.add_paragraph("Onderstaand organogram toont de veiligheidsstructuur van Phoenix Metals B.V.:")
intro_para.style = new_doc.styles['Normal']

# Add organogram image
img_para = new_doc.add_paragraph()
img_para.alignment = WD_ALIGN_PARAGRAPH.CENTER
run = img_para.add_run()
run.add_picture("/tmp/_pbzo_organogram.png", width=Inches(5.5))

# Add space
space_para = new_doc.add_paragraph()
space_para.style = new_doc.styles['Normal']

# Add content from original after "9. Bijlagen"
for i, p in enumerate(orig_doc.paragraphs):
    if p.text.strip() == '9. Bijlagen' and 'Heading 1' in p.style.name:
        start_idx = i + 1
        break
else:
    start_idx = 0

for i in range(start_idx, len(orig_doc.paragraphs)):
    p_orig = orig_doc.paragraphs[i]
    # Skip old Bijlage A
    if p_orig.text.strip().startswith('Bijlage A:') and 'Heading 2' in p_orig.style.name:
        continue
    # Keep everything else
    new_para = new_doc.add_paragraph(p_orig.text)
    new_para.style = p_orig.style

# Save final document
new_doc.save("/root/projects/jg/2026-PM-VBS-Element1/deliverables/PM_VBS01_01_PBZO_Beleidsdocument_v1.1.docx")
print("✅ Rebuilt document with professional Bijlage A organogram")
print("✅ Document structure corrected")