from reportlab.lib.pagesizes import A4
from reportlab.lib import colors
from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
from reportlab.lib.units import cm
from reportlab.platypus import (SimpleDocTemplate, Paragraph, Spacer, Table,
                                 TableStyle, HRFlowable)
from reportlab.lib.enums import TA_CENTER, TA_LEFT

OUTPUT = "/home/daytona/workspace/febrile-seizures/Febrile_Seizures_OSCE.pdf"

doc = SimpleDocTemplate(
    OUTPUT,
    pagesize=A4,
    rightMargin=1.5*cm, leftMargin=1.5*cm,
    topMargin=1.5*cm, bottomMargin=1.5*cm
)

# ── Colour palette ──────────────────────────────────────────────
BLUE      = colors.HexColor("#1A5276")
LIGHT_BLUE= colors.HexColor("#D6EAF8")
ORANGE    = colors.HexColor("#D35400")
RED_LIGHT = colors.HexColor("#FADBD8")
GREEN     = colors.HexColor("#1E8449")
GREEN_LT  = colors.HexColor("#D5F5E3")
GREY_LT   = colors.HexColor("#F2F3F4")
WHITE     = colors.white
BLACK     = colors.black

styles = getSampleStyleSheet()

# Custom styles
def S(name, **kw):
    return ParagraphStyle(name, **kw)

title_style   = S("Title",   fontSize=18, textColor=WHITE,      leading=22, alignment=TA_CENTER, fontName="Helvetica-Bold")
h2_style      = S("H2",      fontSize=11, textColor=WHITE,      leading=14, fontName="Helvetica-Bold")
h3_style      = S("H3",      fontSize=10, textColor=BLUE,       leading=13, fontName="Helvetica-Bold")
body_style    = S("Body",    fontSize=9,  textColor=BLACK,      leading=13, fontName="Helvetica")
bold_style    = S("Bold",    fontSize=9,  textColor=BLACK,      leading=13, fontName="Helvetica-Bold")
small_style   = S("Small",   fontSize=8,  textColor=colors.HexColor("#555555"), leading=11, fontName="Helvetica")
red_style     = S("Red",     fontSize=9,  textColor=colors.HexColor("#C0392B"), leading=13, fontName="Helvetica-Bold")
green_style   = S("Green",   fontSize=9,  textColor=GREEN,      leading=13, fontName="Helvetica-Bold")

story = []

# ════════════════════════════════════════════════
# TITLE BANNER
# ════════════════════════════════════════════════
title_table = Table(
    [[Paragraph("FEBRILE SEIZURES", title_style)],
     [Paragraph("Paediatrics OSCE Counselling — Quick Revision Handout", S("sub", fontSize=11, textColor=LIGHT_BLUE, leading=14, alignment=TA_CENTER, fontName="Helvetica"))]],
    colWidths=[18*cm]
)
title_table.setStyle(TableStyle([
    ("BACKGROUND", (0,0), (-1,-1), BLUE),
    ("ROUNDEDCORNERS", [6]),
    ("TOPPADDING",    (0,0), (-1,-1), 10),
    ("BOTTOMPADDING", (0,0), (-1,-1), 10),
    ("LEFTPADDING",   (0,0), (-1,-1), 12),
]))
story.append(title_table)
story.append(Spacer(1, 0.3*cm))

# ════════════════════════════════════════════════
# SECTION HELPER
# ════════════════════════════════════════════════
def section_header(text, bg=BLUE):
    t = Table([[Paragraph(text, h2_style)]], colWidths=[18*cm])
    t.setStyle(TableStyle([
        ("BACKGROUND",    (0,0), (-1,-1), bg),
        ("TOPPADDING",    (0,0), (-1,-1), 5),
        ("BOTTOMPADDING", (0,0), (-1,-1), 5),
        ("LEFTPADDING",   (0,0), (-1,-1), 8),
    ]))
    return t

def bullet(text, style=body_style):
    return Paragraph(f"• {text}", style)

