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

OUTPUT = "/home/daytona/workspace/study-plan/Med_Summer_Study_Plan.pdf"

# ── Colour palette ──────────────────────────────────────────────────────────
C_DARK_BLUE   = colors.HexColor("#1A237E")
C_MED_BLUE    = colors.HexColor("#1565C0")
C_LIGHT_BLUE  = colors.HexColor("#E3F2FD")
C_TEAL        = colors.HexColor("#00695C")
C_TEAL_LIGHT  = colors.HexColor("#E0F2F1")
C_GREEN       = colors.HexColor("#2E7D32")
C_GREEN_LIGHT = colors.HexColor("#E8F5E9")
C_ORANGE      = colors.HexColor("#E65100")
C_ORANGE_LIGHT= colors.HexColor("#FFF3E0")
C_PURPLE      = colors.HexColor("#4A148C")
C_PURPLE_LIGHT= colors.HexColor("#F3E5F5")
C_RED         = colors.HexColor("#B71C1C")
C_RED_LIGHT   = colors.HexColor("#FFEBEE")
C_GREY_DARK   = colors.HexColor("#37474F")
C_GREY_MED    = colors.HexColor("#78909C")
C_GREY_LIGHT  = colors.HexColor("#ECEFF1")
C_WHITE       = colors.white
C_YELLOW      = colors.HexColor("#F9A825")
C_YELLOW_LIGHT= colors.HexColor("#FFFDE7")

doc = SimpleDocTemplate(
    OUTPUT,
    pagesize=A4,
    rightMargin=1.8*cm, leftMargin=1.8*cm,
    topMargin=1.8*cm, bottomMargin=1.8*cm,
    title="Med School Summer Study Plan 2026",
    author="Orris AI"
)

W = A4[0] - 3.6*cm   # usable page width

styles = getSampleStyleSheet()

def style(name, **kw):
    s = ParagraphStyle(name, **kw)
    return s

S_TITLE = style("Title2",
    fontName="Helvetica-Bold", fontSize=22, textColor=C_WHITE,
    alignment=TA_CENTER, spaceAfter=4, leading=28)

S_SUBTITLE = style("Subtitle2",
    fontName="Helvetica", fontSize=11, textColor=colors.HexColor("#B3E5FC"),
    alignment=TA_CENTER, spaceAfter=2, leading=14)

S_SECTION = style("Section",
    fontName="Helvetica-Bold", fontSize=13, textColor=C_WHITE,
    alignment=TA_LEFT, leading=17, spaceAfter=2)

S_PHASE = style("Phase",
    fontName="Helvetica-Bold", fontSize=11, textColor=C_WHITE,
    alignment=TA_LEFT, leading=14)

S_BODY = style("Body2",
    fontName="Helvetica", fontSize=9, textColor=C_GREY_DARK,
    leading=13, spaceAfter=2)

S_SMALL = style("Small",
    fontName="Helvetica", fontSize=8, textColor=C_GREY_DARK, leading=11)

S_BOLD_SMALL = style("BoldSmall",
    fontName="Helvetica-Bold", fontSize=8, textColor=C_GREY_DARK, leading=11)

S_TH = style("TH",
    fontName="Helvetica-Bold", fontSize=8, textColor=C_WHITE,
    alignment=TA_CENTER, leading=11)

S_TD = style("TD",
    fontName="Helvetica", fontSize=7.5, textColor=C_GREY_DARK,
    leading=10, spaceAfter=1)

S_TD_BOLD = style("TDBold",
    fontName="Helvetica-Bold", fontSize=7.5, textColor=C_DARK_BLUE,
    leading=10)

S_NOTE = style("Note",
    fontName="Helvetica-Oblique", fontSize=8, textColor=C_TEAL,
    leading=11, spaceAfter=2)

S_FOOTER = style("Footer",
    fontName="Helvetica", fontSize=7, textColor=C_GREY_MED,
    alignment=TA_CENTER, leading=9)

def header_table(text, bg=C_DARK_BLUE, sub=None):
    """Full-width coloured header bar."""
    inner = [Paragraph(text, S_SECTION)]
    if sub:
        inner.append(Paragraph(sub, S_NOTE))
    t = Table([[inner]], colWidths=[W])
    t.setStyle(TableStyle([
        ("BACKGROUND", (0,0), (-1,-1), bg),
        ("LEFTPADDING",  (0,0), (-1,-1), 10),
        ("RIGHTPADDING", (0,0), (-1,-1), 10),
        ("TOPPADDING",   (0,0), (-1,-1), 7),
        ("BOTTOMPADDING",(0,0), (-1,-1), 7),
        ("ROUNDEDCORNERS", [4]),
    ]))
    return t

def pill(text, bg, fg=C_WHITE):
    t = Table([[Paragraph(text, style("pill", fontName="Helvetica-Bold",
               fontSize=7.5, textColor=fg, leading=10))]])
    t.setStyle(TableStyle([
        ("BACKGROUND",   (0,0),(-1,-1), bg),
        ("LEFTPADDING",  (0,0),(-1,-1), 5),
        ("RIGHTPADDING", (0,0),(-1,-1), 5),
        ("TOPPADDING",   (0,0),(-1,-1), 2),
        ("BOTTOMPADDING",(0,0),(-1,-1), 2),
    ]))
    return t

