from reportlab.lib.pagesizes import A4
from reportlab.lib import colors
from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
from reportlab.lib.units import mm, cm
from reportlab.platypus import (
    SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle,
    HRFlowable, KeepTogether
)
from reportlab.lib.enums import TA_CENTER, TA_LEFT, TA_JUSTIFY
from reportlab.pdfbase import pdfmetrics
from reportlab.pdfbase.ttfonts import TTFont
import os

OUTPUT = "/home/daytona/workspace/nutrition-pdf/Kwashiorkor_vs_Marasmus_QuickRef.pdf"

doc = SimpleDocTemplate(
    OUTPUT,
    pagesize=A4,
    leftMargin=15*mm,
    rightMargin=15*mm,
    topMargin=12*mm,
    bottomMargin=12*mm,
)

W, H = A4

# ── Colour palette ──────────────────────────────────────────────────
C_GREEN   = colors.HexColor("#1B7A3E")   # Kwashiorkor header
C_ORANGE  = colors.HexColor("#C0550A")   # Marasmus header
C_BLUE    = colors.HexColor("#1A4D8F")   # Section titles
C_GOLD    = colors.HexColor("#F0A500")   # Mnemonic highlight
C_PINK    = colors.HexColor("#FFE4EC")   # Kwashiorkor row bg
C_PEACH   = colors.HexColor("#FFF3E0")   # Marasmus row bg
C_LTBLUE  = colors.HexColor("#E8F0FB")   # Alternating row
C_DKGRAY  = colors.HexColor("#2C2C2C")
C_WHITE   = colors.white
C_LTGREEN = colors.HexColor("#E6F5EC")
C_LTORANGE= colors.HexColor("#FFF0E6")

# ── Styles ──────────────────────────────────────────────────────────
styles = getSampleStyleSheet()

def S(name, **kw):
    return ParagraphStyle(name, **kw)

title_style = S("Title",
    fontSize=20, leading=24, alignment=TA_CENTER,
    textColor=C_WHITE, fontName="Helvetica-Bold",
    spaceAfter=2)

subtitle_style = S("Subtitle",
    fontSize=11, leading=14, alignment=TA_CENTER,
    textColor=colors.HexColor("#FFE680"), fontName="Helvetica-Oblique",
    spaceAfter=4)

section_style = S("Section",
    fontSize=12, leading=15, alignment=TA_LEFT,
    textColor=C_WHITE, fontName="Helvetica-Bold",
    spaceBefore=6, spaceAfter=3,
    leftIndent=4, borderPad=4)

body_style = S("Body",
    fontSize=8.5, leading=11.5, alignment=TA_LEFT,
    textColor=C_DKGRAY, fontName="Helvetica",
    spaceBefore=1, spaceAfter=1)

bold_body = S("BoldBody",
    fontSize=8.5, leading=11.5, alignment=TA_LEFT,
    textColor=C_DKGRAY, fontName="Helvetica-Bold")

mnemonic_style = S("Mnemonic",
    fontSize=9, leading=13, alignment=TA_CENTER,
    textColor=colors.HexColor("#7A3800"), fontName="Helvetica-BoldOblique",
    spaceBefore=2, spaceAfter=2,
    backColor=colors.HexColor("#FFF8DC"),
    borderColor=C_GOLD, borderWidth=1, borderPad=5)

footer_style = S("Footer",
    fontSize=7, leading=9, alignment=TA_CENTER,
    textColor=colors.HexColor("#888888"), fontName="Helvetica-Oblique")

cell_hdr = S("CellHdr",
    fontSize=9, leading=11, alignment=TA_CENTER,
    textColor=C_WHITE, fontName="Helvetica-Bold")

cell_body = S("CellBody",
    fontSize=8, leading=10.5, alignment=TA_LEFT,
    textColor=C_DKGRAY, fontName="Helvetica")

cell_body_c = S("CellBodyC",
    fontSize=8, leading=10.5, alignment=TA_CENTER,
    textColor=C_DKGRAY, fontName="Helvetica")

cell_bold_c = S("CellBoldC",
    fontSize=8.2, leading=11, alignment=TA_CENTER,
    textColor=C_DKGRAY, fontName="Helvetica-Bold")

cell_green = S("CellGreen",
    fontSize=8, leading=10.5, alignment=TA_CENTER,
    textColor=C_GREEN, fontName="Helvetica-Bold")

