from reportlab.lib.pagesizes import A4, landscape
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

OUTPUT = "/tmp/workspace/vitamins-table/Vitamins_Study_Table.pdf"

doc = SimpleDocTemplate(
    OUTPUT,
    pagesize=landscape(A4),
    leftMargin=1.2*cm, rightMargin=1.2*cm,
    topMargin=1.5*cm, bottomMargin=1.5*cm
)

styles = getSampleStyleSheet()

def P(text, bold=False, size=8, color=colors.black, align=TA_LEFT):
    style = ParagraphStyle(
        'custom',
        fontName='Helvetica-Bold' if bold else 'Helvetica',
        fontSize=size,
        textColor=color,
        alignment=align,
        leading=size + 3,
        spaceAfter=0
    )
    return Paragraph(text, style)

# ─── HEADER ──────────────────────────────────────────────────────────────────
title_style = ParagraphStyle('title', fontName='Helvetica-Bold', fontSize=16,
                              alignment=TA_CENTER, textColor=colors.HexColor('#1a1a2e'))
subtitle_style = ParagraphStyle('sub', fontName='Helvetica', fontSize=9,
                                 alignment=TA_CENTER, textColor=colors.grey)

elements = []
elements.append(Paragraph("UNIT-08: VITAMINS — Complete Study Table", title_style))
elements.append(Spacer(1, 0.3*cm))
elements.append(Paragraph("Fat-Soluble Vitamins (A, D, E, K) | Water-Soluble Vitamins (C, B-Complex)", subtitle_style))
elements.append(Spacer(1, 0.4*cm))

# ─── COLOR SCHEME ─────────────────────────────────────────────────────────────
FAT_HEADER   = colors.HexColor('#1a1a2e')   # dark navy
FAT_BG       = colors.HexColor('#e8f4f8')   # light blue
FAT_STRIPE   = colors.HexColor('#d0e8f0')
WATER_HEADER = colors.HexColor('#1b4332')   # dark green
WATER_BG     = colors.HexColor('#e8f5e9')
WATER_STRIPE = colors.HexColor('#c8e6c9')
HEADING_TEXT = colors.white

# ─── TABLE DATA ───────────────────────────────────────────────────────────────
col_header = [
    P("Vitamin", bold=True, size=8, color=HEADING_TEXT),
    P("Also Known As", bold=True, size=8, color=HEADING_TEXT),
    P("Solubility", bold=True, size=8, color=HEADING_TEXT),
    P("Food Sources", bold=True, size=8, color=HEADING_TEXT),
    P("Main Functions", bold=True, size=8, color=HEADING_TEXT),
    P("Deficiency Disease / Signs", bold=True, size=8, color=HEADING_TEXT),
    P("Toxicity / Excess", bold=True, size=8, color=HEADING_TEXT),
    P("Memory Tip", bold=True, size=8, color=HEADING_TEXT),
]

fat_section_header = [
    P("FAT-SOLUBLE VITAMINS (A, D, E, K) — Stored in liver & adipose tissue; absorbed with dietary fat; toxicity possible with excess", bold=True, size=8, color=HEADING_TEXT),
    "", "", "", "", "", "", ""
]

