from reportlab.lib.pagesizes import A4
from reportlab.lib import colors
from reportlab.lib.units import cm
from reportlab.platypus import SimpleDocTemplate, Table, TableStyle, Paragraph, Spacer
from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
from reportlab.lib.enums import TA_CENTER, TA_LEFT
from pypdf import PdfWriter, PdfReader
import os

OUTPUT_DIR = "/tmp/workspace/study-sheets"

PALETTES = {
    "anatomy":    {"header": colors.HexColor("#1A3C5E"), "sub": colors.HexColor("#2E6DA4"), "light": colors.HexColor("#D6E8F7"), "accent": colors.HexColor("#F0F7FF")},
    "physiology": {"header": colors.HexColor("#145A32"), "sub": colors.HexColor("#1E8449"), "light": colors.HexColor("#D5F5E3"), "accent": colors.HexColor("#EAFAF1")},
    "materia":    {"header": colors.HexColor("#6E2C00"), "sub": colors.HexColor("#BA4A00"), "light": colors.HexColor("#FDEBD0"), "accent": colors.HexColor("#FEF9E7")},
    "organon":    {"header": colors.HexColor("#4A235A"), "sub": colors.HexColor("#7D3C98"), "light": colors.HexColor("#E8DAEF"), "accent": colors.HexColor("#F9F0FF")},
    "pharmacy":   {"header": colors.HexColor("#154360"), "sub": colors.HexColor("#1A5276"), "light": colors.HexColor("#D6EAF8"), "accent": colors.HexColor("#EBF5FB")},
}

def build_styles(pal):
    styles = {}
    styles["title"] = ParagraphStyle("title", fontSize=18, fontName="Helvetica-Bold",
        textColor=colors.white, alignment=TA_CENTER, spaceAfter=2, leading=22)
    styles["subtitle"] = ParagraphStyle("subtitle", fontSize=8.5, fontName="Helvetica",
        textColor=colors.HexColor("#CCDDEE"), alignment=TA_CENTER, spaceAfter=4)
    styles["section"] = ParagraphStyle("section", fontSize=9.5, fontName="Helvetica-Bold",
        textColor=colors.white, spaceAfter=2, spaceBefore=2)
    styles["body"] = ParagraphStyle("body", fontSize=7.8, fontName="Helvetica",
        textColor=colors.HexColor("#1A1A1A"), spaceAfter=1, leading=10.5)
    styles["bold"] = ParagraphStyle("bold", fontSize=7.8, fontName="Helvetica-Bold",
        textColor=colors.HexColor("#1A1A1A"), spaceAfter=1, leading=10.5)
    styles["tip"] = ParagraphStyle("tip", fontSize=7.5, fontName="Helvetica-Oblique",
        textColor=colors.HexColor("#444444"), spaceAfter=1, leading=10)
    styles["footer"] = ParagraphStyle("footer", fontSize=7, fontName="Helvetica",
        textColor=colors.HexColor("#666666"), alignment=TA_CENTER)
    return styles

def header_table(title, subtitle, pal):
    s = build_styles(pal)
    data = [[Paragraph(title, s["title"])], [Paragraph(subtitle, s["subtitle"])]]
    t = Table(data, colWidths=[19*cm])
    t.setStyle(TableStyle([
        ("BACKGROUND",    (0,0), (-1,-1), pal["header"]),
        ("TOPPADDING",    (0,0), (-1,-1), 10),
        ("BOTTOMPADDING", (0,0), (-1,-1), 6),
        ("LEFTPADDING",   (0,0), (-1,-1), 12),
        ("RIGHTPADDING",  (0,0), (-1,-1), 12),
    ]))
    return t