# ── Data ───────────────────────────────────────────────────────────────────
phases = [
    {
        "label": "WEEK 0  |  Days 1–4",
        "title": "HISTOLOGY FOUNDATION",
        "color": C_TEAL,
        "light": C_TEAL_LIGHT,
        "goal": "Understand normal tissue before studying disease",
        "days": [
            ("Day 1", "Epithelium & Connective Tissue",
             "Cell types, cell junctions, ECM, collagen, ground substance",
             "Wheater's Functional Histology – Ch. 1-3"),
            ("Day 2", "Muscle, Nerve, Cartilage & Bone",
             "Skeletal/smooth/cardiac muscle ultrastructure; neuron types; bone remodelling",
             "Wheater's Ch. 4-6 + online slides"),
            ("Day 3", "Glands, Blood Vessels & Lymphoid Tissue",
             "Exocrine vs endocrine; artery/vein/capillary walls; lymph node architecture",
             "Wheater's Ch. 7-9"),
            ("Day 4", "Review & Slide Recognition",
             "Go through all 3 days using the free Histology Guide website (histologyguide.com)",
             "Histology Guide online – all organ systems gallery"),
        ],
        "tip": "Tip: Focus on RECOGNISING tissues visually, not memorising names of every cell. Ask yourself 'what tissue is this?' when you see a slide.",
    },
    {
        "label": "WEEKS 1–2  |  Days 5–18",
        "title": "CARDIOVASCULAR SYSTEM",
        "color": C_RED,
        "light": C_RED_LIGHT,
        "goal": "Master the system you'll encounter most in clinical medicine",
        "days": [
            ("Days 5–6", "Cardiac Physiology",
             "Cardiac cycle, action potentials, ECG basics, Frank-Starling law, cardiac output",
             "Guyton & Hall Review – Cardiac Physiology chapters"),
            ("Days 7–8", "Vascular Physiology & BP",
             "Arterial pressure, autoregulation, RAAS, baroreceptors, capillary exchange",
             "Guyton & Hall Review – Vascular chapters"),
            ("Days 9–10", "Heart Anatomy",
             "Chambers, valves, coronary arteries, cardiac conduction system, pericardium",
             "Netter's Atlas – Heart plates + Snell's Clinical Anatomy"),
            ("Days 11–12", "Ischaemic Heart Disease & MI",
             "Atherosclerosis pathogenesis, stable/unstable angina, STEMI/NSTEMI, cardiac enzymes",
             "Robbins Basic Pathology – Cardiovascular chapter"),
            ("Days 13–14", "Heart Failure, Valvular Disease & Cardiomyopathies",
             "Systolic vs diastolic HF, mitral/aortic stenosis and regurgitation, DCM/HCM/RCM",
             "Robbins Basic Pathology – continued"),
            ("Days 15–16", "Infective Endocarditis & Rheumatic Fever",
             "Duke criteria, organisms (Strep viridans, Staph aureus), RHD pathology",
             "Microbiology Made Ridiculously Simple – relevant chapters"),
            ("Days 17–18", "Lipids & Biochemistry",
             "Lipoprotein metabolism, LDL/HDL, statins MOA, cardiac biomarkers (troponin, CK-MB, BNP)",
             "Lippincott's Biochemistry – Lipid chapters"),
        ],
        "tip": "Tip: Draw the cardiac cycle from memory each day. By Day 18, you should be able to explain why heart sounds change in valve disease.",
    },
    {
        "label": "WEEKS 3 + VIROLOGY INTRO  |  Days 19–32",
        "title": "RESPIRATORY SYSTEM + GENERAL VIROLOGY",
        "color": C_MED_BLUE,
        "light": C_LIGHT_BLUE,
        "goal": "Breathe easy in respiratory cases; understand viruses as a class",
        "days": [
            ("Days 19–20", "General Virology",
             "Virus structure, DNA vs RNA viruses, replication cycles, classification overview, antiviral drugs",
             "Microbiology Made Ridiculously Simple – Virology intro"),
            ("Days 21–22", "Respiratory Physiology",
             "Lung volumes/capacities, V/Q ratio, gas exchange, oxygen-haemoglobin curve, acid-base basics",
             "Guyton & Hall Review – Respiratory chapters"),
            ("Days 23–24", "Respiratory Anatomy",
             "Lungs, pleura, mediastinum, diaphragm, bronchopulmonary segments",
             "Netter's Atlas – Respiratory plates"),
            ("Days 25–26", "Pneumonia & TB",
             "Lobar vs bronchopneumonia, organisms, TB pathology (Ghon complex, caseating granuloma)",
             "Robbins Basic Pathology + Micro Made Ridiculously Simple"),
            ("Days 27–28", "COPD, Asthma & Interstitial Lung Disease",
             "Emphysema vs chronic bronchitis, asthma pathophysiology, fibrosis patterns",
             "Robbins Basic Pathology – Lung chapter"),
            ("Days 29–30", "Respiratory Viruses",
             "Influenza (H and N antigens, antigenic drift/shift), RSV, COVID-19, adenovirus",
             "Microbiology Made Ridiculously Simple – Respiratory viruses"),
            ("Days 31–32", "Lung Cancer & Pleural Disease",
             "Squamous, adeno, small cell, large cell – key features; pleural effusion, pneumothorax",
             "Robbins Basic Pathology – continued"),
        ],
        "tip": "Tip: For every respiratory pathology, link it to a lung function test pattern (obstructive vs restrictive). This is asked constantly in Year 3.",
    },
    {
        "label": "WEEK 3.5  |  Days 33–37",
        "title": "IMMUNOLOGY BLOCK",
        "color": C_PURPLE,
        "light": C_PURPLE_LIGHT,
        "goal": "Build the immunological toolkit you'll use in every clinical subject",
        "days": [
            ("Day 33", "Innate Immunity",
             "Physical barriers, complement system (classical/lectin/alternative), NK cells, inflammation mediators",
             "Immunology Made Ridiculously Simple – Ch. 1-2"),
            ("Day 34", "Adaptive Immunity – T Cells",
             "T cell development, MHC I vs II, CD4/CD8, Th1/Th2/Th17/Treg, cytotoxic killing",
             "Immunology Made Ridiculously Simple – Ch. 3-4"),
            ("Day 35", "Adaptive Immunity – B Cells & Antibodies",
             "B cell activation, immunoglobulin classes (IgM/IgG/IgA/IgE/IgD), somatic hypermutation, affinity maturation",
             "Immunology Made Ridiculously Simple – Ch. 5"),
            ("Day 36", "Hypersensitivity Reactions",
             "Type I (anaphylaxis/atopy), Type II (cytotoxic), Type III (immune complex), Type IV (delayed/DTH) – examples of each",
             "Robbins Basic Pathology – Immunology chapter"),
            ("Day 37", "Immunodeficiency & Autoimmunity",
             "Primary immunodeficiencies (SCID, DiGeorge, Bruton's), HIV/AIDS immune destruction, autoimmunity mechanisms",
             "Robbins Basic Pathology – continued"),
        ],
        "tip": "Tip: Make a table for hypersensitivity types (I-IV) with: mechanism, mediator, example disease, and time course. This comes up in EVERY clinical subject.",
    },
    {
        "label": "WEEKS 4–5  |  Days 38–51",
        "title": "GASTROINTESTINAL SYSTEM + LIVER",
        "color": C_GREEN,
        "light": C_GREEN_LIGHT,
        "goal": "Navigate the most common reason patients visit hospital",
        "days": [
            ("Days 38–39", "GI Physiology",
             "Swallowing, gastric secretion (HCl, pepsin, intrinsic factor), intestinal absorption, gut motility",
             "Guyton & Hall Review – GI chapters"),
            ("Days 40–41", "GI Anatomy",
             "Oesophagus to rectum, peritoneum, mesenteries, blood supply (coeliac/SMA/IMA), inguinal canal",
             "Netter's Atlas – GI plates + Snell's"),
            ("Days 42–43", "Peptic Ulcer Disease & Gastritis",
             "H. pylori pathogenesis, NSAIDs, acid-pepsin balance, complications (perforation, bleeding)",
             "Robbins Basic Pathology + Micro Made Ridiculously Simple"),
            ("Days 44–45", "IBD, Colorectal Cancer & Malabsorption",
             "Crohn's vs UC (key differences), CRC staging, coeliac disease, lactase deficiency",
             "Robbins Basic Pathology – GI chapter"),
            ("Days 46–47", "Liver Biochemistry",
             "Bilirubin metabolism, LFTs interpretation, albumin synthesis, clotting factors, ammonia metabolism",
             "Lippincott's Biochemistry – Liver/nitrogen chapters"),
            ("Days 48–49", "Hepatitis Viruses & Liver Pathology",
             "Hep A/B/C/D/E – transmission, serology (HBsAg, anti-HBc, etc.), chronicity, cirrhosis",
             "Microbiology Made Ridiculously Simple + Robbins"),
            ("Days 50–51", "Cirrhosis, Portal Hypertension & Pancreas",
             "Complications of cirrhosis (ascites, varices, HE, SBP), acute/chronic pancreatitis, pancreatic cancer",
             "Robbins Basic Pathology – Liver/Pancreas chapters"),
        ],
        "tip": "Tip: Learn the serology of Hepatitis B cold. You WILL be asked to interpret HBsAg/anti-HBs/HBeAg/anti-HBc in clinical years.",
    },
    {
        "label": "WEEKS 6–7  |  Days 52–65",
        "title": "RENAL SYSTEM + ENDOCRINOLOGY",
        "color": C_ORANGE,
        "light": C_ORANGE_LIGHT,
        "goal": "Interpret lab results and understand metabolic disease",
        "days": [
            ("Days 52–53", "Renal Physiology",
             "GFR, tubular reabsorption/secretion, countercurrent mechanism, ADH/aldosterone regulation",
             "Guyton & Hall Review – Renal chapters"),
            ("Days 54–55", "Acid-Base & Electrolytes",
             "pH/HCO3/pCO2 interpretation, metabolic vs respiratory acidosis/alkalosis, Na+/K+ disorders",
             "Guyton & Hall Review – Acid-Base chapters"),
            ("Days 56–57", "Renal Pathology",
             "Nephrotic vs nephritic syndrome, GN types, AKI vs CKD, UTI/pyelonephritis, renal cell carcinoma",
             "Robbins Basic Pathology – Renal chapter"),
            ("Days 58", "Renal Anatomy",
             "Kidney, ureter, bladder anatomy; renal vasculature; juxtaglomerular apparatus",
             "Netter's Atlas – Renal plates"),
            ("Days 59–60", "Endocrine Physiology",
             "Hypothalamic-pituitary axis, negative feedback, thyroid hormones (T3/T4), adrenal cortex/medulla",
             "Guyton & Hall Review – Endocrine chapters"),
            ("Days 61–62", "Diabetes Mellitus",
             "Type 1 vs 2 pathogenesis, insulin signalling, DKA vs HHS, long-term complications (micro/macrovascular)",
             "Robbins Basic Pathology + Lippincott's Biochem"),
            ("Days 63–64", "Thyroid, Adrenal & Pituitary Disease",
             "Hypo/hyperthyroidism, Graves' vs Hashimoto's, Cushing's vs Addison's, pituitary adenomas",
             "Robbins Basic Pathology – Endocrine chapter"),
            ("Days 65", "Endocrine Biochemistry",
             "Glucose metabolism (glycolysis, gluconeogenesis, glycogen), HbA1c, steroid hormone synthesis",
             "Lippincott's Biochemistry – Carbohydrate/steroid chapters"),
        ],
        "tip": "Tip: Master the acid-base algorithm. Internists use it daily. Practice with 10 blood gas questions – I can give you these.",
    },
    {
        "label": "WEEK 7.5  |  Days 66–70",
        "title": "GENETICS BLOCK",
        "color": C_TEAL,
        "light": C_TEAL_LIGHT,
        "goal": "Understand inheritance patterns before you see patients with genetic conditions",
        "days": [
            ("Day 66", "Mendelian Inheritance",
             "Autosomal dominant (Marfan's, achondroplasia), autosomal recessive (CF, PKU, sickle cell), X-linked (haemophilia, DMD)",
             "Thompson & Thompson Genetics – Ch. 1-4"),
            ("Day 67", "Chromosomal Disorders",
             "Trisomies (Down's/Patau's/Edwards'), monosomy X (Turner's), 47XXY (Klinefelter's), deletions (DiGeorge, Prader-Willi)",
             "Thompson & Thompson – Ch. 5-6"),
            ("Day 68", "Molecular Genetics & Cancer Genetics",
             "Mutations (missense/nonsense/frameshift), oncogenes, tumour suppressor genes (p53, BRCA1/2), DNA repair",
             "Thompson & Thompson – Ch. 7-8 + Robbins Neoplasia chapter"),
            ("Day 69", "Multifactorial Inheritance & Genetic Testing",
             "Polygenic traits, heritability, twin studies, PCR, karyotyping, FISH, next-gen sequencing basics",
             "Thompson & Thompson – Ch. 9-10"),
            ("Day 70", "Clinical Genetics Integration",
             "When to suspect a genetic disorder; family history taking; genetic counselling principles",
             "Thompson & Thompson – Clinical cases"),
        ],
        "tip": "Tip: Draw pedigrees for each inheritance pattern and be able to calculate recurrence risks. This is a practical skill they'll test you on.",
    },
    {
        "label": "WEEK 8  |  Days 71–78",
        "title": "NEUROLOGY + NEURO VIRUSES",
        "color": C_DARK_BLUE,
        "light": C_LIGHT_BLUE,
        "goal": "Stop finding neurology terrifying — build the framework first",
        "days": [
            ("Days 71–72", "Neuroanatomy Overview",
             "CNS divisions, spinal cord tracts (dorsal columns vs spinothalamic), brain lobes, blood supply (Circle of Willis)",
             "Netter's Atlas – Neuroscience plates + Snell's Clinical Neuro"),
            ("Days 73–74", "Cranial Nerves & Reflexes",
             "All 12 cranial nerves (origin, function, lesion), deep tendon reflexes, UMN vs LMN signs",
             "Snell's Clinical Neuroanatomy"),
            ("Days 75–76", "Neurophysiology",
             "Resting membrane potential, action potential, synaptic transmission, neurotransmitters (ACh, DA, GABA, 5-HT)",
             "Guyton & Hall Review – Neuro chapters"),
            ("Day 77", "Stroke & Raised ICP",
             "Ischaemic vs haemorrhagic stroke, territory syndromes (MCA/ACA/PCA), Cushing's reflex, herniation",
             "Robbins Basic Pathology – CNS chapter"),
            ("Day 78", "Meningitis & Neuro Viruses",
             "Bacterial vs viral meningitis (organisms, CSF findings), HSV encephalitis, rabies, polio, HIV neuro",
             "Microbiology Made Ridiculously Simple – Neuro infections"),
        ],
        "tip": "Tip: Learn the UMN vs LMN signs table perfectly. Every single neurological case in Year 3 starts with: 'Is this upper or lower motor neuron?'",
    },
    {
        "label": "WEEK 9  |  Days 79–86",
        "title": "HAEMATOLOGY + HIV/AIDS",
        "color": C_RED,
        "light": C_RED_LIGHT,
        "goal": "Read a full blood count without panic and understand the killer virus",
        "days": [
            ("Days 79–80", "Haematopoiesis & Normal Blood",
             "Stem cell differentiation, RBC/WBC/platelet production, normal ranges, bone marrow structure",
             "Robbins Basic Pathology – Blood chapter"),
            ("Days 81–82", "Anaemias",
             "Microcytic (iron def, thalassaemia), macrocytic (B12/folate), haemolytic (G6PD, spherocytosis, autoimmune)",
             "Robbins Basic Pathology + Lippincott's Biochem (haem synthesis)"),
            ("Day 83", "Leukaemias & Lymphomas",
             "ALL/AML/CLL/CML key features and markers; Hodgkin's vs Non-Hodgkin's lymphoma; Reed-Sternberg cells",
             "Robbins Basic Pathology – Haematology chapter"),
            ("Day 84", "Coagulation & Bleeding Disorders",
             "Coagulation cascade (intrinsic/extrinsic/common), PT vs APTT, haemophilia A/B, von Willebrand disease, DIC",
             "Robbins Basic Pathology + Guyton physiology"),
            ("Days 85–86", "HIV/AIDS + EBV/CMV",
             "HIV structure, life cycle, CD4 decline, AIDS-defining illnesses, HAART MOA; EBV (mono, Burkitt's); CMV (immunocompromised)",
             "Microbiology Made Ridiculously Simple – Retroviridae + Herpesviridae"),
        ],
        "tip": "Tip: Memorise the coagulation cascade as a story, not a list. When PT is prolonged but APTT is normal – that's the extrinsic pathway. Know what that means.",
    },
    {
        "label": "WEEK 10  |  Days 87–75",
        "title": "REVISION + INTEGRATION WEEK",
        "color": C_GREY_DARK,
        "light": C_GREY_LIGHT,
        "goal": "Consolidate, connect the dots, and enter Year 3 with confidence",
        "days": [
            ("Days 87–88", "Weak Area Revisit",
             "Go back to whichever 2 systems felt weakest. Re-read key pathology. Do practice questions.",
             "Your notes + Robbins/Guyton as needed"),
            ("Days 89–90", "Cross-System Integration",
             "How does heart failure affect the kidneys? How does liver disease cause coagulopathy? Systemic thinking.",
             "Your own notes – synthesis exercise"),
            ("Days 91–92", "Clinical Correlation Review",
             "For each system: what does a typical patient presentation look like? What investigations would you order?",
             "Osmosis videos for each system"),
            ("Days 93–94", "Active Recall Sessions",
             "Close your notes. Write everything you remember about each phase. Check gaps. Fill them.",
             "Your own notes only"),
            ("Days 95–75", "Free Days / Rest",
             "Your brain needs consolidation time. Rest, go out, do something non-medical. This is NOT optional.",
             "N/A – full rest"),
        ],
        "tip": "Tip: Teach someone else (or pretend to). If you can explain nephrotic syndrome to a non-medical friend, you actually know it.",
    },
]

