from reportlab.platypus import (
    SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle,
    HRFlowable, KeepTogether
)
from reportlab.lib.pagesizes import A4
from reportlab.lib import colors
from reportlab.lib.styles import ParagraphStyle
from reportlab.lib.units import cm, mm
from reportlab.lib.enums import TA_CENTER, TA_LEFT
from reportlab.pdfbase import pdfmetrics
from reportlab.pdfbase.ttfonts import TTFont
import os

# ── Register Telugu fonts ──────────────────────────────────────────────────────
BASE = "/tmp/workspace/telugu-mealplan"
pdfmetrics.registerFont(TTFont("Telugu", f"{BASE}/NotoSansTelugu-Regular.ttf"))
pdfmetrics.registerFont(TTFont("Telugu-Bold", f"{BASE}/NotoSansTelugu-Bold.ttf"))

# ── Colour palette ─────────────────────────────────────────────────────────────
GREEN_DARK  = colors.HexColor("#1B5E20")
GREEN_MED   = colors.HexColor("#2E7D32")
GREEN_LIGHT = colors.HexColor("#A5D6A7")
GREEN_BG    = colors.HexColor("#E8F5E9")
ORANGE      = colors.HexColor("#E65100")
AMBER       = colors.HexColor("#FF8F00")
WHITE       = colors.white
GREY_TEXT   = colors.HexColor("#424242")
LIGHT_GREY  = colors.HexColor("#F5F5F5")
YELLOW_BG   = colors.HexColor("#FFFDE7")

# ── Styles ─────────────────────────────────────────────────────────────────────
def S(name, font="Telugu", size=10, leading=16, color=GREY_TEXT,
      align=TA_LEFT, spaceAfter=4, spaceBefore=2, bold=False):
    return ParagraphStyle(
        name,
        fontName="Telugu-Bold" if bold else font,
        fontSize=size,
        leading=leading,
        textColor=color,
        alignment=align,
        spaceAfter=spaceAfter,
        spaceBefore=spaceBefore,
    )

title_style     = S("title",    size=22, leading=28, color=WHITE,  align=TA_CENTER, bold=True)
sub_style       = S("sub",      size=12, leading=18, color=WHITE,  align=TA_CENTER)
section_style   = S("section",  size=13, leading=18, color=WHITE,  bold=True, align=TA_CENTER)
body_style      = S("body",     size=10, leading=16)
body_bold       = S("body_b",   size=10, leading=16, bold=True)
small_style     = S("small",    size=9,  leading=14, color=colors.HexColor("#555555"))
note_style      = S("note",     size=9,  leading=14, color=ORANGE)
cal_style       = S("cal",      size=9,  leading=14, color=GREEN_DARK, bold=True, align=TA_CENTER)
total_style     = S("total",    size=10, leading=16, color=GREEN_DARK, bold=True)

# ── Helper: coloured section header ───────────────────────────────────────────
def section_header(text, bg=GREEN_MED):
    data = [[Paragraph(text, section_style)]]
    t = Table(data, colWidths=[17*cm])
    t.setStyle(TableStyle([
        ("BACKGROUND", (0,0), (-1,-1), bg),
        ("ROUNDEDCORNERS", [5]),
        ("TOPPADDING",    (0,0), (-1,-1), 6),
        ("BOTTOMPADDING", (0,0), (-1,-1), 6),
        ("LEFTPADDING",   (0,0), (-1,-1), 10),
        ("RIGHTPADDING",  (0,0), (-1,-1), 10),
    ]))
    return t

# ── Helper: info box (yellow bg) ──────────────────────────────────────────────
def info_box(text):
    data = [[Paragraph(text, note_style)]]
    t = Table(data, colWidths=[17*cm])
    t.setStyle(TableStyle([
        ("BACKGROUND", (0,0), (-1,-1), YELLOW_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),
        ("BOX", (0,0), (-1,-1), 0.5, AMBER),
        ("ROUNDEDCORNERS", [4]),
    ]))
    return t