def section_box(heading, rows, pal, col_widths=None):
    s = build_styles(pal)
    if col_widths is None:
        col_widths = [19*cm]
    n_cols = len(col_widths)

    table_data = [[Paragraph(f"\u25b8  {heading}", s["section"])] + [""] * (n_cols - 1)]
    for r in rows:
        if isinstance(r, str):
            table_data.append([Paragraph(r, s["body"])] + [""] * (n_cols - 1))
        else:
            cells = []
            for c in r:
                cells.append(Paragraph(str(c), s["body"]))
            while len(cells) < n_cols:
                cells.append(Paragraph("", s["body"]))
            table_data.append(cells)

    ts = TableStyle([
        ("BACKGROUND",    (0,0), (-1,0), pal["sub"]),
        ("SPAN",          (0,0), (-1,0)),
        ("ROWBACKGROUNDS",(0,1), (-1,-1), [pal["accent"], pal["light"]]),
        ("GRID",          (0,1), (-1,-1), 0.3, colors.HexColor("#BBBBBB")),
        ("TOPPADDING",    (0,0), (-1,-1), 3),
        ("BOTTOMPADDING", (0,0), (-1,-1), 3),
        ("LEFTPADDING",   (0,0), (-1,-1), 6),
        ("RIGHTPADDING",  (0,0), (-1,-1), 6),
        ("LINEBELOW",     (0,0), (-1,0), 1, pal["header"]),
    ])
    t = Table(table_data, colWidths=col_widths)
    t.setStyle(ts)
    return t

def timetable_box(pal):
    s = build_styles(pal)
    rows = [
        [Paragraph("<b>Time Slot</b>", s["bold"]), Paragraph("<b>Study Focus</b>", s["bold"]), Paragraph("<b>Duration</b>", s["bold"])],
        [Paragraph("5:30-6:30 AM",  s["bold"]), Paragraph("Tough subject / New concepts", s["body"]),  Paragraph("1 hr", s["body"])],
        [Paragraph("7:15-8:15 AM",  s["bold"]), Paragraph("Revision of yesterday's notes", s["body"]), Paragraph("1 hr", s["body"])],
        [Paragraph("6:00-7:00 PM",  s["bold"]), Paragraph("Self-study / Assignments", s["body"]),       Paragraph("1 hr", s["body"])],
        [Paragraph("7:30-10:00 PM", s["bold"]), Paragraph("Main study (BHMS subjects)", s["body"]),     Paragraph("2.5 hrs", s["body"])],
    ]
    t = Table(rows, colWidths=[4.5*cm, 11*cm, 3.5*cm])
    t.setStyle(TableStyle([
        ("BACKGROUND",    (0,0), (-1,0), pal["sub"]),
        ("TEXTCOLOR",     (0,0), (-1,0), colors.white),
        ("ROWBACKGROUNDS",(0,1), (-1,-1), [pal["accent"], pal["light"]]),
        ("GRID",          (0,0), (-1,-1), 0.3, colors.HexColor("#BBBBBB")),
        ("TOPPADDING",    (0,0), (-1,-1), 3),
        ("BOTTOMPADDING", (0,0), (-1,-1), 3),
        ("LEFTPADDING",   (0,0), (-1,-1), 6),
    ]))
    return t

# ──────────────────────────────────────────────────────────────
# SUBJECT DATA
# ──────────────────────────────────────────────────────────────

SUBJECTS = {}