daily_template = [
    ("07:00 – 07:30", "Wake up, light breakfast, review yesterday's notes (30 min)"),
    ("07:30 – 09:30", "BLOCK 1 – New topic reading (textbook / notes)"),
    ("09:30 – 09:45", "Short break – stretch, coffee"),
    ("09:45 – 11:15", "BLOCK 2 – Watch Osmosis / Boards & Beyond video on same topic"),
    ("11:15 – 12:00", "BLOCK 3 – Make flashcards / summarise key points"),
    ("12:00 – 14:00", "LUNCH BREAK – proper break, no studying"),
    ("14:00 – 15:30", "BLOCK 4 – Practice questions or past papers on today's topic"),
    ("15:30 – 15:45", "Short break"),
    ("15:45 – 16:30", "BLOCK 5 – Review mistakes, look up anything unclear"),
    ("16:30 onwards",  "Evening free – social, exercise, rest"),
]

resources = [
    ("Pathology",      "Robbins Basic Pathology (9th/10th ed.)",     "Core disease understanding"),
    ("Physiology",     "Guyton & Hall Textbook of Medical Physiology – Review book", "Mechanisms of function"),
    ("Anatomy",        "Netter's Atlas of Human Anatomy",             "Visual spatial understanding"),
    ("Anatomy (clin)", "Snell's Clinical Anatomy by Regions",         "Clinical correlation focus"),
    ("Microbiology",   "Microbiology Made Ridiculously Simple",       "Bugs + drugs, easy to read"),
    ("Virology",       "Same book – Virology chapters",               "Part of the above"),
    ("Immunology",     "Immunology Made Ridiculously Simple",         "Best intro to immunology"),
    ("Biochemistry",   "Lippincott's Illustrated Reviews: Biochemistry","Metabolism & clinical biochem"),
    ("Histology",      "Wheater's Functional Histology",              "Atlas-style, best for slides"),
    ("Genetics",       "Thompson & Thompson Genetics in Medicine",    "Readable, well-structured"),
    ("Videos",         "Osmosis (YouTube/app) + Boards & Beyond",     "Watch BEFORE reading each topic"),
    ("Questions",      "Amboss (free tier) / Anki pre-made decks",    "Active recall essential"),
]

