from reportlab.lib.pagesizes import A4, landscape
from reportlab.lib import colors
from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
from reportlab.lib.units import cm, mm
from reportlab.platypus import (
    SimpleDocTemplate, Table, TableStyle, Paragraph,
    Spacer, HRFlowable, PageBreak, KeepTogether
)
from reportlab.lib.enums import TA_CENTER, TA_LEFT
from reportlab.platypus import BaseDocTemplate, Frame, PageTemplate
from reportlab.lib.colors import HexColor
import datetime

# ── Colour palette ───────────────────────────────────────────────────────────
TEAL       = HexColor('#006D77')
TEAL_LIGHT = HexColor('#83C5BE')
PINK       = HexColor('#C77DFF')
PINK_LIGHT = HexColor('#E9C8FF')
GOLD       = HexColor('#FFB703')
ORANGE     = HexColor('#FB8500')
GREY_DARK  = HexColor('#2B2D42')
GREY_MID   = HexColor('#5C5F72')
GREY_LIGHT = HexColor('#EDF2F4')
WHITE      = colors.white
RED_LIGHT  = HexColor('#FFE5E5')
GREEN_LIGHT= HexColor('#E5F5E5')
BLUE_LIGHT = HexColor('#E5F0FF')

OUTPUT = '/home/daytona/workspace/obgyn-pdf/OBGyn_Quick_Reference.pdf'

styles = getSampleStyleSheet()

# Custom paragraph styles
title_style = ParagraphStyle('Title', parent=styles['Normal'],
    fontSize=20, textColor=WHITE, fontName='Helvetica-Bold',
    alignment=TA_CENTER, spaceAfter=2)

subtitle_style = ParagraphStyle('Subtitle', parent=styles['Normal'],
    fontSize=10, textColor=HexColor('#CCEEEE'), fontName='Helvetica',
    alignment=TA_CENTER, spaceAfter=0)

section_style = ParagraphStyle('Section', parent=styles['Normal'],
    fontSize=12, textColor=WHITE, fontName='Helvetica-Bold',
    alignment=TA_LEFT, spaceBefore=4, spaceAfter=4)

note_style = ParagraphStyle('Note', parent=styles['Normal'],
    fontSize=6.5, textColor=GREY_MID, fontName='Helvetica',
    alignment=TA_LEFT, leading=9)

footer_style = ParagraphStyle('Footer', parent=styles['Normal'],
    fontSize=7, textColor=GREY_MID, fontName='Helvetica-Oblique',
    alignment=TA_CENTER)

cell_style = ParagraphStyle('Cell', parent=styles['Normal'],
    fontSize=7.2, fontName='Helvetica', leading=9.5,
    textColor=GREY_DARK)

cell_bold = ParagraphStyle('CellBold', parent=styles['Normal'],
    fontSize=7.5, fontName='Helvetica-Bold', leading=10,
    textColor=GREY_DARK)

cell_small = ParagraphStyle('CellSmall', parent=styles['Normal'],
    fontSize=6.5, fontName='Helvetica', leading=8.5,
    textColor=GREY_MID)


def P(text, style=None):
    if style is None:
        style = cell_style
    return Paragraph(str(text), style)

def PB(text):
    return Paragraph(str(text), cell_bold)

def PS(text):
    return Paragraph(str(text), cell_small)


def section_header(title, color=TEAL):
    data = [[Paragraph(title, section_style)]]
    t = Table(data, colWidths=[26*cm])
    t.setStyle(TableStyle([
        ('BACKGROUND', (0,0), (-1,-1), color),
        ('ROWBACKGROUNDS', (0,0), (-1,-1), [color]),
        ('TOPPADDING',    (0,0), (-1,-1), 6),
        ('BOTTOMPADDING', (0,0), (-1,-1), 6),
        ('LEFTPADDING',   (0,0), (-1,-1), 10),
        ('RIGHTPADDING',  (0,0), (-1,-1), 6),
        ('ROUNDEDCORNERS',(0,0), (-1,-1), [4,4,4,4]),
    ]))
    return t