SUBJECTS["anatomy"] = {
    "title": "ANATOMY",
    "subtitle": "BHMS Periodic Assessment  |  High-Yield Quick Reference Sheet  |  Jul 27-31, 2026",
    "pal": PALETTES["anatomy"],
    "sections": [
        ("KEY SYSTEMS TO REVISE", [
            "Musculoskeletal: Bones, joints, muscles, nerve & blood supply of limbs",
            "Neuroanatomy: Brain lobes, spinal cord tracts, cranial nerves I-XII",
            "Cardiovascular: Heart chambers, valves, coronary arteries, conducting system",
            "Respiratory: Lungs, bronchial tree, pleura, diaphragm (nerve supply C3,4,5 - phrenic)",
            "Abdomen: Layers of GI wall, liver segments, kidney position (T12-L3), ureter course",
            "Head & Neck: Triangles of neck, parotid gland, facial nerve branches (TEFMC mnemonic)",
        ], None),
        ("CRANIAL NERVES - QUICK REFERENCE", [
            ["CN", "Name", "Type", "Function"],
            ["I",   "Olfactory",   "Sensory",   "Smell"],
            ["II",  "Optic",       "Sensory",   "Vision"],
            ["III", "Oculomotor",  "Motor",     "Eye movement (SR, IR, MR, IO), pupil constriction"],
            ["V",   "Trigeminal",  "Both",      "Facial sensation (V1/V2/V3), mastication (V3)"],
            ["VII", "Facial",      "Both",      "Facial expression, taste anterior 2/3 tongue"],
            ["VIII","Vestibulocochlear","Sensory","Hearing & balance"],
            ["X",   "Vagus",       "Both",      "Parasympathetic thorax & abdomen, phonation"],
            ["XII", "Hypoglossal", "Motor",     "Tongue movement (deviation to side of lesion)"],
        ], [1.2*cm, 4*cm, 2.8*cm, 11*cm]),
        ("VERTEBRAL LEVELS - HIGH YIELD", [
            ["Structure",              "Level",    "Clinical note"],
            ["Sternal angle (Louis)",  "T4/T5",    "Carina, aortic arch, SVC/IVC junction"],
            ["Transpyloric plane",     "L1",       "Pylorus, hilum of kidneys, fundus of GB"],
            ["Umbilicus",              "L3/L4",    "Bifurcation of abdominal aorta ~ L4"],
            ["Iliac crest",            "L4",       "Surface landmark for lumbar puncture (L3-L4)"],
        ], [5.5*cm, 3*cm, 10.5*cm]),
        ("IMPORTANT NERVE INJURIES", [
            "Radial nerve (Saturday night palsy): Wrist drop - cannot extend wrist",
            "Ulnar nerve (medial epicondyle): Claw hand, loss of little & ring finger sensation",
            "Median nerve (carpal tunnel): Ape hand, loss of thumb opposition, LOAF muscles",
            "Upper brachial plexus (C5-C6): Waiter's tip / Erb's palsy - arm hangs internally rotated",
            "Lower brachial plexus (C8-T1): Klumpke's palsy - claw hand + Horner syndrome",
        ], None),
        ("EXAM TIPS", [
            "* Draw diagrams: brachial plexus, heart valves, renal segments - examiners love visuals",
            "* Know muscle: Origin > Insertion > Action > Nerve supply (format for every answer)",
            "* Vertebral levels are MCQ gold - memorise the table above perfectly",
        ], None),
    ]
}