tips = [
    ("Watch first, read second",
     "For every new topic, watch a 10-15 min Osmosis video FIRST, then open the textbook. The video gives you a mental map so the reading makes sense."),
    ("Organ system thinking",
     "Never study a subject in isolation. For each organ: do the physiology, then anatomy, then pathology, then micro/biochem. Your brain links them."),
    ("One rest day per week",
     "Every Sunday (or your chosen day) = zero studying. Sleep, exercise, socialise. Burnout at week 4 ruins the whole plan."),
    ("Active recall over re-reading",
     "After every topic: close the book and write everything you remember. Check it. Fix gaps. This doubles retention vs re-reading."),
    ("Use me (Orris) daily",
     "Ask me to quiz you, explain anything confusing, or pull diagrams from the textbooks. I'm here throughout your vacation."),
    ("Don't aim for perfection",
     "You're building a foundation, not mastering everything. 70% understanding now + clinical exposure in Year 3 = full understanding."),
]

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

# ── COVER ───────────────────────────────────────────────────────────────────
cover_data = [[
    Paragraph("MEDICAL SCHOOL", style("cov1",
        fontName="Helvetica", fontSize=13, textColor=colors.HexColor("#90CAF9"),
        alignment=TA_CENTER, leading=16)),
    Paragraph("Summer Study Plan", style("cov2",
        fontName="Helvetica-Bold", fontSize=28, textColor=C_WHITE,
        alignment=TA_CENTER, leading=34, spaceAfter=4)),
    Paragraph("Year 2 → Year 3 Bridge  |  2.5 Months  |  4–5 Hours / Day", style("cov3",
        fontName="Helvetica", fontSize=11, textColor=colors.HexColor("#B3E5FC"),
        alignment=TA_CENTER, leading=14)),
    Spacer(1, 0.4*cm),
    Paragraph("Tbilisi, Georgia  •  Medical School  •  2026", style("cov4",
        fontName="Helvetica-Oblique", fontSize=9, textColor=colors.HexColor("#78909C"),
        alignment=TA_CENTER, leading=12)),
]]
ct = Table(cover_data, colWidths=[W])
ct.setStyle(TableStyle([
    ("BACKGROUND",   (0,0),(-1,-1), C_DARK_BLUE),
    ("LEFTPADDING",  (0,0),(-1,-1), 20),
    ("RIGHTPADDING", (0,0),(-1,-1), 20),
    ("TOPPADDING",   (0,0),(-1,-1), 30),
    ("BOTTOMPADDING",(0,0),(-1,-1), 30),
    ("ROUNDEDCORNERS", [6]),
]))
story.append(ct)
story.append(Spacer(1, 0.5*cm))