def make_table(headers, rows, col_widths, alt_colors=(GREY_LIGHT, WHITE)):
    """Build a styled data table."""
    header_row = [PB(h) for h in headers]
    data = [header_row]
    for row in rows:
        data.append([P(c) if not isinstance(c, Paragraph) else c for c in row])

    t = Table(data, colWidths=col_widths, repeatRows=1)

    row_styles = []
    for i in range(1, len(data)):
        bg = alt_colors[i % 2]
        row_styles.append(('BACKGROUND', (0, i), (-1, i), bg))

    t.setStyle(TableStyle([
        # Header
        ('BACKGROUND',    (0, 0), (-1, 0), TEAL),
        ('TEXTCOLOR',     (0, 0), (-1, 0), WHITE),
        ('FONTNAME',      (0, 0), (-1, 0), 'Helvetica-Bold'),
        ('FONTSIZE',      (0, 0), (-1, 0), 8),
        ('TOPPADDING',    (0, 0), (-1, 0), 7),
        ('BOTTOMPADDING', (0, 0), (-1, 0), 7),
        # Body
        ('FONTNAME',      (0, 1), (-1, -1), 'Helvetica'),
        ('FONTSIZE',      (0, 1), (-1, -1), 7.2),
        ('TOPPADDING',    (0, 1), (-1, -1), 5),
        ('BOTTOMPADDING', (0, 1), (-1, -1), 5),
        ('LEFTPADDING',   (0, 0), (-1, -1), 5),
        ('RIGHTPADDING',  (0, 0), (-1, -1), 5),
        ('VALIGN',        (0, 0), (-1, -1), 'TOP'),
        # Grid
        ('GRID',          (0, 0), (-1, -1), 0.35, HexColor('#CCCCCC')),
        ('LINEBELOW',     (0, 0), (-1, 0),  1.2,  TEAL_LIGHT),
    ] + row_styles))
    return t


# ── Data ─────────────────────────────────────────────────────────────────────

GYNAECOLOGY_HEADERS = ['Condition', 'First-Line Treatment', 'Dose & Frequency',
                        'Second-Line Treatment', 'Dose & Frequency', 'Notes']
GYN_COL_W = [3.4*cm, 3.6*cm, 4.2*cm, 3.6*cm, 4.2*cm, 5.0*cm]  # 24 cm total (landscape A4 = ~26 cm usable)