def two_col_table(rows, col1_w=8*cm, col2_w=10*cm, header=None, header_bg=LIGHT_BLUE):
    data = []
    if header:
        data.append([Paragraph(header[0], bold_style), Paragraph(header[1], bold_style)])
    for r in rows:
        data.append([Paragraph(r[0], body_style), Paragraph(r[1], body_style)])
    t = Table(data, colWidths=[col1_w, col2_w])
    style_cmds = [
        ("GRID",          (0,0), (-1,-1), 0.4, colors.HexColor("#CCCCCC")),
        ("ROWBACKGROUNDS",(0,0), (-1,-1), [WHITE, GREY_LT]),
        ("TOPPADDING",    (0,0), (-1,-1), 4),
        ("BOTTOMPADDING", (0,0), (-1,-1), 4),
        ("LEFTPADDING",   (0,0), (-1,-1), 6),
        ("FONTSIZE",      (0,0), (-1,-1), 9),
        ("VALIGN",        (0,0), (-1,-1), "TOP"),
    ]
    if header:
        style_cmds += [
            ("BACKGROUND",  (0,0), (-1,0), header_bg),
            ("FONTNAME",    (0,0), (-1,0), "Helvetica-Bold"),
        ]
    t.setStyle(TableStyle(style_cmds))
    return t

# ════════════════════════════════════════════════
# 1. DEFINITION & BASICS
# ════════════════════════════════════════════════
story.append(section_header("1. DEFINITION & BASICS"))
story.append(Spacer(1, 0.2*cm))

basics = [
    ["Definition", "Seizure triggered by fever (>38°C) in the absence of CNS infection or metabolic cause"],
    ["Incidence", "2–5% of children — most common seizure type in children"],
    ["Age group", "6 months to 6 years (peak: 18 months)"],
    ["Prognosis", "Benign in vast majority; does NOT cause brain damage"],
]
t = Table([[Paragraph(r[0], bold_style), Paragraph(r[1], body_style)] for r in basics],
          colWidths=[4*cm, 14*cm])
t.setStyle(TableStyle([
    ("GRID",          (0,0), (-1,-1), 0.4, colors.HexColor("#CCCCCC")),
    ("ROWBACKGROUNDS",(0,0), (-1,-1), [LIGHT_BLUE, WHITE]),
    ("TOPPADDING",    (0,0), (-1,-1), 4),
    ("BOTTOMPADDING", (0,0), (-1,-1), 4),
    ("LEFTPADDING",   (0,0), (-1,-1), 6),
    ("VALIGN",        (0,0), (-1,-1), "TOP"),
    ("FONTSIZE",      (0,0), (-1,-1), 9),
]))
story.append(t)
story.append(Spacer(1, 0.3*cm))

# ════════════════════════════════════════════════
# 2. SIMPLE vs COMPLEX
# ════════════════════════════════════════════════
story.append(section_header("2. SIMPLE vs COMPLEX FEBRILE SEIZURE"))
story.append(Spacer(1, 0.2*cm))

sc_data = [
    ["Feature", "Simple", "Complex"],
    ["Duration", "< 15 minutes", "> 15 minutes"],
    ["Type", "Generalised (whole body)", "Focal (one side of body)"],
    ["Recurrence (same illness)", "Once only", "More than once in 24 hrs"],
    ["Post-ictal state", "Brief / normal recovery", "May be prolonged"],
    ["Prognosis", "Excellent — benign", "Needs further evaluation"],
]
t = Table([[Paragraph(r[0], bold_style if i==0 else body_style),
            Paragraph(r[1], bold_style if i==0 else body_style),
            Paragraph(r[2], bold_style if i==0 else body_style)] for i, r in enumerate(sc_data)],
          colWidths=[4.5*cm, 6.5*cm, 7*cm])
