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, KeepTogether)
from reportlab.lib.enums import TA_CENTER, TA_LEFT
from reportlab.platypus import PageBreak

OUTPUT = "/home/daytona/workspace/nursing-notes/Antigen_Antibody_Reaction.pdf"

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

# ── Colour palette ──────────────────────────────────────────────
C_DARK_BLUE   = colors.HexColor("#1A3C6E")
C_BLUE        = colors.HexColor("#2563EB")
C_LIGHT_BLUE  = colors.HexColor("#DBEAFE")
C_RED         = colors.HexColor("#DC2626")
C_LIGHT_RED   = colors.HexColor("#FEE2E2")
C_GREEN       = colors.HexColor("#16A34A")
C_LIGHT_GREEN = colors.HexColor("#DCFCE7")
C_ORANGE      = colors.HexColor("#EA580C")
C_LIGHT_ORANGE= colors.HexColor("#FFF7ED")
C_PURPLE      = colors.HexColor("#7C3AED")
C_LIGHT_PURPLE= colors.HexColor("#EDE9FE")
C_YELLOW      = colors.HexColor("#CA8A04")
C_LIGHT_YELLOW= colors.HexColor("#FEF9C3")
C_GRAY        = colors.HexColor("#F1F5F9")
C_DARK_GRAY   = colors.HexColor("#374151")
C_WHITE       = colors.white

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

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

title_style = S("Title", fontSize=20, textColor=C_WHITE,
                fontName="Helvetica-Bold", alignment=TA_CENTER, spaceAfter=4)
subtitle_style = S("Subtitle", fontSize=11, textColor=C_LIGHT_BLUE,
                   fontName="Helvetica", alignment=TA_CENTER, spaceAfter=2)
tag_style = S("Tag", fontSize=9, textColor=C_WHITE,
              fontName="Helvetica-Bold", alignment=TA_CENTER)

h2_style = S("H2", fontSize=13, textColor=C_DARK_BLUE,
             fontName="Helvetica-Bold", spaceBefore=10, spaceAfter=4)
h3_style = S("H3", fontSize=11, textColor=C_BLUE,
             fontName="Helvetica-Bold", spaceBefore=6, spaceAfter=3)

body = S("Body", fontSize=9.5, textColor=C_DARK_GRAY,
         fontName="Helvetica", leading=14, spaceAfter=3)
body_bold = S("BodyBold", fontSize=9.5, textColor=C_DARK_GRAY,
              fontName="Helvetica-Bold", leading=14, spaceAfter=2)
bullet = S("Bullet", fontSize=9.5, textColor=C_DARK_GRAY,
           fontName="Helvetica", leading=14, spaceAfter=2,
           leftIndent=14, bulletIndent=4)
mnemonic = S("Mnemonic", fontSize=11, textColor=C_PURPLE,
             fontName="Helvetica-Bold", alignment=TA_CENTER,
             spaceBefore=4, spaceAfter=4)
note_style = S("Note", fontSize=9, textColor=C_ORANGE,
               fontName="Helvetica-BoldOblique", spaceAfter=3)

def section_header(text, bg=C_DARK_BLUE, fg=C_WHITE):
    tbl = Table([[Paragraph(text, S("sh", fontSize=12, textColor=fg,
                                    fontName="Helvetica-Bold", alignment=TA_CENTER))]], 
                colWidths=[17.4*cm])
    tbl.setStyle(TableStyle([
        ("BACKGROUND", (0,0), (-1,-1), bg),
        ("ROUNDEDCORNERS", [6]),
        ("TOPPADDING", (0,0), (-1,-1), 7),
        ("BOTTOMPADDING", (0,0), (-1,-1), 7),
        ("LEFTPADDING", (0,0), (-1,-1), 10),
    ]))
    return tbl

def info_box(text, bg=C_LIGHT_BLUE, border=C_BLUE):
    tbl = Table([[Paragraph(text, S("ib", fontSize=9.5, textColor=C_DARK_GRAY,
                                    fontName="Helvetica", leading=14))]], 
                colWidths=[17.4*cm])
    tbl.setStyle(TableStyle([
        ("BACKGROUND", (0,0), (-1,-1), bg),
        ("BOX", (0,0), (-1,-1), 1.5, border),
        ("TOPPADDING", (0,0), (-1,-1), 7),
        ("BOTTOMPADDING", (0,0), (-1,-1), 7),
        ("LEFTPADDING", (0,0), (-1,-1), 10),
        ("RIGHTPADDING", (0,0), (-1,-1), 10),
    ]))
    return tbl

def mnemonic_box(text, sub=""):
    content = [Paragraph(text, mnemonic)]
    if sub:
        content.append(Paragraph(sub, S("ms", fontSize=9, textColor=C_DARK_GRAY,
                                        fontName="Helvetica-Oblique", alignment=TA_CENTER)))
    tbl = Table([content] if not sub else [[content[0]], [content[1]]], colWidths=[17.4*cm])
    tbl.setStyle(TableStyle([
        ("BACKGROUND", (0,0), (-1,-1), C_LIGHT_PURPLE),
        ("BOX", (0,0), (-1,-1), 2, C_PURPLE),
        ("TOPPADDING", (0,0), (-1,-1), 8),
        ("BOTTOMPADDING", (0,0), (-1,-1), 8),
        ("LEFTPADDING", (0,0), (-1,-1), 10),
    ]))
    return tbl

# ── Build story ──────────────────────────────────────────────────
story = []

# ═══ TITLE BANNER ════════════════════════════════════════════════
title_tbl = Table([
    [Paragraph("ANTIGEN-ANTIBODY REACTION", title_style)],
    [Paragraph("BSc Nursing 3rd Semester  |  Microbiology - Unit IV (Immunity)", subtitle_style)],
    [Paragraph("QUICK REVISION NOTES", tag_style)],
], colWidths=[17.4*cm])
title_tbl.setStyle(TableStyle([
    ("BACKGROUND", (0,0), (-1,-1), C_DARK_BLUE),
    ("TOPPADDING", (0,0), (-1,-1), 6),
    ("BOTTOMPADDING", (0,0), (-1,-1), 6),
    ("LEFTPADDING", (0,0), (-1,-1), 10),
    ("ROUNDEDCORNERS", [8]),
]))
story.append(title_tbl)
story.append(Spacer(1, 0.3*cm))

# ═══ SECTION 1: DEFINITIONS ══════════════════════════════════════
story.append(section_header("1.  ANTIGEN  &  ANTIBODY  —  DEFINITIONS"))
story.append(Spacer(1, 0.2*cm))

def_data = [
    [Paragraph("<b>ANTIGEN</b>", S("th", fontSize=10, textColor=C_WHITE,
               fontName="Helvetica-Bold", alignment=TA_CENTER)),
     Paragraph("<b>ANTIBODY (Immunoglobulin)</b>", S("th", fontSize=10, textColor=C_WHITE,
               fontName="Helvetica-Bold", alignment=TA_CENTER))],
    [Paragraph("• Foreign substance entering the body<br/>"
               "• Triggers immune response<br/>"
               "• Usually protein / polysaccharide<br/>"
               "• Has <b>EPITOPE</b> = exact binding spot for antibody<br/>"
               "• Examples: bacteria, virus, pollen, toxins", body),
     Paragraph("• Y-shaped protein made by <b>Plasma cells</b> (from B lymphocytes)<br/>"
               "• <b>Fab region</b> (top of Y) = binds antigen<br/>"
               "• <b>Fc region</b> (tail) = effector functions (complement, phagocytes)<br/>"
               "• Highly specific: ONE antibody for ONE antigen<br/>"
               "• Lock-and-key mechanism", body)],
]
def_tbl = Table(def_data, colWidths=[8.5*cm, 8.9*cm])
def_tbl.setStyle(TableStyle([
    ("BACKGROUND", (0,0), (-1,0), C_BLUE),
    ("BACKGROUND", (0,1), (0,1), C_LIGHT_RED),
    ("BACKGROUND", (1,1), (1,1), C_LIGHT_GREEN),
    ("BOX", (0,0), (-1,-1), 1, C_BLUE),
    ("INNERGRID", (0,0), (-1,-1), 0.5, C_BLUE),
    ("TOPPADDING", (0,0), (-1,-1), 7),
    ("BOTTOMPADDING", (0,0), (-1,-1), 7),
    ("LEFTPADDING", (0,0), (-1,-1), 8),
    ("VALIGN", (0,0), (-1,-1), "TOP"),
]))
story.append(def_tbl)
story.append(Spacer(1, 0.3*cm))