GYN_ROWS = [
    ['Primary Dysmenorrhea',
     'NSAIDs\n(Ibuprofen / Mefenamic acid)',
     'Ibuprofen 400-600 mg TID\nMefenamic 500 mg load\nthen 250 mg Q8h',
     'Combined OCP\n(cyclic or continuous)',
     '1 tab OD',
     'Start 1-2 days before menses; continue 2-3 days. ACOG / Harrison 22E'],

    ['Dysmenorrhea\n(refractory)',
     'Progestin-only pill or\nLNG-IUD',
     'Norethisterone 5 mg BD-TID\nor LNG-IUD (insert once)',
     'GnRH agonist (Leuprolide)\n+ add-back therapy',
     'Leuprolide 3.75 mg IM/month\n+ Norethindrone 5 mg OD',
     'Add-back prevents bone loss. Max 6 months GnRH agonist.'],

    ['AUB / Menorrhagia\n(acute)',
     'Tranexamic acid',
     '1g TID, days 1-5 of menses',
     'High-dose progestin\n(Norethisterone)',
     '5 mg TID until bleeding stops,\nthen taper',
     'TXA reduces menstrual loss by ~50%. FDA-approved. Berek & Novak'],

    ['AUB / Menorrhagia\n(chronic/anovulatory)',
     'Combined OCP or\ncyclic progestin (MPA)',
     'MPA 10 mg/day, days 15-26\nor low-dose COC OD',
     'LNG-IUD',
     'Insert once, effective 5 years',
     'LNG-IUD preferred before hysterectomy. Berek & Novak\'s Gynecology'],

    ['Bacterial Vaginosis',
     'Metronidazole oral',
     '500 mg BD x 7 days',
     'Clindamycin vaginal\ncream 2%',
     '5g intravaginally OD\nx 7 nights',
     'Treat partner only if symptomatic. Rosen\'s EM'],

    ['Vulvovaginal Candidiasis',
     'Fluconazole oral\n(single dose)',
     '150 mg once',
     'Clotrimazole pessary\n100 mg intravaginal',
     'OD x 6 nights',
     'Recurrent (>4/yr): Fluconazole 150 mg weekly x 6 months'],

    ['Trichomoniasis',
     'Metronidazole\n2g single dose',
     '2g oral once',
     'Tinidazole 2g once\nor Metronidazole 500 mg BD',
     '2g once OR\n500 mg BD x 7 days',
     'Always treat partner simultaneously. Avoid alcohol.'],

    ['PCOS - Cycle\nregulation',
     'Combined OCP\n(drospirenone-based preferred)',
     '1 tab OD (cyclic)',
     'Cyclic progestin (MPA)\nor Myo-inositol',
     'MPA 10 mg x 12 days/month\nInositol 2g BD',
     '2023 ESHRE/ASRM international guideline'],

    ['PCOS - Metabolic /\nInsulin resistance',
     'Lifestyle modification\n(diet + exercise)',
     '≥5-10% weight loss;\n30 min exercise OD',
     'Metformin',
     '500 mg OD, titrate to\n1000-1500 mg/day',
     'First pharmacologic option for IR/metabolic features. ESHRE 2023'],

    ['PCOS - Ovulation\ninduction',
     'Letrozole',
     '2.5-5 mg, days 3-7 of cycle',
     'Clomiphene citrate',
     '50-150 mg, days 2-6',
     'Letrozole NOW preferred over clomiphene. ESHRE/ASRM 2023 guideline'],

    ['Endometriosis\n(pain management)',
     'NSAIDs + Combined OCP\n(continuous)',
     'Ibuprofen 400 mg TID\n+ OCP 1 tab OD continuous',
     'Progestins continuous\n(Norethisterone / Dienogest)',
     'Norethisterone 5 mg BD\nDienogest 2 mg OD',
     'Empirical treatment without laparoscopy acceptable. ACOG 2026 CPG'],

    ['Endometriosis\n(refractory)',
     'LNG-IUD',
     'Insert once, effective 5 years',
     'GnRH agonist + add-back HRT',
     'Leuprolide 3.75 mg IM/month\n+ Norethindrone 5 mg OD',
     'Max 6 months GnRH; add-back prevents bone loss'],

    ['PMS / PMDD',
     'SSRIs (Sertraline) -\ncontinuous or luteal phase',
     '50-150 mg/day or\nday 14 to menses only',
     'COC (drospirenone 3 mg +\nEE 20 mcg)',
     '1 tab OD x 24 active days',
     'ACOG CPG No. 7, 2023'],

    ['Premature Ovarian\nInsufficiency (POI)',
     'HRT - transdermal estradiol\n+ cyclic progestogen',
     'E2 patch 50-100 mcg/day\n+ MPA 10 mg x 12 days/month',
     'Combined OCP\n(if HRT not preferred)',
     '1 tab OD',
     'Continue until age 50. Protects bone, CV, cognition. ESHRE 2024'],

    ['UTI (non-pregnant)',
     'Nitrofurantoin OR\nFosfomycin',
     'Nitrofurantoin 100 mg BD x 5-7d\nFosfomycin 3g sachet once',
     'Co-trimoxazole OR\nCiprofloxacin',
     'TMP-SMX 960 mg BD x 3d\nCipro 500 mg BD x 3d',
     'Choice based on local antibiogram. Rosen\'s EM'],

    ['PID (outpatient)',
     'Ceftriaxone IM +\nDoxycycline +\nMetronidazole',
     'Ceftriaxone 500 mg IM once\nDoxy 100 mg BD x 14d\nMetro 400 mg BD x 14d',
     'Ofloxacin +\nMetronidazole',
     'Ofloxacin 400 mg BD x 14d\nMetro 400 mg BD x 14d',
     'Avoid quinolones if local gonorrhoea resistance high'],
]