rows_fat = [
    [
        P("Vitamin A", bold=True, size=8),
        P("Retinol / Beta-carotene (pro-vitamin)", size=8),
        P("Fat-soluble", size=8),
        P("Liver, egg yolk, dairy, fish oils;\nBeta-carotene: carrots, leafy greens, mango", size=8),
        P("Vision (rhodopsin synthesis)\nEpithelial cell integrity\nImmune function\nBone growth\nReproduction", size=8),
        P("Night blindness (earliest sign)\nXerophthalmia (dry eyes)\nKeratomalacia (corneal ulcer)\nHyperkeratosis (skin)\nIncreased infection risk", size=8),
        P("Hypervitaminosis A:\nHeadache, nausea\nHair loss, dry skin\nTeratogenic in pregnancy\nHepatotoxicity", size=8),
        P("'A' for 'Antixerophthalmic'\nCarrots → Carotene → A", size=8),
    ],
    [
        P("Vitamin D", bold=True, size=8),
        P("Calciferol\nD2: Ergocalciferol (plant)\nD3: Cholecalciferol (skin/animal)", size=8),
        P("Fat-soluble", size=8),
        P("Sunlight (skin synthesis)\nFish liver oils, fatty fish\nEgg yolk, fortified milk", size=8),
        P("Ca²⁺ & PO₄ absorption (intestine)\nBone mineralization\nBlood calcium regulation\nMuscle & immune function", size=8),
        P("Children: Rickets\n(bowed legs, soft bones)\nAdults: Osteomalacia\n(soft bones, bone pain)\nTetany (low Ca²⁺)", size=8),
        P("Hypercalcemia:\nNausea, vomiting\nCalcification of soft tissues\nKidney stones\nRenal damage", size=8),
        P("'D' for 'Deed' (sunlight)\nRickets → Rickets 'D'eficiency", size=8),
    ],
    [
        P("Vitamin E", bold=True, size=8),
        P("Tocopherol\n(alpha-tocopherol most active)", size=8),
        P("Fat-soluble", size=8),
        P("Vegetable oils (wheat germ, sunflower)\nNuts, seeds\nGreen leafy vegetables", size=8),
        P("Antioxidant (protects cell membranes)\nPrevents lipid peroxidation\nImmune function\nRBC protection", size=8),
        P("Rare in humans\nHemolytic anemia (newborns)\nNeuromuscular degeneration\nAtaxia, peripheral neuropathy", size=8),
        P("Generally low toxicity\nHigh doses may interfere\nwith Vitamin K (anticoagulant\neffect)", size=8),
        P("'E' for antioxidant\n('E'liminates free radicals)", size=8),
    ],
    [
        P("Vitamin K", bold=True, size=8),
        P("K1: Phylloquinone (plant)\nK2: Menaquinone (bacteria)\nK3: Menadione (synthetic)", size=8),
        P("Fat-soluble", size=8),
        P("Green leafy vegetables\n(spinach, kale, broccoli)\nAlso synthesized by gut bacteria", size=8),
        P("Blood clotting (factors II, VII, IX, X)\nCarboxylation of clotting factors\nBone metabolism (osteocalcin)", size=8),
        P("Bleeding tendency / Hemorrhage\nNewborns at risk\n(Hemorrhagic disease of newborn)\nProlonged PT/INR", size=8),
        P("K1/K2: Low toxicity\nK3 (menadione): Hemolytic\nanemia in newborns\n(avoid synthetic K3)", size=8),
        P("'K' for 'Koagulation' (German)\nAnticoagulants (warfarin)\nblock Vitamin K", size=8),
    ],
]

water_section_header = [
    P("WATER-SOLUBLE VITAMINS (C & B-Complex) — Not stored (excreted in urine); toxicity rare; daily intake required", bold=True, size=8, color=HEADING_TEXT),
    "", "", "", "", "", "", ""
]