cell_orange = S("CellOrange",
    fontSize=8, leading=10.5, alignment=TA_CENTER,
    textColor=C_ORANGE, fontName="Helvetica-Bold")

# ── Helper to make coloured section header ──────────────────────────
def section_header(text, bg=C_BLUE):
    tbl = Table([[Paragraph(text, section_style)]], colWidths=[W - 30*mm])
    tbl.setStyle(TableStyle([
        ("BACKGROUND", (0,0), (-1,-1), bg),
        ("ROUNDEDCORNERS", [4]),
        ("TOPPADDING", (0,0), (-1,-1), 5),
        ("BOTTOMPADDING", (0,0), (-1,-1), 5),
        ("LEFTPADDING", (0,0), (-1,-1), 8),
    ]))
    return tbl

# ══════════════════════════════════════════════════════════════════
# BUILD CONTENT
# ══════════════════════════════════════════════════════════════════
story = []

# ── TITLE BANNER ────────────────────────────────────────────────────
banner_data = [[
    Paragraph("🍽️  KWASHIORKOR  vs  MARASMUS", title_style),
]]
banner_data.append([
    Paragraph("Quick Reference Card — PSM / Community Medicine  |  Source: Park's PSM + Robbins Pathology", subtitle_style)
])

banner = Table(banner_data, colWidths=[W - 30*mm])
banner.setStyle(TableStyle([
    ("BACKGROUND", (0,0), (-1,-1), C_BLUE),
    ("ROUNDEDCORNERS", [6]),
    ("TOPPADDING", (0,0), (-1,-1), 8),
    ("BOTTOMPADDING", (0,0), (-1,-1), 8),
    ("LEFTPADDING", (0,0), (-1,-1), 10),
    ("RIGHTPADDING", (0,0), (-1,-1), 10),
]))
story.append(banner)
story.append(Spacer(1, 6))

# ── CONCEPT BOX ─────────────────────────────────────────────────────
concept_rows = [
    [
        Paragraph("KWASHIORKOR", S("kh", fontSize=11, fontName="Helvetica-Bold",
                  textColor=C_WHITE, alignment=TA_CENTER)),
        Paragraph("CONCEPT", S("cc", fontSize=10, fontName="Helvetica-Bold",
                  textColor=C_DKGRAY, alignment=TA_CENTER)),
        Paragraph("MARASMUS", S("mh", fontSize=11, fontName="Helvetica-Bold",
                  textColor=C_WHITE, alignment=TA_CENTER)),
    ],
    [
        Paragraph("Mainly <b>PROTEIN</b> deficient<br/>(some calories present)", cell_body_c),
        Paragraph("Deficiency", cell_bold_c),
        Paragraph("<b>PROTEIN + CALORIES</b><br/>both deficient (starvation)", cell_body_c),
    ],
    [
        Paragraph("Child given only rice/carbs<br/>after weaning from breast", cell_body_c),
        Paragraph("Story", cell_bold_c),
        Paragraph("Total starvation,<br/>early cessation of breastfeeding", cell_body_c),
    ],
    [
        Paragraph("<b>1 – 5 years</b><br/>(after weaning)", cell_green),
        Paragraph("Age", cell_bold_c),
        Paragraph("<b>&lt; 1 year</b><br/>(infancy)", cell_orange),
    ],
]

concept_tbl = Table(concept_rows, colWidths=[72*mm, 36*mm, 72*mm])
concept_tbl.setStyle(TableStyle([
    ("BACKGROUND",    (0,0), (0,0), C_GREEN),
    ("BACKGROUND",    (2,0), (2,0), C_ORANGE),
    ("BACKGROUND",    (1,0), (1,0), colors.HexColor("#DDDDDD")),
    ("BACKGROUND",    (0,1), (0,-1), C_LTGREEN),
    ("BACKGROUND",    (2,1), (2,-1), C_LTORANGE),
    ("BACKGROUND",    (1,1), (1,-1), colors.HexColor("#F5F5F5")),
    ("GRID",          (0,0), (-1,-1), 0.5, colors.HexColor("#CCCCCC")),
    ("TOPPADDING",    (0,0), (-1,-1), 5),
    ("BOTTOMPADDING", (0,0), (-1,-1), 5),
    ("LEFTPADDING",   (0,0), (-1,-1), 6),
    ("RIGHTPADDING",  (0,0), (-1,-1), 6),
    ("ALIGN",         (0,0), (-1,-1), "CENTER"),
    ("VALIGN",        (0,0), (-1,-1), "MIDDLE"),
]))
story.append(concept_tbl)
story.append(Spacer(1, 6))