OBSTETRICS_HEADERS = GYNAECOLOGY_HEADERS
OBS_COL_W = GYN_COL_W

OBS_ROWS = [
    ['Iron Deficiency Anaemia\n(pregnancy)',
     'Oral ferrous sulfate',
     'Ferrous sulfate 200 mg\n(60 mg elemental Fe) BD',
     'IV iron sucrose or\nFerric carboxymaltose',
     'IV iron sucrose 200 mg in\n100 mL NS over 30 min',
     'IV iron if Hb <7 g/dL or oral intolerance'],

    ['Gestational HTN /\nMild preeclampsia',
     'Methyldopa',
     '250-500 mg TID\n(max 3g/day)',
     'Labetalol OR\nNifedipine ER',
     'Labetalol 100-200 mg BD-TID\nNifedipine ER 30-60 mg OD',
     'Target BP <140/90 mmHg. SOMANZ 2023 guideline'],

    ['Severe HTN in pregnancy\n(BP ≥160/110 mmHg)',
     'IV Labetalol',
     '20 mg IV bolus, repeat Q10 min\n(max 300 mg total)',
     'Oral Nifedipine\n(immediate release)',
     '10 mg oral Q20-30 min\n(max 3 doses)',
     'SOMANZ 2023. Keep antidote (calcium gluconate) ready for MgSO4'],

    ['Eclampsia / Seizure\nprophylaxis',
     'Magnesium sulfate',
     '4g IV loading over 20 min,\nthen 1-2g/hr x 24 hours',
     'Phenytoin\n(rarely used now)',
     '15-20 mg/kg IV',
     'MgSO4 antidote: Calcium gluconate 1g IV slow. Monitor urine output, RR, reflexes'],

    ['Gestational Diabetes\n(diet-controlled)',
     'Medical nutrition therapy\n+ exercise',
     'Low GI diet;\n30 min moderate exercise OD',
     'Metformin',
     '500 mg OD, titrate to\n1000-2000 mg/day',
     'ACOG: delivery by 40+6 wks if diet-controlled. ADA 2025 standards'],

    ['Gestational Diabetes\n(medication-required)',
     'Insulin\n(NPH + regular / basal-bolus)',
     'Per glucose monitoring\nprotocol (individualised)',
     'Metformin\n(if insulin declined)',
     '500-2000 mg/day in\ndivided doses',
     'Insulin: gold standard in pregnancy. ACOG delivery 39-39+6 wks'],

    ['Nausea & Vomiting\nof Pregnancy (mild)',
     'Ginger + Pyridoxine (B6)',
     'Ginger 250 mg QID;\nB6 10-25 mg TDS',
     'Doxylamine + B6\n(Diclegis / Bonjesta)',
     'Doxylamine 10 mg + B6 10 mg\n2 tabs at night',
     'ACOG first-line. Take on empty stomach'],

    ['Hyperemesis Gravidarum',
     'IV fluids + Promethazine\nor Metoclopramide',
     'NS/RL rehydration;\nPromethazine 25 mg IM/IV Q6h',
     'Ondansetron\n(refractory cases)',
     '4-8 mg IV/oral Q8h',
     'Reserve ondansetron for refractory cases (QTc monitoring)'],

    ['UTI in Pregnancy',
     'Cephalexin OR\nNitrofurantoin',
     'Cephalexin 500 mg BD x 7d\nNitrofurantoin 100 mg BD x 5-7d',
     'Amoxicillin',
     '500 mg TID x 7d',
     'Avoid nitrofurantoin in 3rd trimester. Rosen\'s EM'],

    ['Preterm Labour\n(tocolysis, 24-34 wks)',
     'Nifedipine',
     '20 mg loading, then\n10-20 mg Q4-6h x 48 hrs',
     'Indomethacin\n(<32 weeks only)',
     '50-100 mg rectal/oral load,\nthen 25-50 mg Q6h',
     'Goal: 48-hr window for corticosteroids. Not for long-term use'],

    ['Fetal Lung Maturity\n(24-34 weeks)',
     'Betamethasone',
     '12 mg IM x 2 doses,\n24 hrs apart',
     'Dexamethasone',
     '6 mg IM Q12h x 4 doses',
     'Single rescue course if <34 wks & >14 days after first course'],

    ['GBS Prophylaxis\nin labour',
     'IV Penicillin G',
     '5 MU IV load, then\n2.5 MU IV Q4h until delivery',
     'IV Ampicillin OR\nClindamycin (PCN allergy)',
     'Ampicillin 2g load, then 1g Q4h\nClindamycin 900 mg IV Q8h',
     'If anaphylaxis risk: Vancomycin 1g IV Q12h'],

    ['PPH Prevention\n(3rd stage)',
     'Oxytocin IM',
     '10 IU IM after delivery\nof anterior shoulder',
     'Carbetocin\n(if cold chain unavailable)',
     '100 mcg IM single dose',
     'WHO 2024-2025 consolidated PPH guidelines (51 recommendations)'],

    ['PPH Treatment\n(active bleeding)',
     'Oxytocin IV +\nuterine massage',
     '20-40 IU in 500 mL NS\nIV infusion + bimanual compression',
     'Tranexamic acid +\nMisoprostol',
     'TXA 1g IV (within 3 hrs);\nMisoprostol 800 mcg sublingual',
     'WHO 2024. Escalate: balloon tamponade, compression sutures, hysterectomy'],

    ['Hypothyroidism\nin pregnancy',
     'Levothyroxine',
     '1.6 mcg/kg/day OD on\nempty stomach',
     'Dose titration upward\n(+25-30% increase)',
     'Increase dose as soon as\npregnancy confirmed',
     'TSH target <2.5 mIU/L (1st trimester). ATA 2026 guidelines'],

    ['Hyperthyroidism\nin pregnancy\n(1st trimester)',
     'Propylthiouracil (PTU)',
     '100-150 mg TDS',
     'Methimazole\n(switch after 16 wks)',
     '5-20 mg OD-BD',
     'PTU preferred in 1st trimester (methimazole teratogenic). ATA 2026'],

    ['Postpartum Depression',
     'SSRI\n(Sertraline / Escitalopram)',
     'Sertraline 50-150 mg OD',
     'Zuranolone (oral) or\nBrexanolone (IV)',
     'Zuranolone 50 mg OD x 14 days',
     'Screen with EPDS each trimester + 6 wks postpartum. CANMAT 2024'],
]