# ── Helper: meal card ─────────────────────────────────────────────────────────
def meal_card(time_label, cal_label, items, bg=GREEN_BG):
    header_data = [[
        Paragraph(time_label, S("mh", size=10, leading=14, color=GREEN_DARK, bold=True)),
        Paragraph(cal_label,  cal_style),
    ]]
    header = Table(header_data, colWidths=[13*cm, 4*cm])
    header.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),
        ("LINEBELOW", (0,0), (-1,-1), 0.5, GREEN_LIGHT),
    ]))

    rows = []
    for item in items:
        rows.append([Paragraph(f"• {item}", body_style)])
    body_tbl = Table(rows, colWidths=[17*cm])
    body_tbl.setStyle(TableStyle([
        ("BACKGROUND", (0,0), (-1,-1), WHITE),
        ("TOPPADDING",    (0,0), (-1,-1), 3),
        ("BOTTOMPADDING", (0,0), (-1,-1), 3),
        ("LEFTPADDING",   (0,0), (-1,-1), 14),
        ("RIGHTPADDING",  (0,0), (-1,-1), 8),
    ]))

    outer = Table([[header], [body_tbl]], colWidths=[17*cm])
    outer.setStyle(TableStyle([
        ("BOX",   (0,0), (-1,-1), 0.8, GREEN_MED),
        ("ROUNDEDCORNERS", [5]),
        ("TOPPADDING",    (0,0), (-1,-1), 0),
        ("BOTTOMPADDING", (0,0), (-1,-1), 0),
        ("LEFTPADDING",   (0,0), (-1,-1), 0),
        ("RIGHTPADDING",  (0,0), (-1,-1), 0),
    ]))
    return outer

# ── Build document ─────────────────────────────────────────────────────────────
OUTPUT = f"{BASE}/Telugu_Meal_Plan.pdf"
doc = SimpleDocTemplate(
    OUTPUT,
    pagesize=A4,
    rightMargin=2*cm, leftMargin=2*cm,
    topMargin=2*cm, bottomMargin=2*cm,
    title="Telugu Weight Loss Meal Plan",
    author="Orris Health",
)

story = []

# ── COVER HEADER ──────────────────────────────────────────────────────────────
cover_data = [[
    Paragraph("🥗 రోజువారీ బరువు తగ్గించుకునే మెనూ", title_style),
    Paragraph("Daily Weight Loss Meal Plan", sub_style),
    Paragraph("స్త్రీ | వయసు 36 | బరువు 68 కిలోలు | భారతీయ", sub_style),
]]
cover_rows = [[Paragraph("🥗 రోజువారీ బరువు తగ్గించుకునే మెనూ", title_style)],
              [Paragraph("Daily Weight Loss Meal Plan", sub_style)],
              [Paragraph("స్త్రీ | వయసు 36 | బరువు 68 కిలోలు | భారతీయ", sub_style)]]
cover = Table(cover_rows, colWidths=[17*cm])
cover.setStyle(TableStyle([
    ("BACKGROUND",    (0,0), (-1,-1), GREEN_DARK),
    ("TOPPADDING",    (0,0), (-1,-1), 8),
    ("BOTTOMPADDING", (0,0), (-1,-1), 8),
    ("LEFTPADDING",   (0,0), (-1,-1), 12),
    ("RIGHTPADDING",  (0,0), (-1,-1), 12),
    ("ROUNDEDCORNERS", [8]),
]))
story.append(cover)
story.append(Spacer(1, 10))