SUBJECTS["physiology"] = {
    "title": "PHYSIOLOGY",
    "subtitle": "BHMS Periodic Assessment  |  High-Yield Quick Reference Sheet  |  Jul 27-31, 2026",
    "pal": PALETTES["physiology"],
    "sections": [
        ("HAEMATOLOGY - NORMAL VALUES", [
            ["Parameter",          "Male",              "Female"],
            ["RBC count",          "4.5-5.5 x10^6/uL",  "4.0-5.0 x10^6/uL"],
            ["Haemoglobin",        "13-17 g/dL",         "12-15 g/dL"],
            ["Haematocrit (PCV)",  "40-54%",             "36-47%"],
            ["WBC (Leucocytes)",   "4000-11000/uL",      "Same"],
            ["Platelets",          "1.5-4.0 lakh/uL",   "Same"],
            ["ESR (Westergren)",   "0-15 mm/hr",         "0-20 mm/hr"],
            ["Blood pH",           "7.35-7.45",          "Same"],
        ], [5.5*cm, 6.75*cm, 6.75*cm]),
        ("CARDIOVASCULAR SYSTEM", [
            "Cardiac Output = HR x Stroke Volume  (Normal: ~5 L/min at rest)",
            "Normal HR: 60-100 bpm  |  Normal BP: 120/80 mmHg  |  PR interval: 0.12-0.20 sec",
            "Conducting system: SA Node (pacemaker 70/min) -> AV Node -> Bundle of His -> Purkinje",
            "Frank-Starling Law: Increased preload -> increased stretch -> increased stroke volume",
            "ECG: P wave = atrial depolarisation | QRS = ventricular depolarisation | T = repolarisation",
            "Korotkoff sounds: Phase I = systolic BP | Phase V (disappearance) = diastolic BP",
        ], None),
        ("RESPIRATORY VOLUMES & CAPACITIES", [
            ["Volume / Capacity",  "Value",    "What it means"],
            ["Tidal Volume (TV)",  "500 mL",   "Air in normal quiet breath"],
            ["IRV",                "3000 mL",  "Max extra air inspired after normal inspiration"],
            ["ERV",                "1100 mL",  "Max extra air expired after normal expiration"],
            ["Residual Volume",    "1200 mL",  "Air remaining after maximum expiration"],
            ["Vital Capacity",     "4600 mL",  "TV + IRV + ERV (max breath in + out)"],
            ["Total Lung Cap",     "5800 mL",  "Vital Capacity + Residual Volume"],
            ["FRC",                "2300 mL",  "ERV + RV (air at end of normal expiration)"],
        ], [5*cm, 3.5*cm, 10.5*cm]),
        ("NERVOUS SYSTEM HIGHLIGHTS", [
            "Resting Membrane Potential: -70 mV (neuron) | Action potential threshold: -55 mV",
            "Key neurotransmitters: ACh (NMJ, parasympathetic) | Dopamine (reward, movement)",
            "Noradrenaline (sympathetic) | Serotonin (mood, sleep) | GABA (inhibitory)",
            "Sympathetic (fight/flight): T1-L2 | Parasympathetic (rest/digest): Craniosacral",
        ], None),
        ("ENDOCRINE - HIGH YIELD HORMONES", [
            ["Hormone",   "Gland",         "Main Action"],
            ["Insulin",   "Pancreas beta", "Lowers blood glucose; promotes glycogen storage"],
            ["Glucagon",  "Pancreas alpha","Raises blood glucose; promotes glycogenolysis"],
            ["Cortisol",  "Adrenal cortex","Stress response, anti-inflammatory, raises glucose"],
            ["ADH",       "Post. pituitary","Water reabsorption in collecting duct"],
            ["Aldosterone","Adrenal cortex","Na+ retention, K+ excretion in renal tubule"],
            ["T3/T4",     "Thyroid",       "Increase BMR, growth, heat production"],
        ], [3*cm, 4.5*cm, 11.5*cm]),
        ("EXAM TIPS", [
            "* Draw flowcharts for: cardiac cycle, nerve impulse, hormonal feedback loops",
            "* Normal lab values are MCQ favourites - memorise all values in the tables above",
            "* Know differences: Sympathetic vs Parasympathetic effects on heart/lung/GI",
        ], None),
    ]
}