# ── MAIN COMPARISON TABLE ───────────────────────────────────────────
story.append(section_header("📊  MASTER COMPARISON TABLE"))
story.append(Spacer(1, 3))

headers = [
    Paragraph("Feature", cell_hdr),
    Paragraph("KWASHIORKOR", S("kh2", fontSize=9, fontName="Helvetica-Bold",
              textColor=C_WHITE, alignment=TA_CENTER)),
    Paragraph("MARASMUS", S("mh2", fontSize=9, fontName="Helvetica-Bold",
              textColor=C_WHITE, alignment=TA_CENTER)),
]

rows = [
    ["Body Weight",         "60–80% of normal (edema masks true loss)",         "< 60% of normal (severe wasting)"],
    ["EDEMA",               "✅ PRESENT — hallmark sign!\n(face, legs, ascites)", "❌ ABSENT"],
    ["Muscle Wasting",      "Mild — relatively SPARED",                           "SEVERE — skin and bones"],
    ["Subcutaneous Fat",    "Relatively preserved (hidden by edema)",              "Completely ABSENT"],
    ["Face",                "\"Moon face\" — puffy, round, swollen",              "\"Old man face\" — sunken, wrinkled"],
    ["Skin",                "\"Flaky paint\" dermatosis\n(hyper + desquamation + hypopigmentation)", "Dry, loose, wrinkled"],
    ["Hair",                "FLAG SIGN — alternating light & dark bands,\nthin, easily pluckable", "Sparse, dry, brittle"],
    ["Liver",               "✅ ENLARGED — fatty liver (hepatomegaly)",           "Normal size"],
    ["Serum Albumin",       "Very LOW (hypoalbuminaemia → edema)",                "Low/normal"],
    ["Behaviour",           "Apathetic, irritable, anorexic, miserable",          "Alert, looks hungry, wants food"],
    ["Appetite",            "POOR / lost",                                         "PRESERVED"],
    ["Immunity",            "Impaired, secondary infections common",               "Impaired, secondary infections common"],
    ["Prognosis",           "Worse — more complications",                          "Slightly better if treated early"],
]

tbl_data = [headers]
for r in rows:
    tbl_data.append([
        Paragraph(r[0], bold_body),
        Paragraph(r[1], cell_body_c),
        Paragraph(r[2], cell_body_c),
    ])

col_w = [36*mm, 80*mm, 64*mm]
main_tbl = Table(tbl_data, colWidths=col_w)

ts = TableStyle([
    # Header row
    ("BACKGROUND",    (0,0), (0,0), C_BLUE),
    ("BACKGROUND",    (1,0), (1,0), C_GREEN),
    ("BACKGROUND",    (2,0), (2,0), C_ORANGE),
    ("TEXTCOLOR",     (0,0), (-1,0), C_WHITE),
    ("FONTNAME",      (0,0), (-1,0), "Helvetica-Bold"),
    ("FONTSIZE",      (0,0), (-1,0), 9),
    # Alternating rows
    ("GRID",          (0,0), (-1,-1), 0.4, colors.HexColor("#CCCCCC")),
    ("TOPPADDING",    (0,0), (-1,-1), 4),
    ("BOTTOMPADDING", (0,0), (-1,-1), 4),
    ("LEFTPADDING",   (0,0), (-1,-1), 5),
    ("RIGHTPADDING",  (0,0), (-1,-1), 5),
    ("VALIGN",        (0,0), (-1,-1), "MIDDLE"),
    # EDEMA row highlight
    ("BACKGROUND",    (1,2), (1,2), colors.HexColor("#C8F5D8")),
    ("BACKGROUND",    (2,2), (2,2), colors.HexColor("#FFE0E0")),
    ("FONTNAME",      (0,2), (-1,2), "Helvetica-Bold"),
])

for i in range(1, len(tbl_data)):
    if i % 2 == 0:
        ts.add("BACKGROUND", (0,i), (0,i), colors.HexColor("#F0F0F0"))
    else:
        ts.add("BACKGROUND", (0,i), (0,i), colors.HexColor("#FAFAFA"))

main_tbl.setStyle(ts)
story.append(main_tbl)
story.append(Spacer(1, 6))