# ── PROFILE BOX ───────────────────────────────────────────────────────────────
profile_rows = [
    [Paragraph("లక్ష్యం", S("ph", size=9, bold=True, color=GREEN_DARK)),
     Paragraph("వారానికి ~0.5 కిలో తగ్గించుకోవడం", small_style)],
    [Paragraph("రోజువారీ కేలరీలు", S("ph2", size=9, bold=True, color=GREEN_DARK)),
     Paragraph("1,400 – 1,500 కేలరీలు (కొవ్వు తగ్గింపు కోసం)", small_style)],
    [Paragraph("మాక్రో నిష్పత్తి", S("ph3", size=9, bold=True, color=GREEN_DARK)),
     Paragraph("కార్బ్స్ 50% | ప్రోటీన్ 25% | కొవ్వు 25%", small_style)],
]
profile = Table(profile_rows, colWidths=[5*cm, 12*cm])
profile.setStyle(TableStyle([
    ("BACKGROUND",    (0,0), (-1,-1), GREEN_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),
    ("BOX",           (0,0), (-1,-1), 1, GREEN_MED),
    ("LINEAFTER",     (0,0), (0,-1), 0.5, GREEN_LIGHT),
]))
story.append(profile)
story.append(Spacer(1, 12))

# ══════════════════════════════════════════════════════════════════════════════
# MEALS
# ══════════════════════════════════════════════════════════════════════════════
story.append(section_header("🍽️ రోజువారీ భోజన వేళలు"))
story.append(Spacer(1, 8))

# 1. Wake-up
story.append(meal_card(
    "🌅 ఉదయం 7:00 – లేచిన వెంటనే",
    "~10 కేలరీలు",
    [
        "గోరువెచ్చని నీళ్ళు 1 గ్లాసు + సగం నిమ్మకాయ రసం + జీలకర్ర పొడి కొంచెం",
        "లేదా రాత్రి నానబెట్టిన మెంతి నీళ్ళు 1 గ్లాసు",
    ]
))
story.append(Spacer(1, 6))

# 2. Breakfast
story.append(meal_card(
    "🍳 అల్పాహారం – ఉదయం 8:00–8:30",
    "~350 కేలరీలు",
    [
        "ఆప్షన్ A: 2 పెసరపప్పు చిల్లా + పుదీనా చట్నీ + తక్కువ కొవ్వు పెరుగు 100గ్రా + గ్రీన్ టీ",
        "ఆప్షన్ B: కూరగాయల వోట్స్ ఉప్మా + 1 ఉడికించిన గుడ్డు లేదా 50గ్రా పనీర్ + బ్లాక్ కాఫీ",
        "ఆప్షన్ C (వారాంతాలు): 2 బేసన్ చిల్లా (పాలకూర+ఉల్లిపాయ) + మజ్జిగ",
    ]
))
story.append(Spacer(1, 6))

# 3. Mid-morning snack
story.append(meal_card(
    "🍎 మధ్యాహ్నం స్నాక్ – 10:30–11:00",
    "~100 కేలరీలు",
    [
        "1 మీడియం సీజన్ పండు: యాపిల్ / జామ / బొప్పాయి / పేర్లు",
        "లేదా నానబెట్టిన బాదం 5–6 గింజలు",
    ]
))
story.append(Spacer(1, 6))

# 4. Lunch
story.append(meal_card(
    "🍛 మధ్యాహ్నం భోజనం – 1:00–1:30",
    "~450 కేలరీలు",
    [
        "1 గిన్నె పప్పు (మసూర్/పెసర/శనగ పప్పు)",
        "1 కాటోరి కూర (లౌకీ, టిండా, బెండకాయ, పాలకూర, మెంతికూర)",
        "1–2 గోధుమ పిండి రొట్టెలు (నెయ్యి తక్కువగా లేదా అసలే వేయకుండా)",
        "1 చిన్న గిన్నె సలాడ్ (దోసకాయ, టమాటా, ఉల్లిపాయ, నిమ్మరసం)",
        "తక్కువ కొవ్వు పెరుగు 100గ్రా",
        "కొన్ని రోజులు: బ్రౌన్ రైస్ 1 కప్పు (రొట్టె బదులు)",
    ]
))
story.append(Spacer(1, 6))

