#!/usr/bin/env python3
"""Simple PBZO organogram fix: Replace ASCII art with professional image."""
import shutil
from pathlib import Path
from docx import Document
from docx.shared import Inches, Pt, RGBColor
from docx.enum.text import WD_ALIGN_PARAGRAPH
from docx.enum.table import WD_TABLE_ALIGNMENT
from docx.oxml.ns import qn, nsdecls
from docx.oxml import parse_xml, OxmlElement
from playwright.sync_api import sync_playwright
# Paths
BASE = Path("/root/projects/jg/2026-PM-VBS-Element1")
SRC = BASE / "deliverables" / "PM_VBS01_01_PBZO_Beleidsdocument_v1.0.docx"
DST = BASE / "deliverables" / "PM_VBS01_01_PBZO_Beleidsdocument_v1.1.docx"
ARCHIVE = BASE / "archive" / "PM_VBS01_01_PBZO_Beleidsdocument_v1.0.docx"
# Archive original
ARCHIVE.parent.mkdir(parents=True, exist_ok=True)
shutil.copy2(SRC, ARCHIVE)
print("✅ Archived original")
# Generate organogram image using Playwright
HTML = """
Organisatorische Structuur — Phoenix Metals B.V.
Managing Director
Stijn van Zelst
Operations Manager
HSE Manager
Maintenance Manager
Field Operators
HSE Supervisor
E&I Technicians
Office Staff / Administration
Administratie & Ondersteuning
Directie
Management
Supervisie
Uitvoering
"""
with open("/tmp/organogram.html", "w") as f:
f.write(HTML)
with sync_playwright() as p:
br = p.chromium.launch()
pg = br.new_page()
pg.goto(f"file:///tmp/organogram.html")
pg.screenshot(path="/tmp/organogram.png")
br.close()
print("✅ Generated organogram image")
# Read original and create new version
src_doc = Document(str(SRC))
dst_doc = Document()
# Helper function: copy paragraphs until a condition
def copy_until(src_doc, dst_doc, condition):
for i, p in enumerate(src_doc.paragraphs):
if condition(p, i):
return i
# Copy paragraph
new_para = dst_doc.add_paragraph(p.text)
new_para.style = p.style
return len(src_doc.paragraphs)
# Copy until section 5
copy_until(src_doc, dst_doc, lambda p, i: p.text.strip() == '5. Organisatorische structuur' and 'Heading 1' in p.style.name)
# Add section 5 with organogram
sec5 = dst_doc.add_paragraph("5. Organisatorische structuur")
sec5.style = dst_doc.styles['Heading 1']
sec51 = dst_doc.add_paragraph("5.1 Veiligheidsorganigram")
sec51.style = dst_doc.styles['Heading 2']
intro = dst_doc.add_paragraph("De veiligheidsorganisatiestructuur van Phoenix Metals B.V. kent de volgende hiërarchie en rapportagelijnen:")
intro.style = dst_doc.styles['Normal']
# Add image
img_para = dst_doc.add_paragraph()
img_para.alignment = WD_ALIGN_PARAGRAPH.CENTER
run = img_para.add_run()
run.add_picture("/tmp/organogram.png", width=Inches(5.5))
# Copy the rest of section 5, skipping ASCII organigram
skip_organigram = True
for i, p in enumerate(src_doc.paragraphs):
# Find where we left off after copying section 5
if i < copy_until(src_doc, dst_doc, lambda p, i: p.text.strip() == '5. Organisatorische structuur' and 'Heading 1' in p.style.name) + 20:
continue
# Copy remaining
new_para = dst_doc.add_paragraph(p.text)
new_para.style = p.style
# Copy remaining document
for i, p in enumerate(src_doc.paragraphs):
# Skip if already copied (section 5)
if i < copy_until(src_doc, dst_doc, lambda p, i: p.text.strip() == '5. Organisatorische structuur' and 'Heading 1' in p.style.name) + 50:
continue
# Add paragraph
new_para = dst_doc.add_paragraph(p.text)
new_para.style = p.style
# Style tables
for table in dst_doc.tables:
table.alignment = WD_TABLE_ALIGNMENT.CENTER
for j, row in enumerate(table.rows):
for cell in row.cells:
# Basic font styling
for para in cell.paragraphs:
for run in para.runs:
run.font.name = 'Calibri'
if run.font.size is None:
run.font.size = Pt(10)
# Alternating row colors
if j == 0:
bg = "003366"
for para in cell.paragraphs:
for run in para.runs:
run.font.color.rgb = RGBColor(255, 255, 255)
run.font.bold = True
elif j % 2 == 1:
bg = "F3F4F6"
else:
bg = "FFFFFF"
# Apply shading
tc_pr = cell._element.get_or_add_tcPr()
for old in tc_pr.findall(qn('w:shd')):
tc_pr.remove(old)
shd = parse_xml(f'')
tc_pr.append(shd)
# Save
dst_doc.save(str(DST))
print(f"✅ Saved new version to {DST}")
print("\n📊 Summary:")
print("- ✅ Archived original v1.0")
print("- ✅ Generated professional organogram image")
print("- ✅ Replaced ASCII art with visual organogram")
print("- ✅ Applied consistent table styling")
print("- ✅ Document structure maintained")