# ── OVERVIEW TABLE ──────────────────────────────────────────────────────────
story.append(header_table("📋  PLAN OVERVIEW", bg=C_GREY_DARK,
    sub="10 phases across 75 days — covering all pre-clinical subjects in organ system order"))
story.append(Spacer(1, 0.25*cm))

ov_header = [
    Paragraph("Phase", S_TH),
    Paragraph("Days", S_TH),
    Paragraph("System / Topic", S_TH),
    Paragraph("Subjects Covered", S_TH),
]
ov_rows = [
    ["Week 0",     "1–4",   "Histology Foundation",           "Histology"],
    ["Weeks 1–2",  "5–18",  "Cardiovascular System",          "Physio, Anatomy, Pathology, Micro, Biochem"],
    ["Week 3",     "19–32", "Respiratory + General Virology",  "Physio, Anatomy, Pathology, Virology"],
    ["Week 3.5",   "33–37", "Immunology Block",                "Immunology, Pathology"],
    ["Weeks 4–5",  "38–51", "GI System + Liver",              "Physio, Anatomy, Pathology, Micro, Biochem"],
    ["Weeks 6–7",  "52–65", "Renal + Endocrinology",          "Physio, Pathology, Biochem, Anatomy"],
    ["Week 7.5",   "66–70", "Genetics Block",                  "Genetics, Molecular Biology"],
    ["Week 8",     "71–78", "Neurology + Neuro Viruses",       "Neuroanatomy, Physio, Pathology, Virology"],
    ["Week 9",     "79–86", "Haematology + HIV/AIDS",          "Pathology, Biochem, Virology, Micro"],
    ["Week 10",    "87–75", "Revision + Integration",          "All subjects – consolidation"],
]
ov_colors = [C_TEAL, C_RED, C_MED_BLUE, C_PURPLE, C_GREEN,
             C_ORANGE, C_TEAL, C_DARK_BLUE, C_RED, C_GREY_DARK]