# 5. Evening snack
story.append(meal_card(
    "☕ సాయంత్రం స్నాక్ – 4:00–4:30",
    "~100 కేలరీలు",
    [
        "1 కప్పు మసాలా చాయ్ (తక్కువ కొవ్వు పాలతో, చక్కెర లేకుండా)",
        "దానితో: వేయించిన మఖానా 20–25గ్రా",
        "లేదా వేయించిన నల్ల శనగలు / కందిపప్పు",
        "లేదా 2 రైస్ కేక్స్ / మల్టీగ్రెయిన్ క్రాకర్స్",
    ]
))
story.append(Spacer(1, 6))

# 6. Dinner
story.append(meal_card(
    "🥘 రాత్రి భోజనం – 7:30–8:00",
    "~400 కేలరీలు",
    [
        "1 గిన్నె కూరగాయల సూప్ (టమాటా / పాలకూర / మిక్స్డ్ వెజ్, క్రీం లేకుండా)",
        "1–2 గోధుమ పిండి రొట్టెలు",
        "1 గిన్నె ఆకుకూర / నీళ్ళు ఎక్కువగా ఉండే కూరగాయల కూర",
        "50గ్రా పనీర్ లేదా కొంచెం పప్పు (ప్రోటీన్ కోసం)",
        "తేలికైన ఆప్షన్: ఖిచిడీ (పెసరపప్పు+బియ్యం) + సలాడ్ + మజ్జిగ",
    ]
))
story.append(Spacer(1, 6))

# 7. Bedtime
story.append(meal_card(
    "🌙 నిద్రపోయే ముందు – రాత్రి 10:00",
    "~50 కేలరీలు",
    [
        "1 చిన్న కప్పు పసుపు పాలు (తక్కువ కొవ్వు పాలతో, చక్కెర లేకుండా)",
    ],
    bg=colors.HexColor("#EDE7F6")
))

story.append(Spacer(1, 12))

# ══════════════════════════════════════════════════════════════════════════════
# CALORIE SUMMARY TABLE
# ══════════════════════════════════════════════════════════════════════════════
story.append(section_header("📋 రోజువారీ కేలరీల సారాంశం", bg=ORANGE))
story.append(Spacer(1, 8))

cal_data = [
    [Paragraph("భోజనం", S("th", size=10, bold=True, color=WHITE)),
     Paragraph("కేలరీలు (సుమారు)", S("th2", size=10, bold=True, color=WHITE, align=TA_CENTER))],
    [Paragraph("లేచిన వెంటనే పానీయం", body_style),     Paragraph("10",       cal_style)],
    [Paragraph("అల్పాహారం",            body_style),     Paragraph("350",      cal_style)],
    [Paragraph("మధ్యాహ్నం స్నాక్",     body_style),     Paragraph("100",      cal_style)],
    [Paragraph("మధ్యాహ్నం భోజనం",      body_style),     Paragraph("450",      cal_style)],
    [Paragraph("సాయంత్రం స్నాక్",      body_style),     Paragraph("100",      cal_style)],
    [Paragraph("రాత్రి భోజనం",          body_style),     Paragraph("400",      cal_style)],
    [Paragraph("నిద్రముందు పానీయం",    body_style),     Paragraph("50",       cal_style)],
    [Paragraph("మొత్తం", total_style),                   Paragraph("~1,460",   total_style)],
]
cal_table = Table(cal_data, colWidths=[12*cm, 5*cm])
cal_table.setStyle(TableStyle([
    ("BACKGROUND", (0,0), (-1,0), ORANGE),
    ("BACKGROUND", (0,1), (-1,-2), WHITE),
    ("BACKGROUND", (0,-1), (-1,-1), GREEN_BG),
    ("ROWBACKGROUNDS", (0,1), (-1,-2), [WHITE, LIGHT_GREY]),
    ("BOX",         (0,0), (-1,-1), 0.8, ORANGE),
    ("INNERGRID",   (0,0), (-1,-1), 0.3, colors.HexColor("#CCCCCC")),
    ("TOPPADDING",    (0,0), (-1,-1), 6),
    ("BOTTOMPADDING", (0,0), (-1,-1), 6),
    ("LEFTPADDING",   (0,0), (-1,-1), 10),
    ("RIGHTPADDING",  (0,0), (-1,-1), 10),
    ("LINEABOVE",   (0,-1), (-1,-1), 1.5, ORANGE),
]))
story.append(cal_table)
story.append(Spacer(1, 12))