# ═══ SECTION 2: FLOWCHART ════════════════════════════════════════
story.append(section_header("2.  HOW  AG-AB  REACTION  HAPPENS  (Flowchart)"))
story.append(Spacer(1, 0.2*cm))

flow_steps = [
    ("Antigen enters body",         C_LIGHT_RED,    C_RED),
    ("Recognized as FOREIGN",       C_LIGHT_ORANGE, C_ORANGE),
    ("B Lymphocytes activated",     C_LIGHT_YELLOW, C_YELLOW),
    ("B cells → PLASMA CELLS",      C_LIGHT_GREEN,  C_GREEN),
    ("Plasma cells secrete ANTIBODIES (Immunoglobulins)", C_LIGHT_GREEN, C_GREEN),
    ("Antibody binds Antigen  (Lock & Key)", C_LIGHT_BLUE, C_BLUE),
    ("Ag-Ab COMPLEX FORMED  →  ANTIGEN DESTROYED", C_LIGHT_PURPLE, C_PURPLE),
]

for i, (text, bg, border) in enumerate(flow_steps):
    row_tbl = Table([[Paragraph(f"<b>{text}</b>",
                                S("fs", fontSize=10, textColor=C_DARK_GRAY,
                                  fontName="Helvetica-Bold", alignment=TA_CENTER))]], 
                    colWidths=[17.4*cm])
    row_tbl.setStyle(TableStyle([
        ("BACKGROUND", (0,0), (-1,-1), bg),
        ("BOX", (0,0), (-1,-1), 1.5, border),
        ("TOPPADDING", (0,0), (-1,-1), 5),
        ("BOTTOMPADDING", (0,0), (-1,-1), 5),
    ]))
    story.append(row_tbl)
    if i < len(flow_steps) - 1:
        arrow = Table([[Paragraph("<b>↓</b>", S("ar", fontSize=14, textColor=C_BLUE,
                                                 fontName="Helvetica-Bold", alignment=TA_CENTER))]], 
                      colWidths=[17.4*cm])
        arrow.setStyle(TableStyle([("TOPPADDING",(0,0),(-1,-1),1),("BOTTOMPADDING",(0,0),(-1,-1),1)]))
        story.append(arrow)

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

# ═══ SECTION 3: RESULTS (APNLO) ══════════════════════════════════
story.append(section_header("3.  RESULTS / ACTIONS  OF  AG-AB  REACTION"))
story.append(Spacer(1, 0.15*cm))
story.append(mnemonic_box('🧠  Mnemonic:  "A P N L O"',
                          "Agglutination  |  Precipitation  |  Neutralization  |  Lysis  |  Opsonization"))
story.append(Spacer(1, 0.2*cm))

results_data = [
    [Paragraph("<b>Action</b>", S("rh", fontSize=10, textColor=C_WHITE, fontName="Helvetica-Bold", alignment=TA_CENTER)),
     Paragraph("<b>What Happens</b>", S("rh", fontSize=10, textColor=C_WHITE, fontName="Helvetica-Bold")),
     Paragraph("<b>Example</b>", S("rh", fontSize=10, textColor=C_WHITE, fontName="Helvetica-Bold"))],
    [Paragraph("<b>1. Agglutination</b>", body_bold),
     Paragraph("Antibody CLUMPS multiple antigens (bacteria/RBCs) into a lump → phagocytes eat them", body),
     Paragraph("Blood group testing, Widal test (Typhoid)", body)],
    [Paragraph("<b>2. Precipitation</b>", body_bold),
     Paragraph("Antibody binds SOLUBLE antigen → complex becomes insoluble & precipitates out", body),
     Paragraph("Tetanus toxin neutralized in fluid", body)],
    [Paragraph("<b>3. Neutralization</b>", body_bold),
     Paragraph("Antibody COVERS the toxic site of antigen → antigen becomes harmless", body),
     Paragraph("Antitoxins for tetanus, polio virus antibodies", body)],
    [Paragraph("<b>4. Lysis</b>", body_bold),
     Paragraph("Antibody + Complement system BURST the bacterial cell membrane", body),
     Paragraph("Bacterial cell wall destruction", body)],
    [Paragraph("<b>5. Opsonization</b>", body_bold),
     Paragraph("Antibody COATS antigen → makes it easy for phagocytes to eat ('tasty')", body),
     Paragraph("Enhanced phagocytosis by neutrophils & macrophages", body)],
]

row_colors = [C_BLUE, C_LIGHT_RED, C_LIGHT_ORANGE, C_LIGHT_YELLOW, C_LIGHT_GREEN, C_LIGHT_BLUE]
res_tbl = Table(results_data, colWidths=[4.0*cm, 8.2*cm, 5.2*cm])
res_tbl.setStyle(TableStyle([
    ("BACKGROUND", (0,0), (-1,0), C_BLUE),
    ("BACKGROUND", (0,1), (-1,1), C_LIGHT_RED),
    ("BACKGROUND", (0,2), (-1,2), C_LIGHT_ORANGE),
    ("BACKGROUND", (0,3), (-1,3), C_LIGHT_YELLOW),
    ("BACKGROUND", (0,4), (-1,4), C_LIGHT_GREEN),
    ("BACKGROUND", (0,5), (-1,5), C_LIGHT_BLUE),
    ("BOX", (0,0), (-1,-1), 1, C_BLUE),
    ("INNERGRID", (0,0), (-1,-1), 0.5, colors.HexColor("#CBD5E1")),
    ("TOPPADDING", (0,0), (-1,-1), 6),
    ("BOTTOMPADDING", (0,0), (-1,-1), 6),
    ("LEFTPADDING", (0,0), (-1,-1), 7),
    ("VALIGN", (0,0), (-1,-1), "TOP"),
]))
story.append(res_tbl)
story.append(Spacer(1, 0.3*cm))

# ═══ SECTION 4: IMMUNOGLOBULINS ══════════════════════════════════
story.append(section_header("4.  TYPES  OF  IMMUNOGLOBULINS  (Antibodies)"))
story.append(Spacer(1, 0.15*cm))
story.append(mnemonic_box('🧠  Mnemonic:  "G A D M E"',
                          '"Go And Defend Me Everywhere"  —  IgG, IgA, IgD, IgM, IgE'))
story.append(Spacer(1, 0.2*cm))

ig_data = [
    [Paragraph("<b>Ig</b>", S("ih", fontSize=10, textColor=C_WHITE, fontName="Helvetica-Bold", alignment=TA_CENTER)),
     Paragraph("<b>% in Blood</b>", S("ih", fontSize=10, textColor=C_WHITE, fontName="Helvetica-Bold", alignment=TA_CENTER)),
     Paragraph("<b>Where Found</b>", S("ih", fontSize=10, textColor=C_WHITE, fontName="Helvetica-Bold")),
     Paragraph("<b>Key Functions</b>", S("ih", fontSize=10, textColor=C_WHITE, fontName="Helvetica-Bold"))],
    [Paragraph("<b>IgG</b>", S("igg", fontSize=11, textColor=C_BLUE, fontName="Helvetica-Bold", alignment=TA_CENTER)),
     Paragraph("75-80%", body),
     Paragraph("Blood, tissues", body),
     Paragraph("• Most abundant<br/>• Memory antibody (2nd response)<br/>• <b>CROSSES PLACENTA</b> → protects newborn<br/>• Main defense against bacteria & viruses", body)],
    [Paragraph("<b>IgA</b>", S("iga", fontSize=11, textColor=C_GREEN, fontName="Helvetica-Bold", alignment=TA_CENTER)),
     Paragraph("10-15%", body),
     Paragraph("Secretions: saliva, tears, breast milk, mucus", body),
     Paragraph("• Protects mucous membranes<br/>• Found in <b>colostrum</b> (1st breast milk)<br/>• Guards entry points of body", body)],
    [Paragraph("<b>IgD</b>", S("igd", fontSize=11, textColor=C_YELLOW, fontName="Helvetica-Bold", alignment=TA_CENTER)),
     Paragraph("<1%", body),
     Paragraph("Surface of B cells", body),
     Paragraph("• Acts as <b>receptor on B lymphocytes</b><br/>• Activates B cells when antigen binds", body)],
    [Paragraph("<b>IgM</b>", S("igm", fontSize=11, textColor=C_RED, fontName="Helvetica-Bold", alignment=TA_CENTER)),
     Paragraph("5-10%", body),
     Paragraph("Blood (intravascular)", body),
     Paragraph("• <b>FIRST antibody to appear</b> in any new infection<br/>• Largest antibody (pentamer)<br/>• Blood group antibodies (ABO)<br/>• Primary immune response", body)],
    [Paragraph("<b>IgE</b>", S("ige", fontSize=11, textColor=C_PURPLE, fontName="Helvetica-Bold", alignment=TA_CENTER)),
     Paragraph("Trace", body),
     Paragraph("Mast cells, basophils", body),
     Paragraph("• Responsible for <b>ALLERGIC reactions</b><br/>• Fights parasites/worms<br/>• Raised in asthma, hay fever, anaphylaxis", body)],
]

