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

# ── Colour palette ──────────────────────────────────────────────────────────
ORANGE      = colors.HexColor("#C8622A")   # terracotta / western
TEAL        = colors.HexColor("#2A7A6A")   # ayurvedic green
LIGHT_ORG   = colors.HexColor("#FDF0E8")
LIGHT_TEAL  = colors.HexColor("#E8F5F2")
MID_GRAY    = colors.HexColor("#555555")
DARK        = colors.HexColor("#1A1A1A")
HEADER_BG   = colors.HexColor("#2B2B2B")
WHITE       = colors.white

OUTPUT = "/tmp/workspace/digital-puberty/Western_vs_Ayurvedic_Management.pdf"

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

styles = getSampleStyleSheet()

# Custom styles
def S(name, **kw):
    return ParagraphStyle(name, parent=styles["Normal"], **kw)

title_style = S("Title2", fontSize=16, textColor=WHITE,
                fontName="Helvetica-Bold", alignment=TA_CENTER, leading=20)
sub_style   = S("Sub",    fontSize=8.5, textColor=colors.HexColor("#CCCCCC"),
                fontName="Helvetica", alignment=TA_CENTER, leading=12)
sec_style   = S("Sec",    fontSize=10, textColor=WHITE,
                fontName="Helvetica-Bold", alignment=TA_CENTER, leading=14)
body_w      = S("BW",     fontSize=8, textColor=DARK, fontName="Helvetica",
                leading=11, leftIndent=4)
body_a      = S("BA",     fontSize=8, textColor=DARK, fontName="Helvetica",
                leading=11, leftIndent=4)
bullet_w    = S("BullW",  fontSize=8, textColor=DARK, fontName="Helvetica",
                leading=11, leftIndent=6, bulletIndent=2)
bullet_a    = S("BullA",  fontSize=8, textColor=DARK, fontName="Helvetica",
                leading=11, leftIndent=6, bulletIndent=2)
col_hdr_w   = S("CHW",    fontSize=9, textColor=WHITE, fontName="Helvetica-Bold",
                alignment=TA_CENTER, leading=12)
col_hdr_a   = S("CHA",    fontSize=9, textColor=WHITE, fontName="Helvetica-Bold",
                alignment=TA_CENTER, leading=12)
note_style  = S("Note",   fontSize=7, textColor=MID_GRAY, fontName="Helvetica-Oblique",
                alignment=TA_CENTER, leading=10)

# ── helper: bullet para ──────────────────────────────────────────────────────
def bw(txt): return Paragraph(f"• {txt}", bullet_w)
def ba(txt): return Paragraph(f"• {txt}", bullet_a)

# ── helper: two-column comparison table ─────────────────────────────────────
W = 8.5*cm  # each column width

def make_compare_table(rows):
    """
    rows: list of (western_content, ayurvedic_content)
          content can be a Paragraph or list of Paragraphs
    """
    data = [[Paragraph("WESTERN / MODERN", col_hdr_w),
             Paragraph("AYURVEDIC", col_hdr_a)]]
    for w, a in rows:
        lw = w if isinstance(w, list) else [w]
        la = a if isinstance(a, list) else [a]
        data.append([lw, la])

    ts = TableStyle([
        # header row
        ("BACKGROUND",  (0,0), (0,0), ORANGE),
        ("BACKGROUND",  (1,0), (1,0), TEAL),
        ("TOPPADDING",  (0,0), (-1,0), 6),
        ("BOTTOMPADDING",(0,0),(-1,0), 6),
        # data rows alternating
        *[("BACKGROUND", (0,i), (0,i), LIGHT_ORG) for i in range(1, len(data), 2)],
        *[("BACKGROUND", (1,i), (1,i), LIGHT_TEAL) for i in range(1, len(data), 2)],
        ("GRID",        (0,0), (-1,-1), 0.3, colors.HexColor("#CCCCCC")),
        ("VALIGN",      (0,0), (-1,-1), "TOP"),
        ("TOPPADDING",  (0,1), (-1,-1), 5),
        ("BOTTOMPADDING",(0,1),(-1,-1), 5),
        ("LEFTPADDING", (0,0), (-1,-1), 6),
        ("RIGHTPADDING",(0,0), (-1,-1), 6),
        ("LINEBELOW",   (0,0), (-1,0), 1, WHITE),
    ])
    return Table(data, colWidths=[W, W], style=ts, hAlign="CENTER")