ov_table_data = [ov_header]
for row in ov_rows:
    ov_table_data.append([Paragraph(c, S_TD) for c in row])

ov_t = Table(ov_table_data, colWidths=[W*0.12, W*0.08, W*0.32, W*0.48])
ov_style = [
    ("BACKGROUND",    (0,0), (-1,0),  C_GREY_DARK),
    ("TEXTCOLOR",     (0,0), (-1,0),  C_WHITE),
    ("ROWBACKGROUNDS",(0,1), (-1,-1), [C_WHITE, C_GREY_LIGHT]),
    ("GRID",          (0,0), (-1,-1), 0.3, colors.HexColor("#CFD8DC")),
    ("LEFTPADDING",   (0,0), (-1,-1), 5),
    ("RIGHTPADDING",  (0,0), (-1,-1), 5),
    ("TOPPADDING",    (0,0), (-1,-1), 4),
    ("BOTTOMPADDING", (0,0), (-1,-1), 4),
    ("VALIGN",        (0,0), (-1,-1), "MIDDLE"),
]
for i, c in enumerate(ov_colors):
    ov_style.append(("BACKGROUND", (0, i+1), (0, i+1), c))
    ov_style.append(("TEXTCOLOR",  (0, i+1), (0, i+1), C_WHITE))
    ov_style.append(("FONTNAME",   (0, i+1), (0, i+1), "Helvetica-Bold"))

ov_t.setStyle(TableStyle(ov_style))
story.append(ov_t)
story.append(Spacer(1, 0.3*cm))

# ── DAILY ROUTINE ───────────────────────────────────────────────────────────
story.append(header_table("⏰  DAILY STUDY ROUTINE (4–5 Hours)", bg=C_MED_BLUE,
    sub="Use this template every weekday — adjust on weekends if needed"))
story.append(Spacer(1, 0.2*cm))