# ── Surgical escalation table ────────────────────────────────────────────────
SURGICAL_HEADERS = ['Condition', 'When Medical Lines Fail -> Surgical Escalation']
SURGICAL_COL_W = [4.5*cm, 21.5*cm]
SURGICAL_ROWS = [
    ['AUB / Menorrhagia',        'Endometrial ablation  →  Hysterectomy'],
    ['Uterine Fibroids',         'Myomectomy (hysteroscopic / laparoscopic)  →  UAE (uterine artery embolisation)  →  Hysterectomy'],
    ['Endometriosis',            'Laparoscopic lesion excision / ablation  →  Hysterectomy + bilateral salpingo-oophorectomy (BSO)'],
    ['Severe PMDD',              'Bilateral oophorectomy (last resort, after exhausting all medical options)'],
    ['Ectopic Pregnancy',        'Salpingotomy (fertility-sparing)  →  Salpingectomy'],
    ['PPH (refractory)',         'Uterine compression sutures (B-Lynch)  →  Uterine / iliac artery devascularisation  →  Peripartum hysterectomy'],
]


# ── Abbreviations ─────────────────────────────────────────────────────────────
ABBREV_TEXT = (
    "<b>Abbreviations:</b> "
    "AUB = Abnormal Uterine Bleeding | BD = Twice daily | BSO = Bilateral salpingo-oophorectomy | "
    "COC/OCP = Combined oral contraceptive | E2 = Estradiol | EE = Ethinyl estradiol | "
    "ER = Extended release | IM = Intramuscular | IR = Insulin resistance | "
    "IV = Intravenous | LNG-IUD = Levonorgestrel intrauterine device | "
    "MPA = Medroxyprogesterone acetate | NS = Normal saline | OD = Once daily | "
    "PID = Pelvic inflammatory disease | PCOS = Polycystic ovary syndrome | "
    "POI = Premature ovarian insufficiency | PPH = Postpartum haemorrhage | "
    "PTU = Propylthiouracil | QID = Four times daily | TDS/TID = Three times daily | "
    "TXA = Tranexamic acid | UAE = Uterine artery embolisation"
)

