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

OUTPUT = "/tmp/workspace/skincare-pdf/Skincare_Routine.pdf"

# ── Colour palette ──────────────────────────────────────────────────────────
NAVY      = colors.HexColor("#1A2B4A")   # headings / header bg
SKY       = colors.HexColor("#E8F4FD")   # morning block bg
MOON      = colors.HexColor("#F0EBF8")   # night block bg
GOLD      = colors.HexColor("#F5A623")   # accent / step numbers
LIGHT_GREY= colors.HexColor("#F7F7F7")   # table row alt
MID_GREY  = colors.HexColor("#CCCCCC")   # rule / border
WHITE     = colors.white
GREEN     = colors.HexColor("#27AE60")
RED_SOFT  = colors.HexColor("#E74C3C")

doc = SimpleDocTemplate(
    OUTPUT,
    pagesize=A4,
    leftMargin=18*mm, rightMargin=18*mm,
    topMargin=14*mm, bottomMargin=14*mm,
    title="Daily Skincare Routine",
    author="Orris Skincare"
)

W, H = A4
content_w = W - 36*mm

styles = getSampleStyleSheet()

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

TITLE_S   = PS("Title2",    fontSize=22, fontName="Helvetica-Bold",
               textColor=WHITE,       alignment=TA_CENTER, spaceAfter=2)
SUBTITLE_S= PS("Subtitle2", fontSize=10, fontName="Helvetica",
               textColor=colors.HexColor("#BCD9F5"), alignment=TA_CENTER)
SECTION_S = PS("Section",   fontSize=13, fontName="Helvetica-Bold",
               textColor=WHITE, alignment=TA_CENTER, spaceAfter=0)
LABEL_S   = PS("Label",     fontSize=9,  fontName="Helvetica-Bold",
               textColor=NAVY, spaceAfter=1)
BODY_S    = PS("Body2",     fontSize=9,  fontName="Helvetica",
               textColor=colors.HexColor("#333333"), leading=13)
SMALL_S   = PS("Small",     fontSize=8,  fontName="Helvetica",
               textColor=colors.HexColor("#555555"), leading=11)
TIP_S     = PS("Tip",       fontSize=8,  fontName="Helvetica-Oblique",
               textColor=colors.HexColor("#555555"), leading=11)
STEP_S    = PS("Step",      fontSize=10, fontName="Helvetica-Bold",
               textColor=NAVY)
WARN_S    = PS("Warn",      fontSize=8,  fontName="Helvetica-Bold",
               textColor=RED_SOFT)
TIMELINE_S= PS("Timeline",  fontSize=8,  fontName="Helvetica",
               textColor=colors.HexColor("#333333"), leading=12)

story = []

# ── HEADER BANNER ───────────────────────────────────────────────────────────
header_data = [[
    Paragraph("MY DAILY SKINCARE ROUTINE", TITLE_S),
]]
header_sub  = [[
    Paragraph("Oily / Acne-Prone Skin  •  Age 21  •  Derma Co Products", SUBTITLE_S),
]]
banner = Table(
    [
        [Paragraph("MY DAILY SKINCARE ROUTINE", TITLE_S)],
        [Paragraph("Oily / Acne-Prone Skin  •  Age 21  •  Derma Co Products", SUBTITLE_S)],
    ],
    colWidths=[content_w]
)
banner.setStyle(TableStyle([
    ("BACKGROUND", (0,0), (-1,-1), NAVY),
    ("TOPPADDING",    (0,0), (-1,-1), 10),
    ("BOTTOMPADDING", (0,0), (-1,-1), 10),
    ("LEFTPADDING",   (0,0), (-1,-1), 8),
    ("RIGHTPADDING",  (0,0), (-1,-1), 8),
    ("ROUNDEDCORNERS",(0,0), (-1,-1), [6,6,6,6]),  # ignored but harmless
]))
story.append(banner)
story.append(Spacer(1, 5*mm))


# ── helper: section header ──────────────────────────────────────────────────
def section_header(text, bg):
    tbl = Table([[Paragraph(text, SECTION_S)]], colWidths=[content_w])
    tbl.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),
        ("RIGHTPADDING",  (0,0), (-1,-1), 8),
    ]))
    return tbl