dr_data = [[Paragraph("Time", S_TH), Paragraph("Activity", S_TH)]]
for time, act in daily_template:
    dr_data.append([
        Paragraph(time, S_TD_BOLD),
        Paragraph(act, S_TD),
    ])
dr_t = Table(dr_data, colWidths=[W*0.28, W*0.72])
dr_t.setStyle(TableStyle([
    ("BACKGROUND",    (0,0), (-1,0),  C_MED_BLUE),
    ("ROWBACKGROUNDS",(0,1), (-1,-1), [C_WHITE, C_LIGHT_BLUE]),
    ("GRID",          (0,0), (-1,-1), 0.3, colors.HexColor("#CFD8DC")),
    ("LEFTPADDING",   (0,0), (-1,-1), 6),
    ("RIGHTPADDING",  (0,0), (-1,-1), 6),
    ("TOPPADDING",    (0,0), (-1,-1), 4),
    ("BOTTOMPADDING", (0,0), (-1,-1), 4),
    ("VALIGN",        (0,0), (-1,-1), "MIDDLE"),
]))
story.append(dr_t)
story.append(Spacer(1, 0.4*cm))

# ── RESOURCES ───────────────────────────────────────────────────────────────
story.append(header_table("📚  RECOMMENDED RESOURCES", bg=C_GREEN,
    sub="You don't need all of these — use the ones listed per phase"))
story.append(Spacer(1, 0.2*cm))

res_data = [[
    Paragraph("Subject", S_TH),
    Paragraph("Book / Resource", S_TH),
    Paragraph("Why Use It", S_TH),
]]
for subj, book, why in resources:
    res_data.append([
        Paragraph(subj, S_TD_BOLD),
        Paragraph(book, S_TD),
        Paragraph(why, S_TD),
    ])
res_t = Table(res_data, colWidths=[W*0.16, W*0.47, W*0.37])
res_t.setStyle(TableStyle([
    ("BACKGROUND",    (0,0), (-1,0),  C_GREEN),
    ("ROWBACKGROUNDS",(0,1), (-1,-1), [C_WHITE, C_GREEN_LIGHT]),
    ("GRID",          (0,0), (-1,-1), 0.3, colors.HexColor("#CFD8DC")),
    ("LEFTPADDING",   (0,0), (-1,-1), 5),
    ("RIGHTPADDING",  (0,0), (-1,-1), 5),
    ("TOPPADDING",    (0,0), (-1,-1), 4),
    ("BOTTOMPADDING", (0,0), (-1,-1), 4),
    ("VALIGN",        (0,0), (-1,-1), "MIDDLE"),
]))
story.append(res_t)
story.append(PageBreak())

# ── PHASE DETAILS ───────────────────────────────────────────────────────────
for phase in phases:
    col  = phase["color"]
    lite = phase["light"]

    # Phase header
    hdr = Table([[
        Paragraph(phase["label"], style("ph_label",
            fontName="Helvetica", fontSize=8, textColor=colors.HexColor("#BBDEFB"),
            leading=10)),
        Paragraph(phase["title"], S_SECTION),
        Paragraph(phase["goal"], style("ph_goal",
            fontName="Helvetica-Oblique", fontSize=8,
            textColor=colors.HexColor("#E3F2FD"), leading=10)),
    ]], colWidths=[W])
    hdr.setStyle(TableStyle([
        ("BACKGROUND",   (0,0),(-1,-1), col),
        ("LEFTPADDING",  (0,0),(-1,-1), 10),
        ("RIGHTPADDING", (0,0),(-1,-1), 10),
        ("TOPPADDING",   (0,0),(-1,-1), 8),
        ("BOTTOMPADDING",(0,0),(-1,-1), 8),
        ("ROUNDEDCORNERS", [4]),
    ]))
    story.append(KeepTogether([hdr]))
    story.append(Spacer(1, 0.2*cm))

    # Day rows
    day_data = [[
        Paragraph("Day(s)", S_TH),
        Paragraph("Topic", S_TH),
        Paragraph("Key Concepts", S_TH),
        Paragraph("Resource", S_TH),
    ]]
    for day, topic, concepts, resource in phase["days"]:
        day_data.append([
            Paragraph(day, S_TD_BOLD),
            Paragraph(f"<b>{topic}</b>", S_TD),
            Paragraph(concepts, S_TD),
            Paragraph(resource, style("res_td",
                fontName="Helvetica-Oblique", fontSize=7,
                textColor=C_TEAL, leading=10)),
        ])

    day_t = Table(day_data, colWidths=[W*0.11, W*0.22, W*0.40, W*0.27])
    day_t.setStyle(TableStyle([
        ("BACKGROUND",    (0,0), (-1,0),  col),
        ("ROWBACKGROUNDS",(0,1), (-1,-1), [C_WHITE, lite]),
        ("GRID",          (0,0), (-1,-1), 0.3, colors.HexColor("#CFD8DC")),
        ("LEFTPADDING",   (0,0), (-1,-1), 5),
        ("RIGHTPADDING",  (0,0), (-1,-1), 5),
        ("TOPPADDING",    (0,0), (-1,-1), 4),
        ("BOTTOMPADDING", (0,0), (-1,-1), 4),
        ("VALIGN",        (0,0), (-1,-1), "TOP"),
    ]))
    story.append(day_t)
    story.append(Spacer(1, 0.2*cm))

    # Tip box
    tip_t = Table([[Paragraph(f"💡  {phase['tip']}", style("tip_s",
        fontName="Helvetica-Oblique", fontSize=8,
        textColor=col, leading=11))]], colWidths=[W])
    tip_t.setStyle(TableStyle([
        ("BACKGROUND",   (0,0),(-1,-1), lite),
        ("LEFTPADDING",  (0,0),(-1,-1), 10),
        ("RIGHTPADDING", (0,0),(-1,-1), 10),
        ("TOPPADDING",   (0,0),(-1,-1), 6),
        ("BOTTOMPADDING",(0,0),(-1,-1), 6),
        ("ROUNDEDCORNERS", [4]),
        ("BOX", (0,0),(-1,-1), 0.5, col),
    ]))
    story.append(tip_t)
    story.append(Spacer(1, 0.5*cm))