ig_tbl = Table(ig_data, colWidths=[1.8*cm, 2.2*cm, 4.5*cm, 8.9*cm])
ig_tbl.setStyle(TableStyle([
    ("BACKGROUND", (0,0), (-1,0), C_DARK_BLUE),
    ("BACKGROUND", (0,1), (-1,1), C_LIGHT_BLUE),
    ("BACKGROUND", (0,2), (-1,2), C_LIGHT_GREEN),
    ("BACKGROUND", (0,3), (-1,3), C_LIGHT_YELLOW),
    ("BACKGROUND", (0,4), (-1,4), C_LIGHT_RED),
    ("BACKGROUND", (0,5), (-1,5), C_LIGHT_PURPLE),
    ("BOX", (0,0), (-1,-1), 1.2, C_DARK_BLUE),
    ("INNERGRID", (0,0), (-1,-1), 0.5, colors.HexColor("#CBD5E1")),
    ("TOPPADDING", (0,0), (-1,-1), 6),
    ("BOTTOMPADDING", (0,0), (-1,-1), 6),
    ("LEFTPADDING", (0,0), (-1,-1), 7),
    ("VALIGN", (0,0), (-1,-1), "TOP"),
    ("ALIGN", (0,0), (1,-1), "CENTER"),
]))
story.append(ig_tbl)
story.append(Spacer(1, 0.3*cm))

# ═══ SECTION 5: COMPLEMENT SYSTEM ════════════════════════════════
story.append(section_header("5.  COMPLEMENT  SYSTEM", bg=C_GREEN))
story.append(Spacer(1, 0.15*cm))
story.append(info_box(
    "<b>What is it?</b>  ~20 proteins in blood that AMPLIFY the antibody response.<br/>"
    "<b>Activated by:</b>  Ag-Ab complex → Classical Pathway<br/><br/>"
    "<b>Key Results:</b><br/>"
    "• <b>C3b</b> → Opsonization (coats bacteria → phagocytes eat them easily)<br/>"
    "• <b>MAC (C5b6789)</b> = Membrane Attack Complex → punches holes in bacterial membrane → bacteria BURST and die<br/>"
    "• Agglutination, Neutralization of viruses, Chemotaxis (attract more WBCs)",
    bg=C_LIGHT_GREEN, border=C_GREEN
))
story.append(Spacer(1, 0.3*cm))

# ═══ SECTION 6: KEY DIFFERENCES TABLE ════════════════════════════
story.append(section_header("6.  KEY  FACTS  TO  REMEMBER  (Exam Favorites)", bg=C_ORANGE))
story.append(Spacer(1, 0.15*cm))

key_data = [
    [Paragraph("<b>Question</b>", S("kh", fontSize=10, textColor=C_WHITE, fontName="Helvetica-Bold")),
     Paragraph("<b>Answer</b>", S("kh", fontSize=10, textColor=C_WHITE, fontName="Helvetica-Bold"))],
    [Paragraph("Which Ig crosses the placenta?", body), Paragraph("<b>IgG</b> → protects the newborn baby", body_bold)],
    [Paragraph("Which Ig appears FIRST in infection?", body), Paragraph("<b>IgM</b> → primary immune response", body_bold)],
    [Paragraph("Which Ig is found in breast milk/saliva?", body), Paragraph("<b>IgA</b> → secretory antibody", body_bold)],
    [Paragraph("Which Ig causes allergic reactions?", body), Paragraph("<b>IgE</b> → allergy + parasites", body_bold)],
    [Paragraph("Which Ig is the most abundant?", body), Paragraph("<b>IgG</b> → 75-80% of all antibodies", body_bold)],
    [Paragraph("What is EPITOPE?", body), Paragraph("The specific region on an antigen where antibody binds", body)],
    [Paragraph("What is OPSONIN?", body), Paragraph("Substance (antibody/complement) that coats antigen to make phagocytosis easier", body)],
    [Paragraph("Affinity vs Avidity?", body), Paragraph("<b>Affinity</b> = strength of ONE binding site | <b>Avidity</b> = overall strength (multiple sites)", body)],
]

key_tbl = Table(key_data, colWidths=[8.0*cm, 9.4*cm])
key_tbl.setStyle(TableStyle([
    ("BACKGROUND", (0,0), (-1,0), C_ORANGE),
    ("ROWBACKGROUNDS", (0,1), (-1,-1), [C_WHITE, C_LIGHT_ORANGE]),
    ("BOX", (0,0), (-1,-1), 1, C_ORANGE),
    ("INNERGRID", (0,0), (-1,-1), 0.5, colors.HexColor("#FED7AA")),
    ("TOPPADDING", (0,0), (-1,-1), 6),
    ("BOTTOMPADDING", (0,0), (-1,-1), 6),
    ("LEFTPADDING", (0,0), (-1,-1), 8),
    ("VALIGN", (0,0), (-1,-1), "TOP"),
]))
story.append(key_tbl)
story.append(Spacer(1, 0.3*cm))

# ═══ SECTION 7: EXAM QUESTIONS ═══════════════════════════════════
story.append(section_header("7.  LIKELY  EXAM  QUESTIONS", bg=C_RED))
story.append(Spacer(1, 0.15*cm))

questions = [
    ("5 marks", "Describe antigen-antibody reaction and explain its results."),
    ("5 marks", "Classify immunoglobulins with their properties and functions."),
    ("3 marks", "Write a short note on the complement system."),
    ("2 marks", "Which immunoglobulin crosses the placenta? Which appears first in infection?"),
    ("2 marks", "Define opsonization. What is its significance?"),
    ("2 marks", "Differentiate between affinity and avidity."),
]

q_data = [[Paragraph("<b>Marks</b>", S("qh", fontSize=10, textColor=C_WHITE, fontName="Helvetica-Bold", alignment=TA_CENTER)),
           Paragraph("<b>Question</b>", S("qh", fontSize=10, textColor=C_WHITE, fontName="Helvetica-Bold"))]]
for marks, q in questions:
    q_data.append([
        Paragraph(f"<b>{marks}</b>", S("qm", fontSize=9, textColor=C_RED, fontName="Helvetica-Bold", alignment=TA_CENTER)),
        Paragraph(q, body)
    ])

q_tbl = Table(q_data, colWidths=[2.5*cm, 14.9*cm])
q_tbl.setStyle(TableStyle([
    ("BACKGROUND", (0,0), (-1,0), C_RED),
    ("ROWBACKGROUNDS", (0,1), (-1,-1), [C_WHITE, C_LIGHT_RED]),
    ("BOX", (0,0), (-1,-1), 1, C_RED),
    ("INNERGRID", (0,0), (-1,-1), 0.5, colors.HexColor("#FECACA")),
    ("TOPPADDING", (0,0), (-1,-1), 6),
    ("BOTTOMPADDING", (0,0), (-1,-1), 6),
    ("LEFTPADDING", (0,0), (-1,-1), 8),
    ("VALIGN", (0,0), (-1,-1), "MIDDLE"),
    ("ALIGN", (0,0), (0,-1), "CENTER"),
]))
story.append(q_tbl)
story.append(Spacer(1, 0.3*cm))

# ═══ FOOTER ══════════════════════════════════════════════════════
footer_tbl = Table([[Paragraph(
    "BSc Nursing 3rd Semester  •  Microbiology Unit IV  •  Quick Revision Notes  •  All the best for your exam! 💪",
    S("ft", fontSize=8, textColor=C_WHITE, fontName="Helvetica-Oblique", alignment=TA_CENTER)
)]], colWidths=[17.4*cm])
footer_tbl.setStyle(TableStyle([
    ("BACKGROUND", (0,0), (-1,-1), C_DARK_BLUE),
    ("TOPPADDING", (0,0), (-1,-1), 6),
    ("BOTTOMPADDING", (0,0), (-1,-1), 6),
    ("ROUNDEDCORNERS", [6]),
]))
story.append(footer_tbl)

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

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