t.setStyle(TableStyle([
    ("BACKGROUND",    (0,0), (-1,0), BLUE),
    ("TEXTCOLOR",     (0,0), (-1,0), WHITE),
    ("FONTNAME",      (0,0), (-1,0), "Helvetica-Bold"),
    ("GRID",          (0,0), (-1,-1), 0.4, colors.HexColor("#CCCCCC")),
    ("ROWBACKGROUNDS",(0,1), (-1,-1), [WHITE, GREY_LT]),
    ("TOPPADDING",    (0,0), (-1,-1), 4),
    ("BOTTOMPADDING", (0,0), (-1,-1), 4),
    ("LEFTPADDING",   (0,0), (-1,-1), 6),
    ("FONTSIZE",      (0,0), (-1,-1), 9),
    ("VALIGN",        (0,0), (-1,-1), "TOP"),
]))
story.append(t)
story.append(Spacer(1, 0.3*cm))

# ════════════════════════════════════════════════
# 3. COUNSELLING SCRIPT
# ════════════════════════════════════════════════
story.append(section_header("3. OSCE COUNSELLING — KEY POINTS TO COVER"))
story.append(Spacer(1, 0.2*cm))

counselling_points = [
    ("Reassure first", "\"This is common and does NOT cause brain damage.\""),
    ("Recurrence risk", "30–40% chance of recurrence. Higher risk: age <1 yr, family history, daycare attendance."),
    ("Epilepsy risk", "Only 2–5% risk of later epilepsy (general population: 1–2%). Low risk in simple FS."),
    ("During a seizure", "Place on side (recovery position) • Note timing • Loosen clothes • Do NOT put anything in mouth • Do NOT restrain"),
    ("Fever management", "Paracetamol 15 mg/kg/dose when fever starts • Keep hydrated • Light clothing"),
    ("Investigations", "Simple FS: No EEG, No MRI, No LP routinely. Only investigate to find fever source."),
    ("Follow up", "Regular follow-up with paediatrician. Record duration and nature of any future episode."),
]

for title, content in counselling_points:
    row_t = Table([[Paragraph(title, bold_style), Paragraph(content, body_style)]],
                  colWidths=[4.5*cm, 13.5*cm])
    row_t.setStyle(TableStyle([
        ("GRID",          (0,0), (-1,-1), 0.4, colors.HexColor("#CCCCCC")),
        ("BACKGROUND",    (0,0), (0,0),   LIGHT_BLUE),
        ("TOPPADDING",    (0,0), (-1,-1), 5),
        ("BOTTOMPADDING", (0,0), (-1,-1), 5),
        ("LEFTPADDING",   (0,0), (-1,-1), 6),
        ("VALIGN",        (0,0), (-1,-1), "TOP"),
        ("FONTSIZE",      (0,0), (-1,-1), 9),
    ]))
    story.append(row_t)

story.append(Spacer(1, 0.3*cm))

# ════════════════════════════════════════════════
# 4. RED FLAGS
# ════════════════════════════════════════════════
story.append(section_header("4. RED FLAGS — WHEN TO BRING CHILD TO HOSPITAL", bg=colors.HexColor("#C0392B")))
story.append(Spacer(1, 0.2*cm))

red_flags = [
    "Seizure lasts > 5 minutes",
    "More than one seizure in 24 hours",
    "Child remains unconscious / confused after seizure",
    "Focal seizure (only one side of body involved)",
    "Signs of meningitis: neck stiffness, non-blanching rash, high-pitched cry",
    "Child is under 6 months of age",
    "Child is not fully vaccinated",
]

rf_data = [[Paragraph(f"⚠  {flag}", S("RF", fontSize=9, textColor=colors.HexColor("#C0392B"), leading=13, fontName="Helvetica-Bold"))] for flag in red_flags]
rf_table = Table(rf_data, colWidths=[18*cm])
rf_table.setStyle(TableStyle([
    ("BACKGROUND",    (0,0), (-1,-1), RED_LIGHT),
    ("GRID",          (0,0), (-1,-1), 0.3, colors.HexColor("#E8B4B8")),
    ("TOPPADDING",    (0,0), (-1,-1), 4),
    ("BOTTOMPADDING", (0,0), (-1,-1), 4),
    ("LEFTPADDING",   (0,0), (-1,-1), 10),
]))
story.append(rf_table)
story.append(Spacer(1, 0.3*cm))