# ── GOLDEN RULES ────────────────────────────────────────────────────────────
story.append(PageBreak())
story.append(header_table("⭐  6 GOLDEN RULES FOR THIS PLAN", bg=C_YELLOW,
    sub="Follow these and the plan will work. Skip them and it won't."))
story.append(Spacer(1, 0.25*cm))

for i, (title, body) in enumerate(tips):
    num_t = Table([[
        Paragraph(str(i+1), style("num", fontName="Helvetica-Bold",
            fontSize=14, textColor=C_WHITE, alignment=TA_CENTER, leading=18)),
        [Paragraph(title, style("tip_title", fontName="Helvetica-Bold",
             fontSize=10, textColor=C_DARK_BLUE, leading=13)),
         Paragraph(body, style("tip_body", fontName="Helvetica",
             fontSize=8.5, textColor=C_GREY_DARK, leading=12))],
    ]], colWidths=[W*0.07, W*0.93])
    num_t.setStyle(TableStyle([
        ("BACKGROUND",   (0,0),(0,0), C_YELLOW),
        ("BACKGROUND",   (1,0),(1,0), C_YELLOW_LIGHT),
        ("LEFTPADDING",  (0,0),(-1,-1), 8),
        ("RIGHTPADDING", (0,0),(-1,-1), 8),
        ("TOPPADDING",   (0,0),(-1,-1), 8),
        ("BOTTOMPADDING",(0,0),(-1,-1), 8),
        ("VALIGN",       (0,0),(-1,-1), "MIDDLE"),
        ("BOX",          (0,0),(-1,-1), 0.5, C_YELLOW),
        ("ROUNDEDCORNERS", [4]),
    ]))
    story.append(num_t)
    story.append(Spacer(1, 0.15*cm))

# ── PROGRESS TRACKER ────────────────────────────────────────────────────────
story.append(Spacer(1, 0.3*cm))
story.append(header_table("✅  WEEKLY PROGRESS TRACKER  (tick off each week)", bg=C_PURPLE))
story.append(Spacer(1, 0.2*cm))

tracker_rows = [
    ["Week 0",   "Histology Foundation",        "[ ]"],
    ["Week 1",   "Cardiovascular – Part 1",     "[ ]"],
    ["Week 2",   "Cardiovascular – Part 2",     "[ ]"],
    ["Week 3",   "Respiratory + Virology Intro","[ ]"],
    ["Week 3.5", "Immunology Block",            "[ ]"],
    ["Week 4",   "GI Tract",                    "[ ]"],
    ["Week 5",   "Liver + Hepatitis Viruses",   "[ ]"],
    ["Week 6",   "Renal System",                "[ ]"],
    ["Week 7",   "Endocrinology",               "[ ]"],
    ["Week 7.5", "Genetics Block",              "[ ]"],
    ["Week 8",   "Neurology + Neuro Viruses",   "[ ]"],
    ["Week 9",   "Haematology + HIV/AIDS",      "[ ]"],
    ["Week 10",  "Revision + Integration",      "[ ]"],
]
tr_header = [Paragraph(h, S_TH) for h in ["Phase", "Topic", "Done?"]]
tr_data   = [tr_header]
for row in tracker_rows:
    tr_data.append([Paragraph(c, S_TD) for c in row])

tr_t = Table(tr_data, colWidths=[W*0.15, W*0.70, W*0.15])
tr_t.setStyle(TableStyle([
    ("BACKGROUND",    (0,0), (-1,0),  C_PURPLE),
    ("ROWBACKGROUNDS",(0,1), (-1,-1), [C_WHITE, C_PURPLE_LIGHT]),
    ("GRID",          (0,0), (-1,-1), 0.3, colors.HexColor("#CFD8DC")),
    ("LEFTPADDING",   (0,0), (-1,-1), 6),
    ("RIGHTPADDING",  (0,0), (-1,-1), 6),
    ("TOPPADDING",    (0,0), (-1,-1), 5),
    ("BOTTOMPADDING", (0,0), (-1,-1), 5),
    ("ALIGN",         (2,0), (2,-1),  "CENTER"),
    ("FONTNAME",      (2,1), (2,-1),  "Helvetica-Bold"),
    ("FONTSIZE",      (2,1), (2,-1),  11),
]))
story.append(tr_t)

story.append(Spacer(1, 0.5*cm))
story.append(Paragraph(
    "Generated by Orris AI  •  Good luck with Year 3!  •  You've got this. 💪",
    S_FOOTER
))

# ── Build PDF ────────────────────────────────────────────────────────────────
doc.build(story)
print("PDF created successfully:", OUTPUT)