SUBJECTS["materia"] = {
    "title": "MATERIA MEDICA",
    "subtitle": "BHMS Periodic Assessment  |  High-Yield Remedy Keynotes  |  Jul 27-31, 2026",
    "pal": PALETTES["materia"],
    "sections": [
        ("POLYCHREST REMEDIES - KEYNOTES", [
            ["Remedy",         "Constitution / Type",           "Key Symptoms & Modalities"],
            ["Sulphur",        "Dirty, lean/fat, lazy",          "Burning everywhere; worse heat, 11 AM hunger, bathing; better cold"],
            ["Calc Carb",      "Fat, fair, flabby child/adult",  "Cold sweaty head; sour sweat/stool; desire eggs; fear of disease"],
            ["Lycopodium",     "Intellectual coward",            "Right-sided; 4-8 PM aggravation; flatulence; desire sweets/hot drinks"],
            ["Pulsatilla",     "Mild, yielding, weeping",        "Thirstless; changeable symptoms; better open air, worse heat & rich food"],
            ["Natrum Mur",     "Reserved, suppressed grief",     "Desire salt; worse at sea, worse consolation; headache like hammers"],
            ["Sepia",          "Indifferent, worn out mother",   "Bearing-down sensation; worse cold, standing; better vigorous exercise"],
            ["Arsenicum Alb",  "Anxious, restless, fastidious",  "Burning relieved by heat; midnight (12-2 AM) aggravation; fear of death"],
            ["Nux Vomica",     "Irritable, sedentary, type-A",   "Oversensitive to everything; worse morning, cold, stimulants; craves spicy"],
            ["Phosphorus",     "Tall, thin, open, artistic",     "Haemorrhagic tendency; fear of dark/alone; desire cold drinks (vomited warm)"],
            ["Belladonna",     "Plethoric, red, hot",            "Sudden violent onset; red-hot-dry-throbbing; photophobia; high fever"],
            ["Ignatia",        "Sensitive, idealistic",          "Grief & disappointed love; contradictory symptoms; sighing; hysteria"],
        ], [3*cm, 4.5*cm, 11.5*cm]),
        ("REMEDY COMPARISONS (Differentiation)", [
            ["Theme",          "Remedy 1",        "Remedy 2",       "Key Difference"],
            ["Weeping",        "Pulsatilla",       "Natrum Mur",    "Puls: better consolation | Nat-m: WORSE consolation"],
            ["Grief",          "Ignatia",          "Natrum Mur",    "Ign: acute/hysterical | Nat-m: chronic/suppressed"],
            ["Burning",        "Arsenicum",        "Sulphur",       "Ars: burning RELIEVED by heat | Sul: burning WORSE heat"],
            ["Right-sided",    "Lycopodium",       "Belladonna",    "Lyc: chronic/slow | Bell: acute/sudden/violent"],
        ], [3*cm, 3.5*cm, 3.5*cm, 9*cm]),
        ("MIASM CLASSIFICATION", [
            ["Miasm",     "Founder remedy",  "Character",         "Key remedies"],
            ["Psora",     "Psorinum",        "Functional, deficiency, itch", "Sulphur, Calc Carb, Lycopodium, Pulsatilla"],
            ["Sycosis",   "Medorrhinum",     "Excess, overgrowth, secretion", "Nux Vomica, Thuja, Natrum Sulph"],
            ["Syphilis",  "Syphilinum",      "Destructive, ulceration",       "Arsenicum, Mercurius, Nitric Acid"],
            ["Tubercular","Tuberculinum",    "Wandering, changing, weakly",   "Phosphorus, Calc Phos, Natrum Mur"],
        ], [2.5*cm, 3.5*cm, 5*cm, 8*cm]),
        ("EXAM TIPS - MATERIA MEDICA", [
            "* PQRS symptoms: Peculiar, Queer, Rare, Strange - these individualise and guide prescription",
            "* Know MODALITIES for each remedy: Aggravations (< worse) and Ameliorations (> better)",
            "* Remedy relationships: Complementary, Antidotes, Follows well - frequently asked",
            "* Miasm of each remedy is a common short-answer / MCQ question",
        ], None),
    ]
}

SUBJECTS["organon"] = {
    "title": "ORGANON OF MEDICINE & PSYCHOLOGY",
    "subtitle": "BHMS Periodic Assessment  |  Key Aphorisms & Concepts  |  Jul 27-31, 2026",
    "pal": PALETTES["organon"],
    "sections": [
        ("KEY APHORISMS - ORGANON 6th EDITION (Hahnemann)", [
            ["Aph", "Topic",                   "Core Statement"],
            ["1",   "Physician's Duty",         "Restore the sick to health - rapidly, gently, permanently"],
            ["2",   "Highest Ideal",            "Rapid, gentle, permanent restoration; removal of whole disease"],
            ["9",   "Vital Force",              "Spirit-like vital force governs body; when deranged = disease"],
            ["11",  "Nature of Disease",        "Disease = dynamic derangement of vital force expressed as symptoms"],
            ["22",  "Law of Cure",              "Medicines cure only by their power to produce similar symptoms"],
            ["26",  "Law of Similars",          "Similia similibus curentur - a stronger similar disease extinguishes weaker"],
            ["27",  "Curative Power",           "Only the law of similars can effect a true and lasting cure"],
            ["63-65","Primary & Secondary",     "Drug produces primary action; vital force responds with secondary (opposite)"],
            ["96-99","Drug Provings",           "Symptoms on healthy provers = true medicinal picture"],
            ["145", "Totality of Symptoms",     "Sum total of all symptoms = true picture of the disease"],
            ["153", "Characteristic Symptoms",  "Peculiar, uncommon, singular symptoms most important for remedy selection"],
            ["246", "Repetition of Dose",       "Repeat dose when improvement stops; do not repeat while improving"],
            ["269", "Potentisation",            "Mechanical process (trituration/succussion) releases latent medicinal power"],
            ["285", "Minimum Dose",             "Smallest quantity capable of producing gentle cure - avoid unnecessary aggravation"],
        ], [1.5*cm, 4.5*cm, 13*cm]),
        ("HERING'S LAW OF CURE (Direction of Cure)", [
            "1. From ABOVE downwards (head to foot)",
            "2. From WITHIN outwards (vital organs to skin/extremities)",
            "3. From MORE vital to LESS vital organ",
            "4. In REVERSE ORDER of appearance (last symptom to disappear first)",
        ], None),
        ("KENT'S HIERARCHY OF SYMPTOMS", [
            ["Rank", "Category",          "Examples"],
            ["1st",  "Mental Generals",   "Will, intellect, emotional state - MOST important"],
            ["2nd",  "Physical Generals", "Thermal reaction, sleep, appetite, thirst, perspiration"],
            ["3rd",  "Particulars",       "Symptoms of individual organs - LEAST weight"],
        ], [2*cm, 5*cm, 12*cm]),
        ("PSYCHOLOGY - KEY CONCEPTS", [
            ["Concept",             "Key Points"],
            ["Miasms & Mind",       "Psora: anxiety/fear  |  Sycosis: fixed ideas, suspicion  |  Syphilis: destructive impulses"],
            ["Will & Understanding","Hahnemann: disease first affects the Will, then Understanding"],
            ["Case Taking",         "Mental symptoms > Physical generals > Particulars  (Kent's hierarchy)"],
            ["Psychosomatic link",  "Mind affects body: grief, fear, anger can suppress physical symptoms"],
            ["Obstacles to Cure",   "Bad diet, grief, sedentary lifestyle, unhealthy environment (Aph 252-256)"],
        ], [4*cm, 15*cm]),
        ("EXAM TIPS - ORGANON", [
            "* Know aphorism NUMBER + concept - examiners ask 'Which aphorism states...'",
            "* Understand logical flow: Vital Force -> Derangement -> Symptoms -> Similimum -> Cure",
            "* Hering's Law - all 4 directions must be memorised, frequently asked in short answers",
            "* Second prescription: Never change remedy if patient is improving per Hering's directions",
        ], None),
    ]
}