OUTPUT = "/home/daytona/workspace/nursing-notes/Stoma_Care.pdf"

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

# ── Colours ──────────────────────────────────────────────────────
DARK_BLUE   = colors.HexColor("#1A3C6E")
BLUE        = colors.HexColor("#2563EB")
LT_BLUE     = colors.HexColor("#DBEAFE")
RED         = colors.HexColor("#DC2626")
LT_RED      = colors.HexColor("#FEE2E2")
GREEN       = colors.HexColor("#16A34A")
LT_GREEN    = colors.HexColor("#DCFCE7")
ORANGE      = colors.HexColor("#EA580C")
LT_ORANGE   = colors.HexColor("#FFF7ED")
PURPLE      = colors.HexColor("#7C3AED")
LT_PURPLE   = colors.HexColor("#EDE9FE")
TEAL        = colors.HexColor("#0F766E")
LT_TEAL     = colors.HexColor("#CCFBF1")
YELLOW      = colors.HexColor("#B45309")
LT_YELLOW   = colors.HexColor("#FEF3C7")
GRAY        = colors.HexColor("#F1F5F9")
DARK_GRAY   = colors.HexColor("#1F2937")
MID_GRAY    = colors.HexColor("#6B7280")
WHITE       = colors.white

W = 17.8*cm   # usable page width

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

# ── Reusable builders ────────────────────────────────────────────
def hdr(text, bg=DARK_BLUE, fg=WHITE, fs=12):
    t = Table([[Paragraph(text, S("h", fontSize=fs, textColor=fg,
               fontName="Helvetica-Bold", alignment=TA_CENTER))]], colWidths=[W])
    t.setStyle(TableStyle([
        ("BACKGROUND",(0,0),(-1,-1),bg),
        ("TOPPADDING",(0,0),(-1,-1),6), ("BOTTOMPADDING",(0,0),(-1,-1),6),
        ("LEFTPADDING",(0,0),(-1,-1),8),
    ]))
    return t

def box(content_para, bg=LT_BLUE, border=BLUE):
    t = Table([[content_para]], colWidths=[W])
    t.setStyle(TableStyle([
        ("BACKGROUND",(0,0),(-1,-1),bg),
        ("BOX",(0,0),(-1,-1),1.5,border),
        ("TOPPADDING",(0,0),(-1,-1),6), ("BOTTOMPADDING",(0,0),(-1,-1),6),
        ("LEFTPADDING",(0,0),(-1,-1),9), ("RIGHTPADDING",(0,0),(-1,-1),9),
    ]))
    return t

def mnem_box(text, sub=""):
    rows = [[Paragraph(text, S("m", fontSize=11, textColor=PURPLE,
                               fontName="Helvetica-Bold", alignment=TA_CENTER))]]
    if sub:
        rows.append([Paragraph(sub, S("ms", fontSize=9, textColor=DARK_GRAY,
                                      fontName="Helvetica-Oblique", alignment=TA_CENTER))])
    t = Table(rows, colWidths=[W])
    t.setStyle(TableStyle([
        ("BACKGROUND",(0,0),(-1,-1),LT_PURPLE),
        ("BOX",(0,0),(-1,-1),2,PURPLE),
        ("TOPPADDING",(0,0),(-1,-1),7), ("BOTTOMPADDING",(0,0),(-1,-1),7),
        ("LEFTPADDING",(0,0),(-1,-1),8),
    ]))
    return t

def p(text, fs=9.5, bold=False, color=DARK_GRAY, align=TA_LEFT, leading=14):
    fn = "Helvetica-Bold" if bold else "Helvetica"
    return Paragraph(text, S("p", fontSize=fs, textColor=color,
                              fontName=fn, leading=leading, alignment=align))

def sp(h=0.2): return Spacer(1, h*cm)

# ════════════════════════════════════════════════════════════════
story = []

# ── TITLE BANNER ─────────────────────────────────────────────────
title_t = Table([
    [p("STOMA CARE", fs=20, bold=True, color=WHITE, align=TA_CENTER)],
    [p("BSc Nursing 3rd Semester  |  MSN - Unit V (GI System)", fs=10,
       color=colors.HexColor("#BFDBFE"), align=TA_CENTER)],
    [p("QUICK REVISION NOTES", fs=9, bold=True,
       color=colors.HexColor("#FDE68A"), align=TA_CENTER)],
], colWidths=[W])
title_t.setStyle(TableStyle([
    ("BACKGROUND",(0,0),(-1,-1),DARK_BLUE),
    ("TOPPADDING",(0,0),(-1,-1),5), ("BOTTOMPADDING",(0,0),(-1,-1),5),
]))
story += [title_t, sp(0.25)]

# ── SECTION 1: DEFINITION + INDICATIONS ──────────────────────────
story += [hdr("1.  WHAT IS A STOMA?  —  DEFINITION & INDICATIONS"), sp(0.15)]

story.append(box(
    p("<b>STOMA</b> = Greek for 'mouth/opening' → A <b>surgically created opening on the abdominal wall</b> "
      "that allows waste (stool or urine) to exit the body when the normal route is blocked or removed.<br/>"
      "<b>Simple:</b> An artificial opening made on the belly for waste to come out."),
    bg=LT_BLUE, border=BLUE))
story.append(sp(0.15))

ind_data = [
    [p("<b>Indication</b>", bold=True, color=WHITE), p("<b>Reason</b>", bold=True, color=WHITE)],
    [p("Colorectal / rectal cancer"), p("Tumour blocks the bowel")],
    [p("Crohn's disease / Ulcerative Colitis"), p("Bowel too inflamed / damaged")],
    [p("Bowel obstruction / perforation"), p("Emergency surgery")],
    [p("Abdominal trauma"), p("Road accidents, stab wounds")],
    [p("Hirschsprung's disease (newborns)"), p("Part of bowel doesn't work")],
    [p("Bladder cancer / injury"), p("Urostomy needed")],
]
ind_t = Table(ind_data, colWidths=[9*cm, 8.8*cm])
ind_t.setStyle(TableStyle([
    ("BACKGROUND",(0,0),(-1,0),BLUE),
    ("ROWBACKGROUNDS",(0,1),(-1,-1),[WHITE, LT_BLUE]),
    ("BOX",(0,0),(-1,-1),1,BLUE), ("INNERGRID",(0,0),(-1,-1),0.5,colors.HexColor("#BFDBFE")),
    ("TOPPADDING",(0,0),(-1,-1),5), ("BOTTOMPADDING",(0,0),(-1,-1),5),
    ("LEFTPADDING",(0,0),(-1,-1),7), ("VALIGN",(0,0),(-1,-1),"MIDDLE"),
]))
story += [ind_t, sp(0.25)]

# ── SECTION 2: TYPES OF STOMA ─────────────────────────────────────
story += [hdr("2.  TYPES OF STOMA", bg=TEAL), sp(0.15)]

types_data = [
    [p("<b>Type</b>", bold=True, color=WHITE),
     p("<b>Organ</b>", bold=True, color=WHITE),
     p("<b>Side</b>", bold=True, color=WHITE),
     p("<b>Output</b>", bold=True, color=WHITE),
     p("<b>Bag Type</b>", bold=True, color=WHITE)],
    [p("<b>Colostomy</b>", bold=True, color=TEAL), p("Large intestine"),
     p("LEFT side"), p("Semi-formed stool"), p("Closed / Non-drainable")],
    [p("<b>Ileostomy</b>", bold=True, color=BLUE), p("Small intestine (ileum)"),
     p("RIGHT side"), p("Liquid / watery"), p("Drainable bag")],
    [p("<b>Urostomy</b>", bold=True, color=PURPLE), p("Urinary tract"),
     p("Either side"), p("Urine"), p("Drainable bag")],
]
types_t = Table(types_data, colWidths=[3.0*cm, 3.5*cm, 2.8*cm, 4.5*cm, 4.0*cm])
types_t.setStyle(TableStyle([
    ("BACKGROUND",(0,0),(-1,0),TEAL),
    ("BACKGROUND",(0,1),(-1,1),LT_TEAL),
    ("BACKGROUND",(0,2),(-1,2),LT_BLUE),
    ("BACKGROUND",(0,3),(-1,3),LT_PURPLE),
    ("BOX",(0,0),(-1,-1),1,TEAL), ("INNERGRID",(0,0),(-1,-1),0.5,colors.HexColor("#99F6E4")),
    ("TOPPADDING",(0,0),(-1,-1),6), ("BOTTOMPADDING",(0,0),(-1,-1),6),
    ("LEFTPADDING",(0,0),(-1,-1),7), ("VALIGN",(0,0),(-1,-1),"MIDDLE"),
]))
story += [types_t, sp(0.12)]