rows_water = [
    [
        P("Vitamin C", bold=True, size=8),
        P("Ascorbic acid", size=8),
        P("Water-soluble", size=8),
        P("Citrus fruits (lemon, orange)\nStrawberries, kiwi\nBroccoli, bell peppers, potatoes", size=8),
        P("Collagen synthesis\nAntioxidant\nIron absorption (Fe³⁺→Fe²⁺)\nWound healing\nImmune function", size=8),
        P("Scurvy:\nBleeding gums, loose teeth\nPerifollicular hemorrhage\nPoor wound healing\nAnemia, fatigue", size=8),
        P("Kidney stones (oxalate)\nGI upset, diarrhea\nRebound scurvy on abrupt stop", size=8),
        P("'C' for Citrus & Collagen\nScurvy → Sailors (no fruit at sea)", size=8),
    ],
    [
        P("Vitamin B1", bold=True, size=8),
        P("Thiamine", size=8),
        P("Water-soluble", size=8),
        P("Pork, legumes\nWhole grains, nuts\nBrewer's yeast", size=8),
        P("Coenzyme in CHO metabolism\n(Pyruvate dehydrogenase,\nα-ketoglutarate DH, TK)\nNerve conduction", size=8),
        P("Beriberi:\n- Wet: cardiac (edema, heart failure)\n- Dry: peripheral neuropathy\nWernicke-Korsakoff (alcoholics):\nAtaxia, confusion, ophthalmoplegia", size=8),
        P("Rare (water-soluble)\nExcess excreted in urine", size=8),
        P("B1 → Beriberi\n'Thiamine = Think'\n(neurological symptoms)", size=8),
    ],
    [
        P("Vitamin B2", bold=True, size=8),
        P("Riboflavin", size=8),
        P("Water-soluble", size=8),
        P("Dairy products, eggs\nMeat, poultry, fish\nGreen leafy vegetables", size=8),
        P("Component of FAD & FMN\n(electron carriers)\nEnergy metabolism\nAntioxidant (via glutathione)", size=8),
        P("Ariboflavinosis:\nAngular stomatitis (cracked mouth)\nCheilosis (lip inflammation)\nGlossitis (magenta tongue)\nCorneal vascularization", size=8),
        P("No known toxicity\nExcess excreted in urine\n(turns urine yellow-green)", size=8),
        P("B2 = '2 lips' cracked\nRiboflavin → Red-colored urine\n'Flavi' = yellow (Latin)", size=8),
    ],
    [
        P("Vitamin B3", bold=True, size=8),
        P("Niacin\n(Nicotinic acid /\nNicotinamide)", size=8),
        P("Water-soluble", size=8),
        P("Meat, poultry, fish\nWhole grains, mushrooms\nCan be made from Tryptophan", size=8),
        P("Component of NAD⁺ & NADP⁺\nEnergy metabolism\nFatty acid synthesis\nDNA repair", size=8),
        P("Pellagra — 3 D's:\n1. Dermatitis (sun-exposed skin)\n2. Diarrhea\n3. Dementia\n(+ Death if untreated)", size=8),
        P("Flushing (prostaglandin)\nHyperglycemia\nHepatotoxicity (high dose)\nGout (hyperuricemia)", size=8),
        P("Pellagra = 3 D's\nNiacin treats dyslipidemia\n(raises HDL)", size=8),
    ],
    [
        P("Vitamin B5", bold=True, size=8),
        P("Pantothenic acid", size=8),
        P("Water-soluble", size=8),
        P("Widespread in foods\n(panto = everywhere)\nMeat, whole grains, avocado\nBroccoli, eggs", size=8),
        P("Component of Coenzyme A (CoA)\nFatty acid synthesis & oxidation\nCarbohydrate & protein metabolism\nSteroid hormone synthesis", size=8),
        P("Extremely rare\n'Burning feet' syndrome\nFatigue, GI disturbances\nIrritability", size=8),
        P("No known toxicity\nMay cause diarrhea at\nvery high doses", size=8),
        P("'Panto' = everywhere\n(found in almost all foods)\nB5 → CoA", size=8),
    ],
    [
        P("Vitamin B6", bold=True, size=8),
        P("Pyridoxine\n(Pyridoxal, Pyridoxamine)", size=8),
        P("Water-soluble", size=8),
        P("Meat, poultry, fish\nPotatoes, bananas\nFortified cereals", size=8),
        P("Coenzyme in amino acid\ntransaminases (transamination)\nGlycogenolysis\nNeurotransmitter synthesis\n(serotonin, GABA, dopamine)", size=8),
        P("Peripheral neuropathy\nSeborrheic dermatitis\nGlossitis, cheilosis\nMicrocytic anemia\nIrritability, confusion", size=8),
        P("Peripheral sensory neuropathy\n(high doses, >200 mg/day)\nUnsteady gait", size=8),
        P("B6 → Pyridox → 'Pride'\n(transamination & neurotransmitters)\nIsoniazid (TB drug) causes B6 deficiency", size=8),
    ],
    [
        P("Vitamin B7", bold=True, size=8),
        P("Biotin", size=8),
        P("Water-soluble", size=8),
        P("Egg yolk, liver\nNuts, seeds\nSome vegetables\nGut bacteria synthesize it", size=8),
        P("Coenzyme for carboxylases\nFatty acid synthesis\nGluconeogenesis\nAmino acid catabolism", size=8),
        P("Rare deficiency\nHair loss (alopecia)\nDermatitis (scaly red rash)\nNeurological symptoms\nEgg white injury (avidin binds B7)", size=8),
        P("No known toxicity\nExcess excreted in urine", size=8),
        P("'Biotin = Beautiful hair'\nRaw egg white blocks absorption\n(avidin protein)", size=8),
    ],
    [
        P("Vitamin B9", bold=True, size=8),
        P("Folic acid\n(Folate)", size=8),
        P("Water-soluble", size=8),
        P("Green leafy vegetables\n(spinach, asparagus)\nLegumes, citrus fruits\nFortified cereals, liver", size=8),
        P("DNA synthesis (thymidine)\nCell division & growth\nAmino acid metabolism\n(homocysteine → methionine)\nRBC formation", size=8),
        P("Megaloblastic (macrocytic) anemia\nNeural tube defects (fetus)\n(spina bifida, anencephaly)\nGlossitis, glossitis\nHyperhomocysteinemia", size=8),
        P("May mask B12 deficiency\nNo known direct toxicity\nMay lower seizure threshold", size=8),
        P("Folate → Fetal (neural tube)\nMust supplement BEFORE pregnancy\nMTX blocks folate (cancer Rx)", size=8),
    ],
    [
        P("Vitamin B12", bold=True, size=8),
        P("Cobalamin\n(contains cobalt)", size=8),
        P("Water-soluble", size=8),
        P("ONLY in animal products:\nMeat, fish, poultry\nDairy, eggs, shellfish\n(Vegans at high risk!)", size=8),
        P("DNA synthesis (with folate)\nMyelin sheath maintenance\nNerve function\nRBC formation\nHomocysteine metabolism", size=8),
        P("Megaloblastic anemia\nSubacute combined degeneration\n(spinal cord demyelination)\nPeripheral neuropathy\nGlossitis, neuropsychiatric Sx", size=8),
        P("No toxicity known\nExcess stored in liver\n(years of stores)", size=8),
        P("B12 = needs Intrinsic Factor (IF)\nPernicious anemia = no IF\nVegans + B12 deficiency\nContains Cobalt (Co-balamin)", size=8),
    ],
]