# ── helper: step row ────────────────────────────────────────────────────────
def step_row(num, step_name, product, qty, notes, bg=WHITE):
    num_cell   = Paragraph(f'<font color="#F5A623"><b>{num}</b></font>', PS("N", fontSize=16, fontName="Helvetica-Bold", alignment=TA_CENTER))
    name_cell  = Paragraph(f"<b>{step_name}</b>", PS("SN", fontSize=10, fontName="Helvetica-Bold", textColor=NAVY))
    prod_cell  = Paragraph(product, PS("SP", fontSize=9,  fontName="Helvetica-Oblique", textColor=colors.HexColor("#1A6B3A")))
    qty_cell   = Paragraph(qty,     PS("SQ", fontSize=9,  fontName="Helvetica-Bold",   textColor=RED_SOFT))
    note_cell  = Paragraph(notes,   SMALL_S)

    inner = Table(
        [
            [name_cell, qty_cell],
            [prod_cell, ""],
            [note_cell, ""],
        ],
        colWidths=[content_w - 24*mm - 4*mm, 24*mm],
        rowHeights=None
    )
    inner.setStyle(TableStyle([
        ("SPAN",         (0,2), (1,2)),
        ("VALIGN",       (0,0), (-1,-1), "TOP"),
        ("LEFTPADDING",  (0,0), (-1,-1), 0),
        ("RIGHTPADDING", (0,0), (-1,-1), 0),
        ("TOPPADDING",   (0,0), (-1,-1), 1),
        ("BOTTOMPADDING",(0,0), (-1,-1), 1),
        ("ALIGN",        (1,0), (1,0), "RIGHT"),
    ]))

    row = Table(
        [[num_cell, inner]],
        colWidths=[12*mm, content_w - 12*mm]
    )
    row.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), 6),
        ("RIGHTPADDING", (0,0), (-1,-1), 8),
        ("VALIGN",       (0,0), (-1,-1), "MIDDLE"),
        ("LINEBELOW",    (0,0), (-1,-1), 0.4, MID_GREY),
    ]))
    return row


# ════════════════════════════════════════════════════════════════════════════
# MORNING SECTION
# ════════════════════════════════════════════════════════════════════════════
story.append(section_header("☀  MORNING ROUTINE", colors.HexColor("#2471A3")))
story.append(Spacer(1, 2*mm))

morning_steps = [
    (
        "1", "CLEANSE",
        "Derma Co 2% Sali-Cinamide Face Wash",
        "Pea-sized amount",
        "Use ONLY if face feels oily. Otherwise rinse with plain water.\n"
        "Massage gently for 60 sec, rinse with cool/lukewarm water.",
        SKY
    ),
    (
        "2", "MOISTURIZE",
        "Derma Co 5% Nia-Ceramide Mattifying Moisturizer",
        "1 fingertip unit",
        "Apply while face is still slightly damp for better absorption.\n"
        "Blend outward — don't forget jawline and neck.",
        WHITE
    ),
    (
        "3", "SUNSCREEN",
        "Derma Co 1% Hyaluronic Sunscreen SPF 50",
        "2-finger rule (face + neck)",
        "Apply 15 min before sun exposure. Reapply every 2-3 hrs if outdoors.\n"
        "Even indoors near windows — UV-A passes through glass.",
        SKY
    ),
]

for s in morning_steps:
    story.append(step_row(*s))
    story.append(Spacer(1, 1*mm))

story.append(Spacer(1, 4*mm))


# ════════════════════════════════════════════════════════════════════════════
# NIGHT SECTION
# ════════════════════════════════════════════════════════════════════════════
story.append(section_header("🌙  NIGHT ROUTINE", NAVY))
story.append(Spacer(1, 2*mm))

night_steps = [
    (
        "1", "CLEANSE",
        "Derma Co 2% Sali-Cinamide Face Wash",
        "Pea-sized amount",
        "Use every night — removes sweat, sunscreen, and daily buildup.\n"
        "Massage for 60 seconds. Rinse thoroughly.",
        MOON
    ),
    (
        "2", "SPOT TREATMENT  (if needed)",
        "Benzoyl Peroxide 2.5% Gel  (Oxy / Persol 2.5 — ~₹80)",
        "Tiny dot — active pimples only",
        "Apply ONLY on active red pimples (especially the nostril area).\n"
        "Do NOT spread on whole face. Let dry 2 min before moisturizer.",
        WHITE
    ),
    (
        "3", "MOISTURIZE",
        "Derma Co 5% Nia-Ceramide Mattifying Moisturizer",
        "1 fingertip unit",
        "Niacinamide at night works to fade acne marks while you sleep.\n"
        "Apply as final step.",
        MOON
    ),
]

for s in night_steps:
    story.append(step_row(*s))
    story.append(Spacer(1, 1*mm))

story.append(Spacer(1, 4*mm))


# ════════════════════════════════════════════════════════════════════════════
# WEEKLY  (1-2x)
# ════════════════════════════════════════════════════════════════════════════
story.append(section_header("📅  WEEKLY  (1–2× per week, Night Only)", colors.HexColor("#7D3C98")))
story.append(Spacer(1, 2*mm))