story.append(mnem_box("💡  Memory Trick:",
                       "Ileostomy = 'I' = Inside/Right side = lIquid   |   Colostomy = 'C' = Colon/Left side = formed stool"))
story.append(sp(0.25))

# ── SECTION 3: COLOSTOMY vs ILEOSTOMY ─────────────────────────────
story += [hdr("3.  COLOSTOMY  vs  ILEOSTOMY  — KEY DIFFERENCES", bg=ORANGE), sp(0.15)]

diff_data = [
    [p("<b>Feature</b>", bold=True, color=WHITE),
     p("<b>Colostomy</b>", bold=True, color=WHITE),
     p("<b>Ileostomy</b>", bold=True, color=WHITE)],
    [p("Position"), p("LEFT side abdomen"), p("RIGHT side abdomen")],
    [p("Stool consistency"), p("Soft → formed"), p("Liquid / watery")],
    [p("Daily output"), p("300–500 ml"), p("500–1500 ml")],
    [p("Skin irritation"), p("Less (alkaline)"), p("<b>More</b> (digestive enzymes)")],
    [p("Bag type"), p("Closed/non-drainable"), p("<b>Drainable</b> bag")],
    [p("Bag change"), p("2–3 times/day"), p("Every 48 hours")],
    [p("Appearance"), p("Flat, round"), p("<b>Spouted</b> (raised/protruding)")],
    [p("Irrigation"), p("Possible (sigmoid)"), p("Not possible")],
    [p("Dehydration risk"), p("Low"), p("<b>HIGH</b> — drink extra fluids!")],
]
diff_t = Table(diff_data, colWidths=[4.0*cm, 6.8*cm, 7.0*cm])
diff_t.setStyle(TableStyle([
    ("BACKGROUND",(0,0),(-1,0),ORANGE),
    ("ROWBACKGROUNDS",(0,1),(-1,-1),[WHITE, LT_ORANGE]),
    ("BOX",(0,0),(-1,-1),1,ORANGE), ("INNERGRID",(0,0),(-1,-1),0.5,colors.HexColor("#FED7AA")),
    ("TOPPADDING",(0,0),(-1,-1),5), ("BOTTOMPADDING",(0,0),(-1,-1),5),
    ("LEFTPADDING",(0,0),(-1,-1),7), ("VALIGN",(0,0),(-1,-1),"MIDDLE"),
]))
story += [diff_t, sp(0.25)]

# ── SECTION 4: STOMA CARE PROCEDURE ──────────────────────────────
story += [hdr("4.  STOMA CARE PROCEDURE  (Step-by-Step)", bg=GREEN), sp(0.15)]

steps = [
    ("STEP 1", "REMOVE OLD BAG", LT_RED, RED,
     "Use adhesive remover wipe → peel gently top to bottom → empty contents first if drainable → dispose in clinical waste"),
    ("STEP 2", "CLEAN STOMA", LT_GREEN, GREEN,
     "Warm water + soft gauze → clean around stoma in circular motion outward → mild soap if needed → rinse well → PAT DRY (not rub)"),
    ("STEP 3", "ASSESS STOMA", LT_YELLOW, YELLOW,
     "Observe: Colour (pink-red = healthy) | Size | Shape | Protrusion | Peristomal skin condition"),
    ("STEP 4", "MEASURE + CUT FLANGE", LT_BLUE, BLUE,
     "Use stoma measuring guide → cut hole 2–3 mm LARGER than stoma size (too small = pressure, too large = skin contact with stool)"),
    ("STEP 5", "APPLY BARRIER PASTE", LT_PURPLE, PURPLE,
     "Apply karaya/stomahesive paste around the opening → protects peristomal skin from leakage"),
    ("STEP 6", "APPLY NEW APPLIANCE", LT_TEAL, TEAL,
     "Centre opening over stoma → press firmly from centre outward → hold with warm hand 30 seconds → check for leaks → for 2-piece: attach wafer first, then click bag"),
    ("STEP 7", "DOCUMENT", GRAY, MID_GRAY,
     "Record: stoma colour, size, output type/amount, skin condition, bag changed, patient tolerance"),
]

for num, title, bg, border, detail in steps:
    row = Table([[
        p(f"<b>{num}</b>", fs=9, bold=True, color=border, align=TA_CENTER),
        p(f"<b>{title}</b>", fs=9.5, bold=True, color=border),
        p(detail, fs=9),
    ]], colWidths=[1.5*cm, 4.2*cm, 12.1*cm])
    row.setStyle(TableStyle([
        ("BACKGROUND",(0,0),(-1,-1),bg),
        ("BOX",(0,0),(-1,-1),1,border),
        ("TOPPADDING",(0,0),(-1,-1),5), ("BOTTOMPADDING",(0,0),(-1,-1),5),
        ("LEFTPADDING",(0,0),(-1,-1),6), ("VALIGN",(0,0),(-1,-1),"MIDDLE"),
    ]))
    story.append(row)
    story.append(sp(0.08))

story.append(sp(0.2))

# ── SECTION 5: HEALTHY vs DANGER ─────────────────────────────────
story += [hdr("5.  HEALTHY STOMA  vs  DANGER SIGNS", bg=RED), sp(0.15)]

hd_data = [
    [p("<b>✅ Healthy Stoma</b>", bold=True, color=WHITE),
     p("<b>🚨 Report Immediately</b>", bold=True, color=WHITE)],
    [p("Pink-red colour"), p("<b>Dark red / purple / BLACK</b> = Ischaemia/Necrosis")],
    [p("Moist and glistening surface"), p("Dry, dull appearance")],
    [p("Slightly raised (ileostomy)"), p("Retracted below skin level")],
    [p("Minimal touch bleeding"), p("Profuse or continuous bleeding")],
    [p("Normal stool/urine output"), p("<b>No output for 4–6 hrs</b> = Obstruction")],
    [p("Intact peristomal skin"), p("Red, raw, broken skin around stoma")],
]
hd_t = Table(hd_data, colWidths=[8.8*cm, 9.0*cm])
hd_t.setStyle(TableStyle([
    ("BACKGROUND",(0,0),(-1,0),RED),
    ("BACKGROUND",(0,1),(0,-1),LT_GREEN),
    ("BACKGROUND",(1,1),(1,-1),LT_RED),
    ("BOX",(0,0),(-1,-1),1,RED), ("INNERGRID",(0,0),(-1,-1),0.5,colors.HexColor("#FECACA")),
    ("TOPPADDING",(0,0),(-1,-1),5), ("BOTTOMPADDING",(0,0),(-1,-1),5),
    ("LEFTPADDING",(0,0),(-1,-1),7), ("VALIGN",(0,0),(-1,-1),"MIDDLE"),
]))
story += [hd_t, sp(0.25)]

# ── SECTION 6: COMPLICATIONS ──────────────────────────────────────
story += [hdr("6.  COMPLICATIONS  OF  STOMA", bg=PURPLE), sp(0.15)]
story.append(mnem_box('🧠  Mnemonic:  "P R S S – H F"',
                       "Prolapse | Retraction | Stenosis | Skin excoriation | Hernia | Fistula (+Ischaemia, Obstruction)"))
story.append(sp(0.15))

comp_data = [
    [p("<b>Complication</b>", bold=True, color=WHITE),
     p("<b>Description</b>", bold=True, color=WHITE),
     p("<b>Nursing Action</b>", bold=True, color=WHITE)],
    [p("<b>Prolapse</b>", bold=True), p("Stoma protrudes excessively outward"), p("Reduce manually + surgical review")],
    [p("<b>Retraction</b>", bold=True), p("Stoma sinks below skin level"), p("Difficult to fit bag → surgical review")],
    [p("<b>Stenosis</b>", bold=True), p("Narrowing of stoma opening"), p("Dilation or surgical revision")],
    [p("<b>Skin Excoriation</b>", bold=True), p("Peristomal skin red, raw, sore (esp. ileostomy)"), p("Correct bag sizing, barrier cream/paste")],
    [p("<b>Parastomal Hernia</b>", bold=True), p("Bulge around stoma under skin"), p("Support belt + surgical repair")],
    [p("<b>Ischaemia/Necrosis</b>", bold=True), p("Stoma turns DARK/BLACK — blood supply cut"), p("🚨 Emergency — inform doctor immediately!")],
    [p("<b>Obstruction</b>", bold=True), p("No output for 4–6 hours"), p("Hydration, position change, surgical review")],
    [p("<b>Fistula</b>", bold=True), p("Abnormal tract near stoma"), p("Surgical management")],
]
comp_t = Table(comp_data, colWidths=[3.8*cm, 7.2*cm, 6.8*cm])
comp_t.setStyle(TableStyle([
    ("BACKGROUND",(0,0),(-1,0),PURPLE),
    ("ROWBACKGROUNDS",(0,1),(-1,-1),[WHITE, LT_PURPLE]),
    ("BOX",(0,0),(-1,-1),1,PURPLE), ("INNERGRID",(0,0),(-1,-1),0.5,colors.HexColor("#DDD6FE")),
    ("TOPPADDING",(0,0),(-1,-1),5), ("BOTTOMPADDING",(0,0),(-1,-1),5),
    ("LEFTPADDING",(0,0),(-1,-1),7), ("VALIGN",(0,0),(-1,-1),"TOP"),
]))
story += [comp_t, sp(0.25)]