# ── helper: section header banner ────────────────────────────────────────────
def section_banner(title, subtitle=""):
    data = [[Paragraph(title, sec_style)]]
    if subtitle:
        data.append([Paragraph(subtitle, sub_style)])
    ts = TableStyle([
        ("BACKGROUND",   (0,0), (-1,-1), HEADER_BG),
        ("TOPPADDING",   (0,0), (-1,-1), 7),
        ("BOTTOMPADDING",(0,0), (-1,-1), 7),
        ("LEFTPADDING",  (0,0), (-1,-1), 8),
        ("RIGHTPADDING", (0,0), (-1,-1), 8),
        ("ROUNDEDCORNERS", [4]),
    ])
    return Table(data, colWidths=[17*cm], style=ts, hAlign="CENTER")

# ════════════════════════════════════════════════════════════════════════════
# BUILD STORY
# ════════════════════════════════════════════════════════════════════════════
story = []

# ── COVER HEADER ─────────────────────────────────────────────────────────────
cover_data = [
    [Paragraph("QUICK REFERENCE GUIDE", title_style)],
    [Paragraph("Western vs Ayurvedic Management Protocols", sub_style)],
    [Paragraph("Role of Digital Addiction, Sleep Disturbance & Physical Activity in Puberty",
               S("Cov2", fontSize=9, textColor=colors.HexColor("#FFB87A"),
                 fontName="Helvetica-BoldOblique", alignment=TA_CENTER, leading=13))],
]
cover_ts = TableStyle([
    ("BACKGROUND",   (0,0), (-1,-1), HEADER_BG),
    ("TOPPADDING",   (0,0), (-1,-1), 10),
    ("BOTTOMPADDING",(0,0), (-1,-1), 10),
])
story.append(Table(cover_data, colWidths=[17*cm], style=cover_ts))
story.append(Spacer(1, 0.35*cm))

# ── LEGEND ───────────────────────────────────────────────────────────────────
legend_data = [[
    Paragraph("■  WESTERN / MODERN MEDICINE", S("LW", fontSize=8, textColor=ORANGE,
              fontName="Helvetica-Bold", leading=11)),
    Paragraph("■  AYURVEDIC MEDICINE", S("LA", fontSize=8, textColor=TEAL,
              fontName="Helvetica-Bold", leading=11)),
]]
story.append(Table(legend_data, colWidths=[8.5*cm, 8.5*cm],
                   style=TableStyle([("VALIGN",(0,0),(-1,-1),"MIDDLE")]),
                   hAlign="CENTER"))
story.append(Spacer(1, 0.25*cm))

# ════════════════════════════════════════════════════════════════════════════
# SECTION 1 — UNDERSTANDING THE CONDITION
# ════════════════════════════════════════════════════════════════════════════
story.append(section_banner("1.  UNDERSTANDING THE CONDITION IN PUBERTY"))
story.append(Spacer(1, 0.15*cm))

rows_s1 = [
    (
        [bw("Pubertal transition: hormonal (GnRH, LH, FSH, estrogen/testosterone)"),
         bw("Prefrontal cortex still maturing — impulse control immature"),
         bw("Heightened neuroplasticity = vulnerability to addictive behaviours")],
        [ba("Kapha → Pitta Avastha transition (Kishora Avastha)"),
         ba("Agni (metabolic fire) and Shukra/Artava Dhatu are forming"),
         ba("Ojas (vital essence) still maturing — easily depleted by overuse")]
    ),
]
story.append(make_compare_table(rows_s1))
story.append(Spacer(1, 0.3*cm))

# ════════════════════════════════════════════════════════════════════════════
# SECTION 2 — DIGITAL ADDICTION
# ════════════════════════════════════════════════════════════════════════════
story.append(section_banner("2.  DIGITAL ADDICTION — PATHOLOGY & MANAGEMENT"))
story.append(Spacer(1, 0.15*cm))

rows_s2 = [
    (
        [Paragraph("<b>Pathology</b>", body_w),
         bw("Dopamine reward loop — compulsive device use"),
         bw("Reduced gray matter in prefrontal/executive regions"),
         bw("Disrupted default mode & executive control networks"),
         bw("Criteria: preoccupation, withdrawal, tolerance, impairment")],
        [Paragraph("<b>Pathology</b>", body_a),
         ba("Asatmyendriyartha Samyoga — unwholesome sense-object contact"),
         ba("Prajnaparadha — intellectual transgression (knowing it's harmful)"),
         ba("Ojas Kshaya — vital essence depletion"),
         ba("Rajas/Tamas dominate; Satva (clarity) weakened")]
    ),
    (
        [Paragraph("<b>Management</b>", body_w),
         bw("CBT / Cognitive Behavioural Therapy"),
         bw("Screen time limits & parental controls"),
         bw("Motivational interviewing"),
         bw("Digital detox programmes"),
         bw("Mindfulness-based interventions")],
        [Paragraph("<b>Management</b>", body_a),
         ba("Satwavajaya Chikitsa (mind-based therapy)"),
         ba("Sadvritta Palana — ethical conduct code"),
         ba("Indriya Nigraha — deliberate sensory restraint"),
         ba("Rasayana: Ashwagandha, Brahmi, Guduchi to rebuild Ojas"),
         ba("Outdoor sports & team games (Krida)")]
    ),
]
story.append(make_compare_table(rows_s2))
story.append(Spacer(1, 0.3*cm))

