#!/usr/bin/env python3
content = open('/root/projects/jg/2026-contractor-management-offshore/working/gen_infographic.py').read()

# Fix 1: add_colored_table function - handle variable col_colors length
old_func = '''def add_colored_table(doc, headers, rows, col_colors, cw=None):
    """Tabel met gekleurde kolomheaders"""
    t = doc.add_table(rows=1+len(rows), cols=len(headers)); t.style = 'Table Grid'; t.alignment = WD_TABLE_ALIGNMENT.CENTER
    for i,h in enumerate(headers):
        c = t.rows[0].cells[i]; c.text = ''; p = c.paragraphs[0]; r = p.add_run(h)
        r.bold = True; r.font.color.rgb = WHITE; r.font.size = Pt(10); scs(c, col_colors[i])'''

new_func = '''def add_colored_table(doc, headers, rows, col_colors, cw=None):
    """Tabel met gekleurde kolomheaders"""
    t = doc.add_table(rows=1+len(rows), cols=len(headers)); t.style = 'Table Grid'; t.alignment = WD_TABLE_ALIGNMENT.CENTER
    for i,h in enumerate(headers):
        c = t.rows[0].cells[i]; c.text = ''; p = c.paragraphs[0]; r = p.add_run(h)
        r.bold = True; r.font.color.rgb = WHITE; r.font.size = Pt(10)
        color = col_colors[i] if i < len(col_colors) else PRIMARY
        scs(c, f"{color[0]:02X}{color[1]:02X}{color[2]:02X}")'''

content = content.replace(old_func, new_func)

# Fix 2: legenda table color
content = content.replace(
    'scs(c, f"{color.red:02X}{color.green:02X}{color.blue:02X}")',
    'scs(c, f"{color[0]:02X}{color[1]:02X}{color[2]:02X}")'
)

open('/root/projects/jg/2026-contractor-management-offshore/working/gen_infographic.py', 'w').write(content)

import py_compile
py_compile.compile('/root/projects/jg/2026-contractor-management-offshore/working/gen_infographic.py', doraise=True)
print('Fixed and compiles OK')