# ── SECTION 7: PATIENT EDUCATION ──────────────────────────────────
story += [hdr("7.  PATIENT  EDUCATION  —  DISCHARGE TEACHING", bg=YELLOW), sp(0.15)]

edu_items = [
    ("Bag Change", "Change when 1/3 to 1/2 full. Ileostomy every 48 hrs. Colostomy 2–3×/day."),
    ("Diet", "Avoid gas-forming foods (beans, cabbage). Ileostomy patients: DRINK MORE WATER (dehydration risk)."),
    ("Activity", "Normal life — work, swimming, exercise all allowed. Avoid heavy lifting for 6 weeks post-op."),
    ("Skin Care", "Keep peristomal skin dry and intact. Use barrier cream daily."),
    ("Report to nurse/doctor", "Dark stoma, no output 4–6 hrs, bleeding, leakage, severe skin breakdown."),
    ("Medications", "Some tablets may pass whole into bag — ask doctor for liquid/soluble forms."),
    ("Support", "Contact stoma care nurse / colostomy association for help and latest appliances."),
]
edu_data = [[p("<b>Topic</b>", bold=True, color=WHITE), p("<b>Key Points</b>", bold=True, color=WHITE)]]
for topic, detail in edu_items:
    edu_data.append([p(f"<b>{topic}</b>", bold=True), p(detail)])
edu_t = Table(edu_data, colWidths=[3.5*cm, 14.3*cm])
edu_t.setStyle(TableStyle([
    ("BACKGROUND",(0,0),(-1,0),YELLOW),
    ("ROWBACKGROUNDS",(0,1),(-1,-1),[WHITE, LT_YELLOW]),
    ("BOX",(0,0),(-1,-1),1,YELLOW), ("INNERGRID",(0,0),(-1,-1),0.5,colors.HexColor("#FDE68A")),
    ("TOPPADDING",(0,0),(-1,-1),5), ("BOTTOMPADDING",(0,0),(-1,-1),5),
    ("LEFTPADDING",(0,0),(-1,-1),7), ("VALIGN",(0,0),(-1,-1),"TOP"),
]))
story += [edu_t, sp(0.25)]

# ── SECTION 8: EXAM QUESTIONS ─────────────────────────────────────
story += [hdr("8.  LIKELY EXAM QUESTIONS", bg=RED), sp(0.15)]

qs = [
    ("5 marks", "Define stoma. Classify types and give differences between colostomy and ileostomy."),
    ("10 marks", "Describe the nursing care / procedure for a patient with colostomy."),
    ("5 marks", "What are the complications of stoma? How are they managed?"),
    ("10 marks", "Write a nursing care plan for a patient with colostomy."),
    ("5 marks", "What health education will you give to a patient discharged with a colostomy?"),
    ("3 marks", "What is colostomy irrigation? Explain the procedure."),
    ("2 marks", "Difference between prolapse and retraction of stoma."),
]
q_data = [[p("<b>Marks</b>", bold=True, color=WHITE, align=TA_CENTER),
           p("<b>Question</b>", bold=True, color=WHITE)]]
for m, q in qs:
    q_data.append([p(f"<b>{m}</b>", bold=True, color=RED, align=TA_CENTER), p(q)])
q_t = Table(q_data, colWidths=[2.5*cm, 15.3*cm])
q_t.setStyle(TableStyle([
    ("BACKGROUND",(0,0),(-1,0),RED),
    ("ROWBACKGROUNDS",(0,1),(-1,-1),[WHITE, LT_RED]),
    ("BOX",(0,0),(-1,-1),1,RED), ("INNERGRID",(0,0),(-1,-1),0.5,colors.HexColor("#FECACA")),
    ("TOPPADDING",(0,0),(-1,-1),5), ("BOTTOMPADDING",(0,0),(-1,-1),5),
    ("LEFTPADDING",(0,0),(-1,-1),7), ("VALIGN",(0,0),(-1,-1),"MIDDLE"),
    ("ALIGN",(0,0),(0,-1),"CENTER"),
]))
story += [q_t, sp(0.2)]

# ── FOOTER ────────────────────────────────────────────────────────
footer = Table([[p("BSc Nursing 3rd Semester  •  MSN Unit V  •  Stoma Care  •  Quick Revision Notes  •  All the best! 💪",
                   fs=8, color=WHITE, align=TA_CENTER)]], colWidths=[W])
footer.setStyle(TableStyle([
    ("BACKGROUND",(0,0),(-1,-1),DARK_BLUE),
    ("TOPPADDING",(0,0),(-1,-1),6), ("BOTTOMPADDING",(0,0),(-1,-1),6),
]))
story.append(footer)

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

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

OUTPUT = "/tmp/workspace/nursing-notes/Stoma_Care.pdf"

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

DARK_BLUE   = colors.HexColor("#1A3C6E")
BLUE        = colors.HexColor("#2563EB")
LT_BLUE     = colors.HexColor("#DBEAFE")
RED         = colors.HexColor("#DC2626")
LT_RED      = colors.HexColor("#FEE2E2")
GREEN       = colors.HexColor("#16A34A")
LT_GREEN    = colors.HexColor("#DCFCE7")
ORANGE      = colors.HexColor("#EA580C")
LT_ORANGE   = colors.HexColor("#FFF7ED")
PURPLE      = colors.HexColor("#7C3AED")
LT_PURPLE   = colors.HexColor("#EDE9FE")
TEAL        = colors.HexColor("#0F766E")
LT_TEAL     = colors.HexColor("#CCFBF1")
YELLOW      = colors.HexColor("#B45309")
LT_YELLOW   = colors.HexColor("#FEF3C7")
DARK_GRAY   = colors.HexColor("#1F2937")
MID_GRAY    = colors.HexColor("#6B7280")
WHITE       = colors.white
W = 17.8*cm

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

def hdr(text, bg=DARK_BLUE, fg=WHITE, fs=12):
    t = Table([[Paragraph(text, S("h", fontSize=fs, textColor=fg,
               fontName="Helvetica-Bold", alignment=TA_CENTER))]], colWidths=[W])
    t.setStyle(TableStyle([("BACKGROUND",(0,0),(-1,-1),bg),
        ("TOPPADDING",(0,0),(-1,-1),6),("BOTTOMPADDING",(0,0),(-1,-1),6),
        ("LEFTPADDING",(0,0),(-1,-1),8)]))
    return t

def box(txt, bg=LT_BLUE, border=BLUE):
    t = Table([[Paragraph(txt, S("b", fontSize=9.5, textColor=DARK_GRAY,
               fontName="Helvetica", leading=14))]], colWidths=[W])
    t.setStyle(TableStyle([("BACKGROUND",(0,0),(-1,-1),bg),
        ("BOX",(0,0),(-1,-1),1.5,border),
        ("TOPPADDING",(0,0),(-1,-1),6),("BOTTOMPADDING",(0,0),(-1,-1),6),
        ("LEFTPADDING",(0,0),(-1,-1),9),("RIGHTPADDING",(0,0),(-1,-1),9)]))
    return t

def mnem(text, sub=""):
    rows = [[Paragraph(text, S("m", fontSize=11, textColor=PURPLE,
                               fontName="Helvetica-Bold", alignment=TA_CENTER))]]
    if sub:
        rows.append([Paragraph(sub, S("ms", fontSize=9, textColor=DARK_GRAY,
                                      fontName="Helvetica-Oblique", alignment=TA_CENTER))])
    t = Table(rows, colWidths=[W])
    t.setStyle(TableStyle([("BACKGROUND",(0,0),(-1,-1),LT_PURPLE),
        ("BOX",(0,0),(-1,-1),2,PURPLE),
        ("TOPPADDING",(0,0),(-1,-1),7),("BOTTOMPADDING",(0,0),(-1,-1),7),
        ("LEFTPADDING",(0,0),(-1,-1),8)]))
    return t