# ════════════════════════════════════════════════════════════════════════════
# SECTION 3 — SLEEP DISTURBANCE
# ════════════════════════════════════════════════════════════════════════════
story.append(section_banner("3.  SLEEP DISTURBANCE — PATHOLOGY & MANAGEMENT"))
story.append(Spacer(1, 0.15*cm))

rows_s3 = [
    (
        [Paragraph("<b>Pathology</b>", body_w),
         bw("Blue light → melatonin suppression → circadian delay"),
         bw("Social media arousal → delayed sleep onset"),
         bw("Sleep deprivation → disrupted GH release, HPG axis dysregulation"),
         bw("Vicious cycle: poor sleep → poor impulse control → more screens")],
        [Paragraph("<b>Pathology</b>", body_a),
         ba("Vata-Pitta Prakopa → Anidra (insomnia)"),
         ba("Screen Tejas/Agni stimulus suppresses Kapha-Tamas Nidra"),
         ba("Rajas (mental agitation) prevents Pratyahara (sensory withdrawal)"),
         ba("Anidra → Dhatu Kshaya, Ojas depletion, growth impairment")]
    ),
    (
        [Paragraph("<b>Management</b>", body_w),
         bw("Sleep hygiene: consistent schedule, dark cool room"),
         bw("No screens ≥1 hr before bed"),
         bw("CBT-I (Cognitive Behavioural Therapy for Insomnia)"),
         bw("Melatonin supplementation (short-term, physician-guided)"),
         bw("Recommend 8-10 hrs/night (AAP guideline)")],
        [Paragraph("<b>Management</b>", body_a),
         ba("Ratri Charya (night routine): screen-free, warm milk + Ashwagandha"),
         ba("Brahma Muhurta wake (4:30-5:30 AM) — resets circadian rhythm"),
         ba("Shirodhara — medicated oil on forehead; profound Vata-calming"),
         ba("Shankhapushpi + Brahmi — herbal sleep promoters"),
         ba("Yoga Nidra (conscious deep relaxation / Pratyahara)")]
    ),
]
story.append(make_compare_table(rows_s3))
story.append(Spacer(1, 0.3*cm))

# ════════════════════════════════════════════════════════════════════════════
# SECTION 4 — PHYSICAL ACTIVITY
# ════════════════════════════════════════════════════════════════════════════
story.append(section_banner("4.  PHYSICAL ACTIVITY — RECOMMENDATIONS & PROTOCOLS"))
story.append(Spacer(1, 0.15*cm))

rows_s4 = [
    (
        [Paragraph("<b>Recommendations</b>", body_w),
         bw("WHO: 60 min moderate-vigorous activity per day"),
         bw("Strength/resistance training ≥3×/week"),
         bw("Limit recreational screen time <2 hrs/day"),
         bw("Exercise boosts BDNF, dopamine regulation, sleep quality")],
        [Paragraph("<b>Recommendations</b>", body_a),
         ba("Vyayama daily — minimum 30 min in morning (Kapha time: 6-10 AM)"),
         ba("Ardhashakti rule: exercise to HALF capacity only"),
         ba("Stop when forehead sweats & breath deepens"),
         ba("Vyayama: Agni Deepana, Dosha Kshaya, Sthairya (stability)")]
    ),
    (
        [Paragraph("<b>Preferred Activities</b>", body_w),
         bw("Aerobic: running, swimming, cycling, dance"),
         bw("Team sports: football, basketball, volleyball"),
         bw("School-mandated PE programmes"),
         bw("Structured after-school sports clubs")],
        [Paragraph("<b>Preferred Activities</b>", body_a),
         ba("Surya Namaskar (12 rounds) — all-dosha balancing"),
         ba("Team sports: badminton, volleyball, kabaddi (Pitta-channelling)"),
         ba("Abhyanga (oil massage) pre-exercise — Vata-pacifying"),
         ba("Avoid Atiyoga (over-exercise) — causes Vata aggravation")]
    ),
    (
        [Paragraph("<b>Inactivity Consequences</b>", body_w),
         bw("Obesity, insulin resistance, poor cardiovascular fitness"),
         bw("Reduced dopamine regulation → digital rewards more potent"),
         bw("Musculoskeletal: postural problems, spondylosis risk")],
        [Paragraph("<b>Inactivity Consequences</b>", body_a),
         ba("Kapha-Tamas Vriddhi → lethargy, obesity, Agni Mandya"),
         ba("Srotorodha (channel blockage) → metabolic and tissue disease"),
         ba("Reduced Bala (strength) and Dhriti (mental fortitude)")]
    ),
]
story.append(make_compare_table(rows_s4))
story.append(Spacer(1, 0.3*cm))