SOURCES_TEXT = (
    "<b>Sources:</b> ACOG CPG #6 (2023), #7 (2023), #8 (2024), #9 (2025), #10 (2025), #11 (2026) | "
    "ESHRE/ASRM PCOS International Guideline 2023 (PMID 37580861) | "
    "ESHRE POI Guideline 2024 (PMID 39647506) | "
    "SOMANZ Hypertension in Pregnancy 2023 (PMID 38763516) | "
    "WHO PPH Consolidated Guidelines 2024-2025 | "
    "ATA Thyroid Disease in Pregnancy 2026 (PMID 42219800) | "
    "CANMAT Perinatal Mood Disorders 2024 (PMID 39936923) | "
    "Berek & Novak's Gynecology | Harrison's Principles of Internal Medicine 22E | "
    "Rosen's Emergency Medicine"
)


# ── Page header / footer callbacks ───────────────────────────────────────────
def on_first_page(canvas, doc):
    _draw_header(canvas, doc)
    _draw_footer(canvas, doc)

def on_later_pages(canvas, doc):
    _draw_header_small(canvas, doc)
    _draw_footer(canvas, doc)

def _draw_header(canvas, doc):
    W, H = landscape(A4)
    # Banner background
    canvas.setFillColor(TEAL)
    canvas.rect(0, H - 3.2*cm, W, 3.2*cm, fill=1, stroke=0)
    # Red cross icon placeholder (simple cross)
    canvas.setFillColor(WHITE)
    canvas.setFont('Helvetica-Bold', 24)
    canvas.drawString(1.0*cm, H - 2.1*cm, '+')
    # Title
    canvas.setFont('Helvetica-Bold', 19)
    canvas.setFillColor(WHITE)
    canvas.drawCentredString(W/2, H - 1.7*cm, 'OBGyn Quick-Reference: First-Line vs Second-Line Treatments')
    # Subtitle
    canvas.setFont('Helvetica', 9)
    canvas.setFillColor(HexColor('#AADDDD'))
    canvas.drawCentredString(W/2, H - 2.4*cm, 'Gynaecology  |  Obstetrics  |  Evidence-Based  |  2023-2026 Guidelines')
    # Date stamp
    canvas.setFont('Helvetica-Oblique', 7.5)
    canvas.setFillColor(HexColor('#AADDDD'))
    canvas.drawRightString(W - 1.0*cm, H - 2.4*cm,
        f'Generated: {datetime.date.today().strftime("%d %b %Y")}')