def p(txt, fs=9.5, bold=False, color=DARK_GRAY, align=TA_LEFT, leading=14):
    fn = "Helvetica-Bold" if bold else "Helvetica"
    return Paragraph(txt, S("p", fontSize=fs, textColor=color,
                             fontName=fn, leading=leading, alignment=align))

def sp(h=0.2): return Spacer(1, h*cm)

story = []

# TITLE
tb = Table([
    [p("STOMA CARE", fs=20, bold=True, color=WHITE, align=TA_CENTER)],
    [p("BSc Nursing 3rd Semester  |  MSN - Unit V (GI System)", fs=10,
       color=colors.HexColor("#BFDBFE"), align=TA_CENTER)],
    [p("QUICK REVISION NOTES", fs=9, bold=True,
       color=colors.HexColor("#FDE68A"), align=TA_CENTER)],
], colWidths=[W])
tb.setStyle(TableStyle([("BACKGROUND",(0,0),(-1,-1),DARK_BLUE),
    ("TOPPADDING",(0,0),(-1,-1),5),("BOTTOMPADDING",(0,0),(-1,-1),5)]))
story += [tb, sp(0.2)]

# 1. DEFINITION
story += [hdr("1.  WHAT IS A STOMA?"), sp(0.12)]
story.append(box("<b>STOMA</b> (Greek = 'mouth/opening') = A <b>surgically created opening on the abdominal wall</b> "
    "that allows waste (stool/urine) to exit when the normal route is blocked.<br/>"
    "<b>Indications:</b> Colorectal cancer | Crohn's / UC | Bowel obstruction | Trauma | "
    "Hirschsprung's disease | Bladder cancer | After removal of rectum"))
story.append(sp(0.2))

# 2. TYPES TABLE
story += [hdr("2.  TYPES OF STOMA", bg=TEAL), sp(0.12)]
td = [
    [p("<b>Type</b>",bold=True,color=WHITE), p("<b>Organ</b>",bold=True,color=WHITE),
     p("<b>Side</b>",bold=True,color=WHITE), p("<b>Output</b>",bold=True,color=WHITE),
     p("<b>Bag</b>",bold=True,color=WHITE)],
    [p("<b>Colostomy</b>",bold=True,color=TEAL), p("Large intestine"), p("LEFT"),
     p("Semi-formed"), p("Closed bag")],
    [p("<b>Ileostomy</b>",bold=True,color=BLUE), p("Small intestine"), p("RIGHT"),
     p("Liquid/watery"), p("Drainable")],
    [p("<b>Urostomy</b>",bold=True,color=PURPLE), p("Urinary tract"), p("Either"),
     p("Urine"), p("Drainable")],
]
tt = Table(td, colWidths=[3.2*cm, 3.8*cm, 2.5*cm, 4.5*cm, 3.8*cm])
tt.setStyle(TableStyle([("BACKGROUND",(0,0),(-1,0),TEAL),
    ("BACKGROUND",(0,1),(-1,1),LT_TEAL),("BACKGROUND",(0,2),(-1,2),LT_BLUE),
    ("BACKGROUND",(0,3),(-1,3),LT_PURPLE),
    ("BOX",(0,0),(-1,-1),1,TEAL),("INNERGRID",(0,0),(-1,-1),0.5,colors.HexColor("#99F6E4")),
    ("TOPPADDING",(0,0),(-1,-1),5),("BOTTOMPADDING",(0,0),(-1,-1),5),
    ("LEFTPADDING",(0,0),(-1,-1),6),("VALIGN",(0,0),(-1,-1),"MIDDLE")]))
story += [tt, sp(0.1)]
story.append(mnem("Ileostomy = 'I' = rIght = lIquid   |   Colostomy = 'C' = Colon = Left = formed stool"))
story.append(sp(0.2))

# 3. COLOSTOMY vs ILEOSTOMY
story += [hdr("3.  COLOSTOMY  vs  ILEOSTOMY", bg=ORANGE), sp(0.12)]
dd = [
    [p("<b>Feature</b>",bold=True,color=WHITE), p("<b>Colostomy</b>",bold=True,color=WHITE),
     p("<b>Ileostomy</b>",bold=True,color=WHITE)],
    [p("Output consistency"), p("Semi-formed stool"), p("Liquid / watery")],
    [p("Daily output"), p("300–500 ml"), p("500–1500 ml")],
    [p("Skin irritation"), p("Less (alkaline pH)"), p("<b>MORE</b> (digestive enzymes)")],
    [p("Bag type"), p("Closed/non-drainable"), p("<b>Drainable</b> bag")],
    [p("Bag change frequency"), p("2–3 times/day"), p("Every 48 hours")],
    [p("Stoma appearance"), p("Flat, round"), p("<b>Spouted</b> (protruding)")],
    [p("Dehydration risk"), p("Low"), p("<b>HIGH</b> — drink extra fluids!")],
    [p("Irrigation possible?"), p("Yes (sigmoid)"), p("No")],
]
dt = Table(dd, colWidths=[4.5*cm, 6.5*cm, 6.8*cm])
dt.setStyle(TableStyle([("BACKGROUND",(0,0),(-1,0),ORANGE),
    ("ROWBACKGROUNDS",(0,1),(-1,-1),[WHITE,LT_ORANGE]),
    ("BOX",(0,0),(-1,-1),1,ORANGE),("INNERGRID",(0,0),(-1,-1),0.5,colors.HexColor("#FED7AA")),
    ("TOPPADDING",(0,0),(-1,-1),5),("BOTTOMPADDING",(0,0),(-1,-1),5),
    ("LEFTPADDING",(0,0),(-1,-1),6),("VALIGN",(0,0),(-1,-1),"MIDDLE")]))
story += [dt, sp(0.2)]

# 4. PROCEDURE
story += [hdr("4.  STOMA CARE PROCEDURE  (Step-by-Step)", bg=GREEN), sp(0.12)]
steps = [
    ("STEP 1","REMOVE OLD BAG",LT_RED,RED,
     "Adhesive remover wipe → peel gently top to bottom → empty drainable bag first → dispose in clinical waste"),
    ("STEP 2","CLEAN STOMA",LT_GREEN,GREEN,
     "Warm water + soft gauze → circular motion from stoma outward → mild soap if needed → rinse well → PAT DRY"),
    ("STEP 3","ASSESS STOMA",LT_YELLOW,YELLOW,
     "Colour (pink-red=healthy) | Size | Shape | Protrusion | Peristomal skin condition"),
    ("STEP 4","MEASURE + CUT FLANGE",LT_BLUE,BLUE,
     "Use stoma measuring guide → cut hole 2–3 mm LARGER than stoma (too small=pressure | too large=leakage)"),
    ("STEP 5","APPLY BARRIER PASTE",LT_PURPLE,PURPLE,
     "Karaya/stomahesive paste around opening → protects skin from leakage and irritation"),
    ("STEP 6","APPLY NEW APPLIANCE",LT_TEAL,TEAL,
     "Centre over stoma → press firmly outward → hold 30 sec with warm hand → 2-piece: wafer first, then click bag → check for leaks"),
    ("STEP 7","DOCUMENT",colors.HexColor("#F1F5F9"),MID_GRAY,
     "Record: stoma colour, size, output, skin condition, bag changed, patient tolerance"),
]
for num, title, bg, border, detail in steps:
    r = Table([[p(f"<b>{num}</b>",fs=8.5,bold=True,color=border,align=TA_CENTER),
                p(f"<b>{title}</b>",fs=9.5,bold=True,color=border),
                p(detail,fs=9)]], colWidths=[1.5*cm, 4.0*cm, 12.3*cm])
    r.setStyle(TableStyle([("BACKGROUND",(0,0),(-1,-1),bg),
        ("BOX",(0,0),(-1,-1),1,border),
        ("TOPPADDING",(0,0),(-1,-1),5),("BOTTOMPADDING",(0,0),(-1,-1),5),
        ("LEFTPADDING",(0,0),(-1,-1),6),("VALIGN",(0,0),(-1,-1),"MIDDLE")]))
    story += [r, sp(0.07)]
story.append(sp(0.2))