# ════════════════════════════════════════════════════════════════════════════
# SECTION 5 — INTEGRATED / COMBINED PROTOCOL
# ════════════════════════════════════════════════════════════════════════════
story.append(section_banner("5.  INTEGRATED DAILY PROTOCOL FOR ADOLESCENTS",
                            "Western recommendations mapped to Ayurvedic Dinacharya"))
story.append(Spacer(1, 0.15*cm))

time_data = [
    [Paragraph("TIME", S("TH", fontSize=8, textColor=WHITE,
               fontName="Helvetica-Bold", alignment=TA_CENTER)),
     Paragraph("WESTERN PROTOCOL", col_hdr_w),
     Paragraph("AYURVEDIC DINACHARYA", col_hdr_a)],
    ["4:30–6 AM",
     Paragraph("No alarm disruption; gradual wake", body_w),
     Paragraph("Brahma Muhurta — wake, self-reflection, Pranayama", body_a)],
    ["6–7 AM",
     Paragraph("30-60 min physical exercise; breakfast", body_w),
     Paragraph("Vyayama (half capacity) + Abhyanga + Snana (bathing)", body_a)],
    ["7 AM–2 PM",
     Paragraph("School; limit screens; natural light exposure", body_w),
     Paragraph("Pitta time — study, work, Sattvic Ahara (diet)", body_a)],
    ["2–6 PM",
     Paragraph("Outdoor play; homework; social interaction", body_w),
     Paragraph("Vata time — creative activity, team sports, Krida", body_a)],
    ["6–8 PM",
     Paragraph("Family time; light screen use (<1 hr)", body_w),
     Paragraph("Sandhya Vandanam / Dhyana (meditation), light meal", body_a)],
    ["8–9 PM",
     Paragraph("Wind-down: dim lights, no devices, reading", body_w),
     Paragraph("No screens; warm turmeric milk; Shankhapushpi", body_a)],
    ["By 10 PM",
     Paragraph("Sleep onset target (8-10 hrs for teens)", body_w),
     Paragraph("Nidra (Kapha-Pitta time: 10 PM–2 AM = deep repair sleep)", body_a)],
]

time_ts = TableStyle([
    ("BACKGROUND",   (0,0), (0,0), HEADER_BG),
    ("BACKGROUND",   (1,0), (1,0), ORANGE),
    ("BACKGROUND",   (2,0), (2,0), TEAL),
    ("TEXTCOLOR",    (0,0), (-1,0), WHITE),
    ("FONTNAME",     (0,0), (-1,0), "Helvetica-Bold"),
    ("FONTSIZE",     (0,0), (-1,0), 8),
    ("ALIGN",        (0,0), (-1,0), "CENTER"),
    *[("BACKGROUND", (0,i), (0,i), colors.HexColor("#EEEEEE")) for i in range(1,len(time_data))],
    *[("BACKGROUND", (1,i), (1,i), LIGHT_ORG) for i in range(1,len(time_data),2)],
    *[("BACKGROUND", (2,i), (2,i), LIGHT_TEAL) for i in range(1,len(time_data),2)],
    ("GRID",         (0,0), (-1,-1), 0.3, colors.HexColor("#CCCCCC")),
    ("VALIGN",       (0,0), (-1,-1), "MIDDLE"),
    ("FONTSIZE",     (0,1), (0,-1), 7.5),
    ("FONTNAME",     (0,1), (0,-1), "Helvetica-Bold"),
    ("ALIGN",        (0,1), (0,-1), "CENTER"),
    ("TOPPADDING",   (0,0), (-1,-1), 5),
    ("BOTTOMPADDING",(0,0), (-1,-1), 5),
    ("LEFTPADDING",  (0,0), (-1,-1), 5),
    ("RIGHTPADDING", (0,0), (-1,-1), 5),
])
story.append(Table(time_data, colWidths=[2.3*cm, 7.35*cm, 7.35*cm],
                   style=time_ts, hAlign="CENTER"))