# ── TWO COLUMN: WHY EDEMA + WHY FATTY LIVER ─────────────────────────
story.append(section_header("🧠  PATHOPHYSIOLOGY (WHY?)"))
story.append(Spacer(1, 3))

edema_text = [
    Paragraph("<b>WHY EDEMA in Kwashiorkor?</b>", S("eh", fontSize=9,
              fontName="Helvetica-Bold", textColor=C_GREEN, alignment=TA_CENTER)),
    Spacer(1, 3),
    Paragraph("No protein intake", cell_body_c),
    Paragraph("⬇", cell_body_c),
    Paragraph("Liver can't make <b>Albumin</b>", cell_body_c),
    Paragraph("⬇", cell_body_c),
    Paragraph("Low oncotic pressure in blood", cell_body_c),
    Paragraph("⬇", cell_body_c),
    Paragraph("<b>Water leaks into tissues</b>", cell_body_c),
    Paragraph("⬇", cell_body_c),
    Paragraph("<b>EDEMA ✅</b>", S("ee", fontSize=9, fontName="Helvetica-Bold",
              textColor=C_GREEN, alignment=TA_CENTER)),
]

liver_text = [
    Paragraph("<b>WHY FATTY LIVER in Kwashiorkor?</b>", S("lh", fontSize=9,
              fontName="Helvetica-Bold", textColor=C_ORANGE, alignment=TA_CENTER)),
    Spacer(1, 3),
    Paragraph("No protein intake", cell_body_c),
    Paragraph("⬇", cell_body_c),
    Paragraph("Can't make <b>Lipoprotein carrier</b>", cell_body_c),
    Paragraph("⬇", cell_body_c),
    Paragraph("Fat made but <b>can't be exported</b>", cell_body_c),
    Paragraph("⬇", cell_body_c),
    Paragraph("<b>Fat accumulates in liver cells</b>", cell_body_c),
    Paragraph("⬇", cell_body_c),
    Paragraph("<b>FATTY LIVER ✅</b>", S("fl", fontSize=9, fontName="Helvetica-Bold",
              textColor=C_ORANGE, alignment=TA_CENTER)),
]

path_tbl = Table(
    [[edema_text, liver_text]],
    colWidths=[87*mm, 87*mm]
)
path_tbl.setStyle(TableStyle([
    ("VALIGN",        (0,0), (-1,-1), "TOP"),
    ("LEFTPADDING",   (0,0), (-1,-1), 8),
    ("RIGHTPADDING",  (0,0), (-1,-1), 8),
    ("TOPPADDING",    (0,0), (-1,-1), 6),
    ("BOTTOMPADDING", (0,0), (-1,-1), 6),
    ("BACKGROUND",    (0,0), (0,0), C_LTGREEN),
    ("BACKGROUND",    (1,0), (1,0), C_LTORANGE),
    ("GRID",          (0,0), (-1,-1), 0.5, colors.HexColor("#CCCCCC")),
]))
story.append(path_tbl)
story.append(Spacer(1, 6))

# ── WELLCOME CLASSIFICATION ──────────────────────────────────────────
story.append(section_header("📋  WELLCOME CLASSIFICATION (Must Know for MCQ!)"))
story.append(Spacer(1, 3))

wc_data = [
    [
        Paragraph("Weight for Age", cell_hdr),
        Paragraph("WITH Edema", cell_hdr),
        Paragraph("WITHOUT Edema", cell_hdr),
    ],
    [
        Paragraph("60 – 80% of normal", cell_body_c),
        Paragraph("KWASHIORKOR", S("wk", fontSize=9, fontName="Helvetica-Bold",
                  textColor=C_GREEN, alignment=TA_CENTER)),
        Paragraph("Undernutrition", cell_body_c),
    ],
    [
        Paragraph("< 60% of normal", cell_body_c),
        Paragraph("MARASMIC-KWASHIORKOR\n(Mixed)", S("wmk", fontSize=8.5,
                  fontName="Helvetica-Bold", textColor=C_BLUE, alignment=TA_CENTER)),
        Paragraph("MARASMUS", S("wm", fontSize=9, fontName="Helvetica-Bold",
                  textColor=C_ORANGE, alignment=TA_CENTER)),
    ],
]