def _draw_header_small(canvas, doc):
    W, H = landscape(A4)
    canvas.setFillColor(TEAL)
    canvas.rect(0, H - 1.4*cm, W, 1.4*cm, fill=1, stroke=0)
    canvas.setFont('Helvetica-Bold', 10)
    canvas.setFillColor(WHITE)
    canvas.drawCentredString(W/2, H - 1.0*cm,
        'OBGyn Quick-Reference: First-Line vs Second-Line Treatments')

def _draw_footer(canvas, doc):
    W, H = landscape(A4)
    canvas.setFillColor(GREY_LIGHT)
    canvas.rect(0, 0, W, 0.9*cm, fill=1, stroke=0)
    canvas.setFont('Helvetica-Oblique', 7)
    canvas.setFillColor(GREY_MID)
    canvas.drawString(1.0*cm, 0.32*cm,
        'For clinical use. Always verify doses against local formulary and current guidelines. Not a substitute for clinical judgement.')
    canvas.drawRightString(W - 1.0*cm, 0.32*cm,
        f'Page {doc.page}')


# ── Build document ────────────────────────────────────────────────────────────
def build():
    doc = SimpleDocTemplate(
        OUTPUT,
        pagesize=landscape(A4),
        leftMargin=1.0*cm,
        rightMargin=1.0*cm,
        topMargin=3.6*cm,       # space for header banner
        bottomMargin=1.2*cm,    # space for footer
    )
    doc.onFirstPage = on_first_page
    doc.onLaterPages = on_later_pages

    story = []

    # ── GYNAECOLOGY section ──────────────────────────────────────────────────
    story.append(section_header('GYNAECOLOGY', TEAL))
    story.append(Spacer(1, 4))
    story.append(make_table(GYNAECOLOGY_HEADERS, GYN_ROWS, GYN_COL_W))
    story.append(Spacer(1, 10))

    # ── OBSTETRICS section ───────────────────────────────────────────────────
    story.append(PageBreak())
    story.append(section_header('OBSTETRICS', HexColor('#2A6496')))
    story.append(Spacer(1, 4))
    story.append(make_table(OBSTETRICS_HEADERS, OBS_ROWS, OBS_COL_W))
    story.append(Spacer(1, 10))

    # ── Surgical escalation ──────────────────────────────────────────────────
    story.append(section_header('SURGICAL ESCALATION (When All Medical Lines Fail)', ORANGE))
    story.append(Spacer(1, 4))
    surg_t = make_table(SURGICAL_HEADERS, SURGICAL_ROWS, SURGICAL_COL_W)
    story.append(surg_t)
    story.append(Spacer(1, 8))

    # ── Abbreviations + Sources ───────────────────────────────────────────────
    abbrev_p = Paragraph(ABBREV_TEXT, note_style)
    sources_p = Paragraph(SOURCES_TEXT, note_style)
    info_data = [[abbrev_p], [sources_p]]
    info_t = Table(info_data, colWidths=[26*cm])
    info_t.setStyle(TableStyle([
        ('BACKGROUND',    (0, 0), (-1, 0), GREY_LIGHT),
        ('BACKGROUND',    (0, 1), (-1, 1), HexColor('#FFF8E1')),
        ('TOPPADDING',    (0, 0), (-1, -1), 5),
        ('BOTTOMPADDING', (0, 0), (-1, -1), 5),
        ('LEFTPADDING',   (0, 0), (-1, -1), 8),
        ('RIGHTPADDING',  (0, 0), (-1, -1), 8),
        ('BOX',           (0, 0), (-1, -1), 0.5, HexColor('#CCCCCC')),
        ('LINEBELOW',     (0, 0), (-1, 0),  0.5, HexColor('#CCCCCC')),
    ]))
    story.append(info_t)

    doc.build(story, onFirstPage=on_first_page, onLaterPages=on_later_pages)
    print(f'PDF saved to: {OUTPUT}')


if __name__ == '__main__':
    build()