story.append(Spacer(1, 0.3*cm))

# ════════════════════════════════════════════════════════════════════════════
# SECTION 6 — QUICK-GLANCE EQUIVALENCE TABLE
# ════════════════════════════════════════════════════════════════════════════
story.append(section_banner("6.  CONCEPT EQUIVALENCE — AT A GLANCE"))
story.append(Spacer(1, 0.15*cm))

eq_data = [
    [Paragraph("MODERN CONCEPT", col_hdr_w),
     Paragraph("AYURVEDIC EQUIVALENT", col_hdr_a)],
    [Paragraph("Digital addiction", body_w), Paragraph("Asatmyendriyartha Samyoga + Prajnaparadha", body_a)],
    [Paragraph("Insomnia / sleep disorder", body_w), Paragraph("Anidra — Vata-Pitta Prakopa", body_a)],
    [Paragraph("Physical inactivity", body_w), Paragraph("Vyayama Heena + Kapha-Tamas Vriddhi", body_a)],
    [Paragraph("Puberty", body_w), Paragraph("Kishora Avastha — Kapha → Pitta transition", body_a)],
    [Paragraph("Blue light / melatonin suppression", body_w), Paragraph("Tejas overuse suppressing Kapha-Tamas Nidra", body_a)],
    [Paragraph("Dopamine reward loop", body_w), Paragraph("Indriya Atiyoga depleting Ojas", body_a)],
    [Paragraph("Sleep hygiene", body_w), Paragraph("Ratri Charya", body_a)],
    [Paragraph("Mindfulness / CBT", body_w), Paragraph("Satwavajaya Chikitsa", body_a)],
    [Paragraph("Preventive health / wellness", body_w), Paragraph("Swasthavritta + Dinacharya + Sadvritta", body_a)],
    [Paragraph("Antioxidants / neuroprotection", body_w), Paragraph("Rasayana therapy (Ashwagandha, Brahmi, Amalaki)", body_a)],
    [Paragraph("Circadian rhythm", body_w), Paragraph("Kala-based Dosha cycles (Dinacharya)", body_a)],
]

eq_ts = TableStyle([
    ("BACKGROUND",   (0,0), (0,0), ORANGE),
    ("BACKGROUND",   (1,0), (1,0), TEAL),
    ("TEXTCOLOR",    (0,0), (-1,0), WHITE),
    *[("BACKGROUND", (0,i), (-1,i), colors.HexColor("#F7F7F7")) for i in range(1,len(eq_data),2)],
    ("GRID",         (0,0), (-1,-1), 0.3, colors.HexColor("#DDDDDD")),
    ("VALIGN",       (0,0), (-1,-1), "MIDDLE"),
    ("TOPPADDING",   (0,0), (-1,-1), 4),
    ("BOTTOMPADDING",(0,0), (-1,-1), 4),
    ("LEFTPADDING",  (0,0), (-1,-1), 6),
    ("RIGHTPADDING", (0,0), (-1,-1), 6),
    ("FONTNAME",     (0,0), (-1,0), "Helvetica-Bold"),
    ("FONTSIZE",     (0,0), (-1,0), 8.5),
    ("ALIGN",        (0,0), (-1,0), "CENTER"),
])
story.append(Table(eq_data, colWidths=[8.5*cm, 8.5*cm],
                   style=eq_ts, hAlign="CENTER"))
story.append(Spacer(1, 0.35*cm))

# ── FOOTER NOTE ──────────────────────────────────────────────────────────────
story.append(HRFlowable(width="100%", thickness=0.5, color=MID_GRAY))
story.append(Spacer(1, 0.15*cm))
story.append(Paragraph(
    "Sources: Charaka Samhita · Ashtanga Hridayam · Sushruta Samhita · Bhavaprakasha  |  "
    "WHO Physical Activity Guidelines 2022  |  AAP Sleep Guidelines  |  "
    "PLOS Mental Health, UCL 2024  |  BMC Public Health PMC7418409",
    note_style))
story.append(Spacer(1, 0.1*cm))
story.append(Paragraph(
    "For educational / quick-reference use. Clinical decisions should always be guided by a qualified professional.",
    note_style))

# ── BUILD ─────────────────────────────────────────────────────────────────────
doc.build(story)
print("PDF generated:", OUTPUT)