SUBJECTS["pharmacy"] = {
    "title": "HOMOEOPATHIC PHARMACY",
    "subtitle": "BHMS Periodic Assessment  |  Preparation, Potency & Dispensing  |  Jul 27-31, 2026",
    "pal": PALETTES["pharmacy"],
    "sections": [
        ("SOURCES OF HOMOEOPATHIC DRUGS", [
            ["Source",      "Examples",                                              "Notes"],
            ["Vegetable",   "Aconite, Belladonna, Nux Vom, Pulsatilla, Chamomilla", "Largest source group"],
            ["Animal",      "Apis Mel (honeybee), Lachesis (snake venom), Sepia (cuttlefish)", "Potentised whole animal or secretion"],
            ["Mineral",     "Sulphur, Calc Carb, Arsenicum, Natrum Mur, Phosphorus","Elements & compounds"],
            ["Nosode",      "Psorinum (scabies), Medorrhinum (gonorrhoea), Tuberculinum","Made from disease products"],
            ["Sarcode",     "Thyroidinum, Insulinum, Fel Tauri (ox bile)",           "Made from healthy tissue/secretions"],
            ["Imponderabilia","X-ray, Sol (sunlight), Electricitas",                "Physical forces potentised"],
        ], [3*cm, 8.5*cm, 7.5*cm]),
        ("SCALES OF POTENCY", [
            ["Scale",                "Dilution Ratio", "Symbol",           "Common Potencies"],
            ["Decimal (X or D)",     "1:10",           "6X, 12X, 30X",     "Low potency, frequent dose"],
            ["Centesimal (C or CH)", "1:100",          "6C, 30C, 200C, 1M","Most widely used scale"],
            ["LM / Q Potency",       "1:50,000",       "LM1, LM2, LM3...", "Gentlest; daily repetition possible"],
            ["Millesimal (M/CM/MM)", "Higher",         "1M, 10M, 50M, CM", "High potency, rare repetition"],
        ], [4.5*cm, 3.5*cm, 4.5*cm, 6.5*cm]),
        ("PREPARATION METHODS", [
            "Mother Tincture (O): Fresh plant material + purified alcohol - base for all potencies",
            "Trituration: Insoluble substances (eg Calc Carb, Sulphur) ground with lactose up to 3C, then dissolve",
            "Potentisation = Dilution + Succussion (liquid) OR Dilution + Trituration (solid)",
            "Succussion: Vigorous shaking against a firm elastic surface - releases latent medicinal energy",
            "Hahnemannian method: New vial at each potency step (most accurate)",
            "Korsakovian method: Same vial emptied & refilled - rapid, used for high potencies",
        ], None),
        ("VEHICLES / DISPENSING FORMS", [
            ["Form",           "Vehicle",          "Use / Notes"],
            ["Globules/Pills", "Sucrose/Lactose",   "Sizes: 10, 20, 40 (smaller = more surface area for drug)"],
            ["Liquid drops",   "Alcohol + water",   "Dissolved potency; direct or diluted in distilled water"],
            ["Trituration tabs","Lactose",          "For low potencies (1X, 2X, 3X)"],
            ["Ointment",       "Vaseline/lanolin",  "External use only; Calendula, Hypericum ointments"],
        ], [3.5*cm, 4*cm, 11.5*cm]),
        ("STORAGE & ANTIDOTES", [
            "Store in: Amber/dark glass bottles | Away from heat, light, moisture, strong odours",
            "Camphor antidotes MOST homoeopathic remedies - patient must avoid camphor, menthol, eucalyptus",
            "Avoid: Coffee, tobacco, mint, strong perfumes during treatment (may antidote remedy)",
            "Shelf life: Potentised medicines have indefinite shelf life if stored correctly",
        ], None),
        ("EXAM TIPS - PHARMACY", [
            "* Dilution ratios MUST be memorised: X=1:10, C=1:100, LM=1:50000 - direct MCQ",
            "* Nosode vs Sarcode distinction - very commonly tested short question",
            "* Hahnemannian vs Korsakovian - know the difference clearly",
            "* Drug sources: Be able to classify ANY given remedy into its source category",
        ], None),
    ]
}