story.append(step_row(
    "W", "CHEMICAL EXFOLIANT",
    "Minimalist AHA 25% + BHA 5% Peeling Solution  (~₹599)",
    "Thin even layer",
    "Apply to dry face. Leave on 10 min MAX. Rinse off with water.\n"
    "Helps clear forehead bumps (closed comedones). Do NOT use on open/active pimples.\n"
    "Always apply sunscreen the next morning after using this.",
    colors.HexColor("#F5EEF8")
))

story.append(Spacer(1, 5*mm))


# ════════════════════════════════════════════════════════════════════════════
# GOLDEN RULES
# ════════════════════════════════════════════════════════════════════════════
rules_header = Table(
    [[Paragraph("  ✔  GOLDEN RULES", PS("GR", fontSize=11, fontName="Helvetica-Bold",
                                         textColor=WHITE, spaceAfter=0))]],
    colWidths=[content_w]
)
rules_header.setStyle(TableStyle([
    ("BACKGROUND",   (0,0), (-1,-1), colors.HexColor("#1E8449")),
    ("TOPPADDING",   (0,0), (-1,-1), 5),
    ("BOTTOMPADDING",(0,0), (-1,-1), 5),
    ("LEFTPADDING",  (0,0), (-1,-1), 8),
]))
story.append(rules_header)

rules = [
    ("Never skip moisturizer",   "Oily skin still needs hydration. Skipping causes MORE oil production."),
    ("Never skip sunscreen",     "This caused your previous white marks. SPF every day, no exceptions."),
    ("Max 2x cleanse daily",     "Over-washing strips the skin barrier and worsens oiliness."),
    ("60-second rule",           "Salicylic acid needs contact time. Massage for a full minute."),
    ("Don't pick the nose bump", "Nostril crease area scars and gets infected easily. Use spot treatment only."),
    ("One product at a time",    "Wait 1 week before introducing any new product to your routine."),
    ("Patch test first",         "Test all new products on inner arm for 24 hrs before applying to face."),
]

for i, (rule, detail) in enumerate(rules):
    bg = LIGHT_GREY if i % 2 == 0 else WHITE
    row = Table(
        [[
            Paragraph(f"<b>{rule}</b>", PS("RL", fontSize=9, fontName="Helvetica-Bold", textColor=NAVY)),
            Paragraph(detail, SMALL_S)
        ]],
        colWidths=[50*mm, content_w - 50*mm]
    )
    row.setStyle(TableStyle([
        ("BACKGROUND",   (0,0), (-1,-1), bg),
        ("TOPPADDING",   (0,0), (-1,-1), 5),
        ("BOTTOMPADDING",(0,0), (-1,-1), 5),
        ("LEFTPADDING",  (0,0), (-1,-1), 8),
        ("RIGHTPADDING", (0,0), (-1,-1), 8),
        ("VALIGN",       (0,0), (-1,-1), "TOP"),
        ("LINEBELOW",    (0,0), (-1,-1), 0.3, MID_GREY),
    ]))
    story.append(row)

story.append(Spacer(1, 5*mm))


# ════════════════════════════════════════════════════════════════════════════
# RESULTS TIMELINE
# ════════════════════════════════════════════════════════════════════════════
timeline_header = Table(
    [[Paragraph("  ⏱  WHAT TO EXPECT", PS("TH", fontSize=11, fontName="Helvetica-Bold",
                                            textColor=WHITE, spaceAfter=0))]],
    colWidths=[content_w]
)
timeline_header.setStyle(TableStyle([
    ("BACKGROUND",   (0,0), (-1,-1), colors.HexColor("#1A2B4A")),
    ("TOPPADDING",   (0,0), (-1,-1), 5),
    ("BOTTOMPADDING",(0,0), (-1,-1), 5),
    ("LEFTPADDING",  (0,0), (-1,-1), 8),
]))
story.append(timeline_header)

timeline = [
    ("Week 1–2",   "Less shine, skin feels more balanced and less greasy by midday."),
    ("Week 3–4",   "Forehead bumps (closed comedones) start visibly reducing."),
    ("Month 2–3",  "Acne marks and white patch near nose gradually even out (niacinamide + SPF)."),
    ("Month 3+",   "If forehead bumps persist, add Adapalene (Differin) at night — consult a dermatologist."),
]