wc_tbl = Table(wc_data, colWidths=[50*mm, 72*mm, 58*mm])
wc_tbl.setStyle(TableStyle([
    ("BACKGROUND",    (0,0), (-1,0), C_BLUE),
    ("TEXTCOLOR",     (0,0), (-1,0), C_WHITE),
    ("BACKGROUND",    (0,1), (-1,1), C_LTGREEN),
    ("BACKGROUND",    (0,2), (-1,2), C_LTORANGE),
    ("GRID",          (0,0), (-1,-1), 0.5, colors.HexColor("#CCCCCC")),
    ("ALIGN",         (0,0), (-1,-1), "CENTER"),
    ("VALIGN",        (0,0), (-1,-1), "MIDDLE"),
    ("TOPPADDING",    (0,0), (-1,-1), 6),
    ("BOTTOMPADDING", (0,0), (-1,-1), 6),
]))
story.append(wc_tbl)
story.append(Spacer(1, 6))

# ── MUAC + SAM BOX ──────────────────────────────────────────────────
muac_data = [
    [
        Paragraph("MUAC (Mid-Upper Arm Circumference)", S("mh3",
                  fontSize=9, fontName="Helvetica-Bold", textColor=C_WHITE,
                  alignment=TA_CENTER)),
        Paragraph("SAM Definition (WHO)", S("sh",
                  fontSize=9, fontName="Helvetica-Bold", textColor=C_WHITE,
                  alignment=TA_CENTER)),
    ],
    [
        Paragraph(
            "• <b>&gt; 13.5 cm</b> → Normal ✅<br/>"
            "• <b>12.5 – 13.5 cm</b> → Mild-Moderate malnutrition ⚠️<br/>"
            "• <b>&lt; 12.5 cm</b> → Severe malnutrition ❌<br/>"
            "• <b>&lt; 11.5 cm</b> → SAM (critical) 🚨",
            cell_body),
        Paragraph(
            "Any ONE of the following:<br/>"
            "• Weight-for-height Z-score <b>&lt; −3SD</b><br/>"
            "• MUAC <b>&lt; 115 mm</b><br/>"
            "• <b>Nutritional oedema</b> present<br/>"
            "Severely wasted = <b>9× more likely to die</b>",
            cell_body),
    ]
]

muac_tbl = Table(muac_data, colWidths=[87*mm, 87*mm])
muac_tbl.setStyle(TableStyle([
    ("BACKGROUND",    (0,0), (0,0), C_GREEN),
    ("BACKGROUND",    (1,0), (1,0), C_ORANGE),
    ("BACKGROUND",    (0,1), (0,1), C_LTGREEN),
    ("BACKGROUND",    (1,1), (1,1), C_LTORANGE),
    ("GRID",          (0,0), (-1,-1), 0.5, colors.HexColor("#CCCCCC")),
    ("VALIGN",        (0,0), (-1,-1), "TOP"),
    ("TOPPADDING",    (0,0), (-1,-1), 5),
    ("BOTTOMPADDING", (0,0), (-1,-1), 5),
    ("LEFTPADDING",   (0,0), (-1,-1), 7),
    ("RIGHTPADDING",  (0,0), (-1,-1), 7),
]))
story.append(muac_tbl)
story.append(Spacer(1, 6))

# ── MNEMONICS ───────────────────────────────────────────────────────
story.append(section_header("💡  MNEMONICS & MEMORY TRICKS", bg=colors.HexColor("#8B4513")))
story.append(Spacer(1, 3))

mn_rows = [
    ["Kwashiorkor = PUFFY",
     "P-Protein def | U-Underweight (60–80%) | F-Fatty liver | F-Flag sign hair | Y-Young 1–5yr"],
    ["Marasmus = WASTE",
     "W-Wasting severe | A-Alert child | S-Subcutaneous fat gone | T-Total starvation | E-Emaciated"],
    ["K for Kwashiorkor",
     "K = Keeps water (edema) | K = keeps fat in liver | K = Kids 1-5yr"],
    ["Moon face vs Old man",
     "Kwashiorkor = MOON face (round, puffy). Marasmus = OLD MAN face (sunken, wrinkled)"],
    ["Flaky paint skin",
     "ONLY Kwashiorkor. Imagine peeling paint on a wall — that's the skin rash pattern"],
    ["Flag sign",
     "ONLY Kwashiorkor. Hair bands = flag stripes. Each band = one episode of protein deficiency"],
]