# ──────────────────────────────────────────────────────────────
# BUILD PDF PER SUBJECT
# ──────────────────────────────────────────────────────────────

def build_subject_pdf(key, data):
    filepath = os.path.join(OUTPUT_DIR, f"{key}_study_sheet.pdf")
    pal = data["pal"]
    s = build_styles(pal)

    doc = SimpleDocTemplate(
        filepath, pagesize=A4,
        leftMargin=1*cm, rightMargin=1*cm,
        topMargin=0.8*cm, bottomMargin=0.8*cm
    )
    story = []

    story.append(header_table(data["title"], data["subtitle"], pal))
    story.append(Spacer(1, 0.2*cm))
    story.append(timetable_box(pal))
    story.append(Spacer(1, 0.18*cm))

    for (heading, rows, col_widths) in data["sections"]:
        story.append(section_box(heading, rows, pal, col_widths))
        story.append(Spacer(1, 0.15*cm))

    footer_data = [[Paragraph(
        "Assessment: Jul 27-31, 2026  |  Study Slots: 5:30 AM | 7:15 AM | 6:00 PM | 7:30 PM  |  Total: 5.5 hrs/day  |  BHMS Year 1",
        s["footer"]
    )]]
    ft = Table(footer_data, colWidths=[19*cm])
    ft.setStyle(TableStyle([
        ("BACKGROUND",    (0,0), (-1,-1), colors.HexColor("#F0F0F0")),
        ("TOPPADDING",    (0,0), (-1,-1), 4),
        ("BOTTOMPADDING", (0,0), (-1,-1), 4),
        ("LINEABOVE",     (0,0), (-1,0), 0.5, colors.HexColor("#AAAAAA")),
    ]))
    story.append(ft)

    doc.build(story)
    return filepath

files = []
order = ["anatomy", "physiology", "materia", "organon", "pharmacy"]
for key in order:
    fp = build_subject_pdf(key, SUBJECTS[key])
    files.append(fp)
    print(f"Generated: {fp}")

# Merge all into one PDF
writer = PdfWriter()
for fp in files:
    reader = PdfReader(fp)
    for page in reader.pages:
        writer.add_page(page)

combined = os.path.join(OUTPUT_DIR, "BHMS_All_5_Study_Sheets.pdf")
with open(combined, "wb") as f:
    writer.write(f)
print(f"\nCombined PDF ready: {combined}")