# ══════════════════════════════════════════════════════════════════════════════
# DO / DONT TABLE
# ══════════════════════════════════════════════════════════════════════════════
story.append(section_header("✅ పాటించవలసిన నియమాలు"))
story.append(Spacer(1, 8))

rules_data = [
    [Paragraph("ఇవి చేయండి ✔️", S("rh1", size=10, bold=True, color=WHITE)),
     Paragraph("ఇవి మానండి ❌",  S("rh2", size=10, bold=True, color=WHITE))],
    [Paragraph("రోజూ 8–10 గ్లాసుల నీళ్ళు తాగండి", body_style),
     Paragraph("మైదా వస్తువులు – పూరీ, పరాఠా", body_style)],
    [Paragraph("ప్రతి 3–4 గంటలకు తినండి", body_style),
     Paragraph("పాకెట్ బిస్కెట్లు, నమ్కీన్, చిప్స్", body_style)],
    [Paragraph("ఆవాల నూనె / ఆలివ్ ఆయిల్ 1–2 టీ.స్పూన్", body_style),
     Paragraph("వేయించిన తిండి, సమోసా, వడ", body_style)],
    [Paragraph("రోజూ 30–45 నిమిషాలు నడవండి", body_style),
     Paragraph("చక్కెర పానీయాలు, కోల్డ్ డ్రింక్స్", body_style)],
    [Paragraph("నెమ్మదిగా తినండి, బాగా నమలండి", body_style),
     Paragraph("రాత్రి 9 తర్వాత తినడం", body_style)],
    [Paragraph("పప్పు/పెరుగు/పనీర్/గుడ్డు చేర్చండి", body_style),
     Paragraph("భోజనం దాటవేయడం", body_style)],
    [Paragraph("ఆవిరిలో వండటం / గ్రిల్ చేయడం", body_style),
     Paragraph("అధికంగా తెల్ల బియ్యం తినడం", body_style)],
]
rules_table = Table(rules_data, colWidths=[8.5*cm, 8.5*cm])
rules_table.setStyle(TableStyle([
    ("BACKGROUND",  (0,0), (0,0), GREEN_DARK),
    ("BACKGROUND",  (1,0), (1,0), colors.HexColor("#C62828")),
    ("ROWBACKGROUNDS", (0,1), (-1,-1), [WHITE, GREEN_BG]),
    ("BOX",         (0,0), (-1,-1), 0.8, GREEN_MED),
    ("INNERGRID",   (0,0), (-1,-1), 0.3, colors.HexColor("#CCCCCC")),
    ("LINEAFTER",   (0,0), (0,-1), 0.5, GREEN_LIGHT),
    ("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(rules_table)
story.append(Spacer(1, 12))

# ══════════════════════════════════════════════════════════════════════════════
# SAMPLE DAY TABLE
# ══════════════════════════════════════════════════════════════════════════════
story.append(section_header("🥣 నమూనా రోజు (సోమవారం)", bg=colors.HexColor("#6A1B9A")))
story.append(Spacer(1, 8))

sample_data = [
    [Paragraph("సమయం", S("sh1", size=10, bold=True, color=WHITE)),
     Paragraph("తినవలసింది", S("sh2", size=10, bold=True, color=WHITE))],
    [Paragraph("ఉదయం 7:00",      body_bold), Paragraph("గోరువెచ్చని నిమ్మనీళ్ళు", body_style)],
    [Paragraph("ఉదయం 8:00",      body_bold), Paragraph("2 పెసరపప్పు చిల్లా + పెరుగు + గ్రీన్ టీ", body_style)],
    [Paragraph("ఉదయం 10:30",     body_bold), Paragraph("1 జామ పండు", body_style)],
    [Paragraph("మ.హ్నం 1:00",    body_bold), Paragraph("2 రొట్టెలు + మసూర్ పప్పు + లౌకీ కూర + సలాడ్ + పెరుగు", body_style)],
    [Paragraph("సా.త్రం 4:00",   body_bold), Paragraph("వేయించిన మఖానా + మసాలా చాయ్", body_style)],
    [Paragraph("సా.త్రం 7:30",   body_bold), Paragraph("కూరగాయల సూప్ + 1 రొట్టె + పాలక్ పనీర్ (కొంచెం)", body_style)],
    [Paragraph("రాత్రి 10:00",   body_bold), Paragraph("పసుపు పాలు", body_style)],
]
sample_table = Table(sample_data, colWidths=[4.5*cm, 12.5*cm])
sample_table.setStyle(TableStyle([
    ("BACKGROUND",  (0,0), (-1,0), colors.HexColor("#6A1B9A")),
    ("ROWBACKGROUNDS", (0,1), (-1,-1), [WHITE, colors.HexColor("#F3E5F5")]),
    ("BOX",         (0,0), (-1,-1), 0.8, colors.HexColor("#6A1B9A")),
    ("INNERGRID",   (0,0), (-1,-1), 0.3, colors.HexColor("#CCCCCC")),
    ("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(sample_table)
story.append(Spacer(1, 12))

# ══════════════════════════════════════════════════════════════════════════════
# EXERCISE BOX
# ══════════════════════════════════════════════════════════════════════════════
ex_rows = [
    [Paragraph("🏋️ వ్యాయామం సూచన", S("exh", size=11, bold=True, color=WHITE))],
    [Paragraph("ఉదయం: 30–45 నిమిషాలు వేగంగా నడవడం లేదా యోగా (వారానికి 5 రోజులు)", body_style)],
    [Paragraph("సాయంత్రం: 15–20 నిమిషాలు తేలికపాటి స్ట్రెచింగ్ లేదా ఇంట్లో వ్యాయామాలు", body_style)],
]
ex_table = Table(ex_rows, colWidths=[17*cm])
ex_table.setStyle(TableStyle([
    ("BACKGROUND",  (0,0), (0,0), colors.HexColor("#0277BD")),
    ("BACKGROUND",  (0,1), (0,-1), colors.HexColor("#E1F5FE")),
    ("BOX",         (0,0), (-1,-1), 0.8, colors.HexColor("#0277BD")),
    ("TOPPADDING",    (0,0), (-1,-1), 7),
    ("BOTTOMPADDING", (0,0), (-1,-1), 7),
    ("LEFTPADDING",   (0,0), (-1,-1), 10),
    ("RIGHTPADDING",  (0,0), (-1,-1), 10),
]))
story.append(ex_table)
story.append(Spacer(1, 10))

# ── FOOTER NOTE ───────────────────────────────────────────────────────────────
story.append(info_box(
    "💡 ఆశించే ఫలితం: నిరంతరంగా పాటిస్తే వారానికి 0.4–0.8 కిలోలు తగ్గుతుంది. "
    "మీకు థైరాయిడ్, పీసీఓఎస్, లేదా డయాబెటీస్ వంటి సమస్యలు ఉంటే వైద్యుడిని సంప్రదించండి."
))
story.append(Spacer(1, 6))

footer_style = S("footer", size=8, color=colors.HexColor("#9E9E9E"), align=TA_CENTER)
story.append(Paragraph("ఈ మెనూ Orris ద్వారా రూపొందించబడింది • July 2026", footer_style))

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