# ─── ASSEMBLE TABLE ───────────────────────────────────────────────────────────
col_widths = [2.1*cm, 2.5*cm, 1.8*cm, 3.5*cm, 4.0*cm, 4.5*cm, 3.8*cm, 3.8*cm]

all_data = [col_header, fat_section_header] + rows_fat + [water_section_header] + rows_water

table = Table(all_data, colWidths=col_widths, repeatRows=1)

# Row indices
n_col_header = 1
n_fat_sh = 1
n_fat = len(rows_fat)
n_water_sh = 1
n_water = len(rows_water)

fat_start = n_col_header + n_fat_sh  # row 2
fat_end = fat_start + n_fat - 1       # row 5
water_sh_row = fat_end + 1            # row 6
water_start = water_sh_row + 1        # row 7
water_end = water_start + n_water - 1

style = TableStyle([
    # ─── Column header ───
    ('BACKGROUND', (0, 0), (-1, 0), colors.HexColor('#2d2d2d')),
    ('TEXTCOLOR', (0, 0), (-1, 0), colors.white),
    ('FONTNAME', (0, 0), (-1, 0), 'Helvetica-Bold'),
    ('FONTSIZE', (0, 0), (-1, 0), 8),
    ('ALIGN', (0, 0), (-1, 0), 'CENTER'),
    ('VALIGN', (0, 0), (-1, 0), 'MIDDLE'),
    ('ROWBACKGROUND', (0, 0), (-1, 0), colors.HexColor('#2d2d2d')),

    # ─── Fat-soluble section header ───
    ('BACKGROUND', (0, 1), (-1, 1), FAT_HEADER),
    ('SPAN', (0, 1), (-1, 1)),
    ('ALIGN', (0, 1), (-1, 1), 'LEFT'),
    ('VALIGN', (0, 1), (-1, 1), 'MIDDLE'),

    # ─── Fat-soluble rows (alternating) ───
    *[('BACKGROUND', (0, fat_start + i), (-1, fat_start + i),
       FAT_BG if i % 2 == 0 else FAT_STRIPE) for i in range(n_fat)],

    # ─── Water-soluble section header ───
    ('BACKGROUND', (0, water_sh_row), (-1, water_sh_row), WATER_HEADER),
    ('SPAN', (0, water_sh_row), (-1, water_sh_row)),
    ('ALIGN', (0, water_sh_row), (-1, water_sh_row), 'LEFT'),
    ('VALIGN', (0, water_sh_row), (-1, water_sh_row), 'MIDDLE'),

    # ─── Water-soluble rows (alternating) ───
    *[('BACKGROUND', (0, water_start + i), (-1, water_start + i),
       WATER_BG if i % 2 == 0 else WATER_STRIPE) for i in range(n_water)],

    # ─── Global grid & padding ───
    ('GRID', (0, 0), (-1, -1), 0.5, colors.HexColor('#888888')),
    ('VALIGN', (0, 0), (-1, -1), 'TOP'),
    ('LEFTPADDING', (0, 0), (-1, -1), 4),
    ('RIGHTPADDING', (0, 0), (-1, -1), 4),
    ('TOPPADDING', (0, 0), (-1, -1), 4),
    ('BOTTOMPADDING', (0, 0), (-1, -1), 4),

    # ─── Bold vitamin name column ───
    ('FONTNAME', (0, fat_start), (0, fat_end), 'Helvetica-Bold'),
    ('FONTNAME', (0, water_start), (0, water_end), 'Helvetica-Bold'),
])

table.setStyle(style)
elements.append(table)

# ─── FOOTER ───────────────────────────────────────────────────────────────────
elements.append(Spacer(1, 0.3*cm))
footer_style = ParagraphStyle('footer', fontName='Helvetica-Oblique', fontSize=7,
                               textColor=colors.grey, alignment=TA_CENTER)
elements.append(Paragraph(
    "Unit-08 Vitamins | Source: Basic Medical Biochemistry (Lippincott) & Washington Manual of Medical Therapeutics | For study use only",
    footer_style
))

doc.build(elements)
print("PDF generated:", OUTPUT)