# 5. HEALTHY vs DANGER
story += [hdr("5.  HEALTHY STOMA  vs  DANGER SIGNS", bg=RED), sp(0.12)]
hd = [
    [p("<b>Healthy Stoma</b>",bold=True,color=WHITE), p("<b>Report Immediately</b>",bold=True,color=WHITE)],
    [p("Pink-red colour"), p("<b>Dark red / purple / BLACK</b> = Ischaemia/Necrosis")],
    [p("Moist and glistening"), p("Dry, dull appearance")],
    [p("Slightly raised (ileostomy)"), p("Retracted below skin level")],
    [p("Minimal touch bleeding"), p("Profuse or continuous bleeding")],
    [p("Normal output"), p("<b>No output 4–6 hrs</b> = Obstruction")],
    [p("Intact peristomal skin"), p("Red, raw, broken peristomal skin")],
]
ht = Table(hd, colWidths=[8.8*cm, 9.0*cm])
ht.setStyle(TableStyle([("BACKGROUND",(0,0),(-1,0),RED),
    ("BACKGROUND",(0,1),(0,-1),LT_GREEN),("BACKGROUND",(1,1),(1,-1),LT_RED),
    ("BOX",(0,0),(-1,-1),1,RED),("INNERGRID",(0,0),(-1,-1),0.5,colors.HexColor("#FECACA")),
    ("TOPPADDING",(0,0),(-1,-1),5),("BOTTOMPADDING",(0,0),(-1,-1),5),
    ("LEFTPADDING",(0,0),(-1,-1),7),("VALIGN",(0,0),(-1,-1),"MIDDLE")]))
story += [ht, sp(0.2)]

# 6. COMPLICATIONS
story += [hdr("6.  COMPLICATIONS  OF  STOMA  (PRSS-HF)", bg=PURPLE), sp(0.12)]
story.append(mnem('Mnemonic:  "P R S S - H F"',
    "Prolapse | Retraction | Stenosis | Skin excoriation | Hernia | Fistula"))
story.append(sp(0.12))
cd = [
    [p("<b>Complication</b>",bold=True,color=WHITE),
     p("<b>What Happens</b>",bold=True,color=WHITE),
     p("<b>Action</b>",bold=True,color=WHITE)],
    [p("<b>Prolapse</b>",bold=True), p("Stoma protrudes excessively"), p("Reduce + surgical review")],
    [p("<b>Retraction</b>",bold=True), p("Stoma sinks below skin"), p("Difficult to fit bag → surgery")],
    [p("<b>Stenosis</b>",bold=True), p("Narrowing of stoma opening"), p("Dilation or surgery")],
    [p("<b>Skin Excoriation</b>",bold=True), p("Peristomal skin raw/sore"), p("Correct bag size, barrier cream")],
    [p("<b>Parastomal Hernia</b>",bold=True), p("Bulge around stoma"), p("Support belt + surgery")],
    [p("<b>Ischaemia/Necrosis</b>",bold=True,color=RED), p("Stoma turns BLACK",color=RED),
     p("EMERGENCY — call doctor NOW!",color=RED,bold=True)],
    [p("<b>Obstruction</b>",bold=True), p("No output 4–6 hours"), p("Hydration, position, surgery")],
    [p("<b>Fistula</b>",bold=True), p("Abnormal tract near stoma"), p("Surgical management")],
]
ct = Table(cd, colWidths=[3.8*cm, 7.2*cm, 6.8*cm])
ct.setStyle(TableStyle([("BACKGROUND",(0,0),(-1,0),PURPLE),
    ("ROWBACKGROUNDS",(0,1),(-1,-1),[WHITE,LT_PURPLE]),
    ("BOX",(0,0),(-1,-1),1,PURPLE),("INNERGRID",(0,0),(-1,-1),0.5,colors.HexColor("#DDD6FE")),
    ("TOPPADDING",(0,0),(-1,-1),5),("BOTTOMPADDING",(0,0),(-1,-1),5),
    ("LEFTPADDING",(0,0),(-1,-1),6),("VALIGN",(0,0),(-1,-1),"TOP")]))
story += [ct, sp(0.2)]

# 7. PATIENT EDUCATION
story += [hdr("7.  PATIENT EDUCATION  —  DISCHARGE TEACHING", bg=YELLOW), sp(0.12)]
ed = [
    [p("<b>Topic</b>",bold=True,color=WHITE), p("<b>Key Points</b>",bold=True,color=WHITE)],
    [p("<b>Bag Change</b>",bold=True), p("Change when 1/3–1/2 full. Ileostomy: every 48 hrs. Colostomy: 2–3×/day.")],
    [p("<b>Diet</b>",bold=True), p("Avoid gas-forming foods (beans, cabbage). Ileostomy: DRINK MORE WATER (dehydration risk).")],
    [p("<b>Activity</b>",bold=True), p("Normal life allowed — work, swim, exercise. Avoid heavy lifting 6 weeks post-op.")],
    [p("<b>Skin Care</b>",bold=True), p("Keep peristomal skin clean and dry. Use barrier cream daily.")],
    [p("<b>Report Urgently</b>",bold=True,color=RED), p("Dark/black stoma | No output 4–6 hrs | Profuse bleeding | Severe skin breakdown")],
    [p("<b>Medications</b>",bold=True), p("Some tablets pass whole into bag — ask for liquid/soluble forms.")],
    [p("<b>Support</b>",bold=True), p("Contact stoma care nurse or colostomy association for help with appliances and lifestyle.")],
]
et = Table(ed, colWidths=[3.5*cm, 14.3*cm])
et.setStyle(TableStyle([("BACKGROUND",(0,0),(-1,0),YELLOW),
    ("ROWBACKGROUNDS",(0,1),(-1,-1),[WHITE,LT_YELLOW]),
    ("BOX",(0,0),(-1,-1),1,YELLOW),("INNERGRID",(0,0),(-1,-1),0.5,colors.HexColor("#FDE68A")),
    ("TOPPADDING",(0,0),(-1,-1),5),("BOTTOMPADDING",(0,0),(-1,-1),5),
    ("LEFTPADDING",(0,0),(-1,-1),7),("VALIGN",(0,0),(-1,-1),"TOP")]))
story += [et, sp(0.2)]

# 8. EXAM QUESTIONS
story += [hdr("8.  LIKELY EXAM QUESTIONS", bg=RED), sp(0.12)]
qs = [
    ("5 marks","Define stoma. Classify types and compare colostomy vs ileostomy."),
    ("10 marks","Describe the nursing care procedure for a patient with colostomy."),
    ("5 marks","What are the complications of stoma? How are they managed?"),
    ("10 marks","Write a nursing care plan for a patient with colostomy."),
    ("5 marks","What health education will you give to a patient discharged with a colostomy?"),
    ("3 marks","What is colostomy irrigation? Explain the procedure."),
    ("2 marks","Differentiate between prolapse and retraction of stoma."),
]
qd = [[p("<b>Marks</b>",bold=True,color=WHITE,align=TA_CENTER), p("<b>Question</b>",bold=True,color=WHITE)]]
for m, q in qs:
    qd.append([p(f"<b>{m}</b>",bold=True,color=RED,align=TA_CENTER), p(q)])
qt = Table(qd, colWidths=[2.5*cm, 15.3*cm])
qt.setStyle(TableStyle([("BACKGROUND",(0,0),(-1,0),RED),
    ("ROWBACKGROUNDS",(0,1),(-1,-1),[WHITE,LT_RED]),
    ("BOX",(0,0),(-1,-1),1,RED),("INNERGRID",(0,0),(-1,-1),0.5,colors.HexColor("#FECACA")),
    ("TOPPADDING",(0,0),(-1,-1),5),("BOTTOMPADDING",(0,0),(-1,-1),5),
    ("LEFTPADDING",(0,0),(-1,-1),7),("VALIGN",(0,0),(-1,-1),"MIDDLE"),
    ("ALIGN",(0,0),(0,-1),"CENTER")]))
story += [qt, sp(0.15)]

# FOOTER
ft = Table([[p("BSc Nursing 3rd Semester  •  MSN Unit V  •  Stoma Care  •  Quick Revision Notes  •  All the best! 💪",
               fs=8, color=WHITE, align=TA_CENTER)]], colWidths=[W])
ft.setStyle(TableStyle([("BACKGROUND",(0,0),(-1,-1),DARK_BLUE),
    ("TOPPADDING",(0,0),(-1,-1),6),("BOTTOMPADDING",(0,0),(-1,-1),6)]))
story.append(ft)

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