mn_tbl_data = [[
    Paragraph("<b>Mnemonic</b>", cell_hdr),
    Paragraph("<b>Meaning</b>", cell_hdr),
]]
for r in mn_rows:
    mn_tbl_data.append([
        Paragraph(r[0], S("mnk", fontSize=8.2, fontName="Helvetica-Bold",
                  textColor=colors.HexColor("#5C3A00"))),
        Paragraph(r[1], cell_body),
    ])

mn_tbl = Table(mn_tbl_data, colWidths=[55*mm, 120*mm])
mn_tbl.setStyle(TableStyle([
    ("BACKGROUND",    (0,0), (-1,0), colors.HexColor("#8B4513")),
    ("TEXTCOLOR",     (0,0), (-1,0), C_WHITE),
    ("BACKGROUND",    (0,1), (-1,-1), colors.HexColor("#FFF8DC")),
    ("GRID",          (0,0), (-1,-1), 0.4, colors.HexColor("#DEB887")),
    ("TOPPADDING",    (0,0), (-1,-1), 4),
    ("BOTTOMPADDING", (0,0), (-1,-1), 4),
    ("LEFTPADDING",   (0,0), (-1,-1), 6),
    ("RIGHTPADDING",  (0,0), (-1,-1), 6),
    ("VALIGN",        (0,0), (-1,-1), "MIDDLE"),
    ("ROWBACKGROUNDS", (0,1), (-1,-1),
     [colors.HexColor("#FFF8DC"), colors.HexColor("#FFF0C0")]),
]))
story.append(mn_tbl)
story.append(Spacer(1, 6))

# ── 3-LINE EXAM SUMMARY ─────────────────────────────────────────────
story.append(section_header("✍️  3-LINE EXAM SUMMARY (Write this first in your answer!)", bg=C_BLUE))
story.append(Spacer(1, 3))

summary_items = [
    ("1.", "Kwashiorkor",
     "= Mainly PROTEIN deficient, age 1–5 yrs. Key features: EDEMA + Moon face + Fatty liver "
     "+ Flaky paint skin + Flag sign hair + Apathy. Child looks puffy but is wasted underneath."),
    ("2.", "Marasmus",
     "= Total starvation (protein + calories), under 1 year. Key features: NO edema, severe wasting, "
     "Old man face, bones visible, alert child who wants to eat. Weight < 60% of normal."),
    ("3.", "Key difference",
     "EDEMA — present in Kwashiorkor, absent in Marasmus. "
     "Kwashiorkor also has fatty liver, flag sign hair, flaky paint skin which Marasmus does NOT have."),
]

for num, bold, rest in summary_items:
    row_data = [[
        Paragraph(num, S("sn", fontSize=11, fontName="Helvetica-Bold",
                  textColor=C_BLUE, alignment=TA_CENTER)),
        Paragraph(f"<b>{bold}</b>", S("sk", fontSize=9, fontName="Helvetica-Bold",
                  textColor=C_BLUE, alignment=TA_CENTER)),
        Paragraph(rest, cell_body),
    ]]
    row_tbl = Table(row_data, colWidths=[8*mm, 35*mm, 134*mm])
    row_tbl.setStyle(TableStyle([
        ("VALIGN",        (0,0), (-1,-1), "MIDDLE"),
        ("TOPPADDING",    (0,0), (-1,-1), 5),
        ("BOTTOMPADDING", (0,0), (-1,-1), 5),
        ("LEFTPADDING",   (0,0), (-1,-1), 5),
        ("BACKGROUND",    (0,0), (-1,-1), colors.HexColor("#F0F4FF")),
        ("LINEBELOW",     (0,0), (-1,-1), 0.5, colors.HexColor("#CCCCCC")),
    ]))
    story.append(row_tbl)

story.append(Spacer(1, 5))

# ── FOOTER ──────────────────────────────────────────────────────────
story.append(HRFlowable(width="100%", thickness=0.5, color=colors.HexColor("#AAAAAA")))
story.append(Spacer(1, 2))
story.append(Paragraph(
    "Sources: Park's Textbook of Preventive &amp; Social Medicine | Robbins &amp; Kumar Basic Pathology | "
    "Harper's Illustrated Biochemistry  •  Generated by Orris AI for last-minute PSM revision",
    footer_style))

# ── BUILD ────────────────────────────────────────────────────────────
doc.build(story)
print(f"PDF created: {OUTPUT}")