# ════════════════════════════════════════════════
# 5. DO / DO NOT TABLE
# ════════════════════════════════════════════════
story.append(section_header("5. DURING A SEIZURE — DO vs DO NOT"))
story.append(Spacer(1, 0.2*cm))

do_dont = [
    ["✔  Place child on their side (recovery position)", "✘  Put anything in the mouth"],
    ["✔  Note the time — how long it lasts",             "✘  Restrain the child"],
    ["✔  Loosen tight clothing",                         "✘  Give water/food during seizure"],
    ["✔  Stay calm and stay with the child",             "✘  Leave the child alone"],
    ["✔  Call emergency if seizure > 5 minutes",         "✘  Wait indefinitely — act fast"],
]
do_dont_table = Table(
    [[Paragraph(r[0], S("DO",   fontSize=9, textColor=GREEN, leading=13, fontName="Helvetica-Bold")),
      Paragraph(r[1], S("DONT", fontSize=9, textColor=colors.HexColor("#C0392B"), leading=13, fontName="Helvetica-Bold"))]
     for r in do_dont],
    colWidths=[9*cm, 9*cm]
)
do_dont_table.setStyle(TableStyle([
    ("GRID",          (0,0), (-1,-1), 0.4, colors.HexColor("#CCCCCC")),
    ("BACKGROUND",    (0,0), (0,-1), GREEN_LT),
    ("BACKGROUND",    (1,0), (1,-1), RED_LIGHT),
    ("TOPPADDING",    (0,0), (-1,-1), 5),
    ("BOTTOMPADDING", (0,0), (-1,-1), 5),
    ("LEFTPADDING",   (0,0), (-1,-1), 8),
    ("VALIGN",        (0,0), (-1,-1), "MIDDLE"),
]))
story.append(do_dont_table)
story.append(Spacer(1, 0.3*cm))

# ════════════════════════════════════════════════
# 6. OSCE CHECKLIST
# ════════════════════════════════════════════════
story.append(section_header("6. OSCE MARKING CHECKLIST", bg=GREEN))
story.append(Spacer(1, 0.2*cm))

checklist = [
    "Introduced yourself and established rapport",
    "Explained what a febrile seizure is (fever-triggered, not infection)",
    "Differentiated simple vs complex febrile seizure",
    "Stated recurrence risk (30–40%)",
    "Addressed epilepsy risk clearly (low, 2–5%)",
    "Gave first aid instructions for during a seizure",
    "Advised on fever management (paracetamol, fluids)",
    "Listed red flags / when to go to hospital",
    "Explained investigations (none needed for simple FS)",
    "Checked patient understanding and invited questions",
]

check_data = [[Paragraph("☐", bold_style), Paragraph(item, body_style)] for item in checklist]
check_table = Table(check_data, colWidths=[0.7*cm, 17.3*cm])
check_table.setStyle(TableStyle([
    ("ROWBACKGROUNDS",(0,0), (-1,-1), [GREEN_LT, WHITE]),
    ("GRID",          (0,0), (-1,-1), 0.3, colors.HexColor("#CCCCCC")),
    ("TOPPADDING",    (0,0), (-1,-1), 4),
    ("BOTTOMPADDING", (0,0), (-1,-1), 4),
    ("LEFTPADDING",   (0,0), (-1,-1), 6),
    ("VALIGN",        (0,0), (-1,-1), "MIDDLE"),
    ("FONTSIZE",      (0,0), (-1,-1), 9),
]))
story.append(check_table)
story.append(Spacer(1, 0.3*cm))

# ════════════════════════════════════════════════
# FOOTER
# ════════════════════════════════════════════════
story.append(HRFlowable(width="100%", thickness=0.5, color=BLUE))
story.append(Spacer(1, 0.1*cm))
story.append(Paragraph(
    "Paediatrics OSCE Revision  •  Febrile Seizures  •  Generated for MBBS University Exam",
    S("footer", fontSize=7.5, textColor=colors.HexColor("#888888"), alignment=TA_CENTER, leading=10, fontName="Helvetica")
))

# ════════════════════════════════════════════════
# BUILD
# ════════════════════════════════════════════════
doc.build(story)
print(f"PDF saved to: {OUTPUT}")