for i, (period, result) in enumerate(timeline):
    bg = colors.HexColor("#EAF2FF") if i % 2 == 0 else WHITE
    row = Table(
        [[
            Paragraph(f"<b>{period}</b>", PS("TP", fontSize=9, fontName="Helvetica-Bold",
                                              textColor=colors.HexColor("#2471A3"))),
            Paragraph(result, SMALL_S)
        ]],
        colWidths=[28*mm, content_w - 28*mm]
    )
    row.setStyle(TableStyle([
        ("BACKGROUND",   (0,0), (-1,-1), bg),
        ("TOPPADDING",   (0,0), (-1,-1), 5),
        ("BOTTOMPADDING",(0,0), (-1,-1), 5),
        ("LEFTPADDING",  (0,0), (-1,-1), 8),
        ("RIGHTPADDING", (0,0), (-1,-1), 8),
        ("VALIGN",       (0,0), (-1,-1), "MIDDLE"),
        ("LINEBELOW",    (0,0), (-1,-1), 0.3, MID_GREY),
    ]))
    story.append(row)

story.append(Spacer(1, 5*mm))


# ════════════════════════════════════════════════════════════════════════════
# PRODUCT QUICK REFERENCE TABLE
# ════════════════════════════════════════════════════════════════════════════
story.append(section_header("🛒  PRODUCT QUICK REFERENCE", colors.HexColor("#C0392B")))
story.append(Spacer(1, 2*mm))

col_w = [content_w * p for p in [0.38, 0.20, 0.17, 0.25]]
header_row = [
    Paragraph("<b>Product</b>",   PS("PH", fontSize=8, fontName="Helvetica-Bold", textColor=WHITE)),
    Paragraph("<b>When</b>",      PS("PH", fontSize=8, fontName="Helvetica-Bold", textColor=WHITE)),
    Paragraph("<b>Amount</b>",    PS("PH", fontSize=8, fontName="Helvetica-Bold", textColor=WHITE)),
    Paragraph("<b>Est. Cost</b>", PS("PH", fontSize=8, fontName="Helvetica-Bold", textColor=WHITE)),
]

prod_rows = [
    ["Derma Co 2% Sali-Cinamide Face Wash",         "AM (if oily) + PM",   "Pea-sized",          "~₹199"],
    ["Derma Co 5% Nia-Ceramide Moisturizer",         "AM + PM",             "1 fingertip unit",   "~₹349"],
    ["Derma Co SPF 50 Hyaluronic Sunscreen",         "AM only",             "2-finger rule",      "~₹399"],
    ["Benzoyl Peroxide 2.5% (Oxy/Persol 2.5)",       "PM — spot only",      "Tiny dot",           "~₹80"],
    ["Minimalist AHA+BHA Peel (weekly, optional)",   "PM — 1-2x/week",      "Thin layer, 10 min", "~₹599"],
]

table_data = [header_row]
for pr in prod_rows:
    table_data.append([Paragraph(pr[i], PS(f"PC{i}", fontSize=8, fontName="Helvetica",
                                            textColor=colors.HexColor("#222222"))) for i in range(4)])

prod_table = Table(table_data, colWidths=col_w)
prod_table.setStyle(TableStyle([
    ("BACKGROUND",    (0,0), (-1,0), colors.HexColor("#C0392B")),
    ("ROWBACKGROUNDS",(0,1), (-1,-1), [LIGHT_GREY, WHITE]),
    ("TOPPADDING",    (0,0), (-1,-1), 5),
    ("BOTTOMPADDING", (0,0), (-1,-1), 5),
    ("LEFTPADDING",   (0,0), (-1,-1), 6),
    ("RIGHTPADDING",  (0,0), (-1,-1), 6),
    ("GRID",          (0,0), (-1,-1), 0.4, MID_GREY),
    ("VALIGN",        (0,0), (-1,-1), "MIDDLE"),
]))
story.append(prod_table)

story.append(Spacer(1, 4*mm))


# ════════════════════════════════════════════════════════════════════════════
# FOOTER
# ════════════════════════════════════════════════════════════════════════════
footer = Table(
    [[Paragraph(
        "⚠ If the nostril bump doesn't improve in 2 weeks, consult a dermatologist. "
        "Persistent acne responds best to prescription Adapalene or Clindamycin gel. "
        "Generated by Orris · July 2026",
        PS("FT", fontSize=7.5, fontName="Helvetica-Oblique",
           textColor=colors.HexColor("#666666"), alignment=TA_CENTER)
    )]],
    colWidths=[content_w]
)
footer.setStyle(TableStyle([
    ("BACKGROUND",   (0,0), (-1,-1), colors.HexColor("#F0F0F0")),
    ("TOPPADDING",   (0,0), (-1,-1), 6),
    ("BOTTOMPADDING",(0,0), (-1,-1), 6),
    ("LEFTPADDING",  (0,0), (-1,-1), 8),
    ("RIGHTPADDING", (0,0), (-1,-1), 8),
]))
story.append(footer)


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