from docx import Document
from docx.shared import Pt, Inches, RGBColor
from docx.enum.text import WD_ALIGN_PARAGRAPH
from docx.oxml.ns import qn
from docx.oxml import OxmlElement

doc = Document()

# ── Page margins ──────────────────────────────────────────────
section = doc.sections[0]
section.page_width  = Inches(8.27)   # A4
section.page_height = Inches(11.69)
section.left_margin   = Inches(0.9)
section.right_margin  = Inches(0.9)
section.top_margin    = Inches(0.8)
section.bottom_margin = Inches(0.8)

# ── Default style ─────────────────────────────────────────────
normal = doc.styles['Normal']
normal.font.name = 'Arial'
normal.font.size = Pt(11)

# ── Helper functions ──────────────────────────────────────────
def add_colored_heading(doc, text, level=1, r=0, g=80, b=160):
    p = doc.add_heading(text, level=level)
    p.alignment = WD_ALIGN_PARAGRAPH.CENTER if level == 1 else WD_ALIGN_PARAGRAPH.LEFT
    for run in p.runs:
        run.font.color.rgb = RGBColor(r, g, b)
        run.font.bold = True
    return p

def add_colored_para(doc, text, bold=False, color=None, size=11, center=False):
    p = doc.add_paragraph()
    run = p.add_run(text)
    run.bold = bold
    run.font.size = Pt(size)
    if color:
        run.font.color.rgb = RGBColor(*color)
    if center:
        p.alignment = WD_ALIGN_PARAGRAPH.CENTER
    return p

def add_shaded_para(doc, text, shade_color="DCE6F1"):
    p = doc.add_paragraph()
    pPr = p._p.get_or_add_pPr()
    shd = OxmlElement('w:shd')
    shd.set(qn('w:val'), 'clear')
    shd.set(qn('w:color'), 'auto')
    shd.set(qn('w:fill'), shade_color)
    pPr.append(shd)
    run = p.add_run(text)
    run.bold = True
    run.font.size = Pt(11)
    return p

def add_bullet(doc, text, bold_part=None):
    p = doc.add_paragraph(style='List Bullet')
    if bold_part and bold_part in text:
        idx = text.index(bold_part)
        before = text[:idx]
        after = text[idx+len(bold_part):]
        if before:
            p.add_run(before)
        r = p.add_run(bold_part)
        r.bold = True
        if after:
            p.add_run(after)
    else:
        run = p.add_run(text)
    return p

def add_divider(doc):
    p = doc.add_paragraph()
    p.alignment = WD_ALIGN_PARAGRAPH.CENTER
    run = p.add_run('─' * 70)
    run.font.color.rgb = RGBColor(0, 80, 160)
    run.font.size = Pt(9)

# ═══════════════════════════════════════════════════════════════
#  HEADER
# ═══════════════════════════════════════════════════════════════
title = doc.add_paragraph()
title.alignment = WD_ALIGN_PARAGRAPH.CENTER
r1 = title.add_run('THYROID TABLET (LEVOTHYROXINE) - PATIENT GUIDE')
r1.bold = True
r1.font.size = Pt(16)
r1.font.color.rgb = RGBColor(0, 80, 160)

subtitle = doc.add_paragraph()
subtitle.alignment = WD_ALIGN_PARAGRAPH.CENTER
r2 = subtitle.add_run('تھائیرائڈ کی دوائی - مریض کے لیے ہدایات')
r2.bold = True
r2.font.size = Pt(14)
r2.font.color.rgb = RGBColor(180, 0, 0)

add_divider(doc)

# ── IMPORTANT BOX ─────────────────────────────────────────────
doc.add_paragraph()
imp = doc.add_paragraph()
imp.alignment = WD_ALIGN_PARAGRAPH.CENTER
r_imp = imp.add_run('⚠  THIS MEDICINE MUST BE TAKEN EVERY SINGLE DAY  ⚠')
r_imp.bold = True
r_imp.font.size = Pt(12)
r_imp.font.color.rgb = RGBColor(200, 0, 0)

imp_ur = doc.add_paragraph()
imp_ur.alignment = WD_ALIGN_PARAGRAPH.CENTER
r_ur = imp_ur.add_run('یہ دوائی ہر روز لینا ضروری ہے')
r_ur.bold = True
r_ur.font.size = Pt(12)
r_ur.font.color.rgb = RGBColor(180, 0, 0)

add_divider(doc)

# ═══════════════════════════════════════════════════════════════
#  SECTION 1: HOW TO TAKE
# ═══════════════════════════════════════════════════════════════
add_colored_heading(doc, '1.  HOW TO TAKE YOUR TABLET  /  گولی کیسے لیں', level=2, r=0, g=80, b=160)

add_shaded_para(doc, 'STEP 1: Wake up in the morning  |  صبح اٹھیں', shade_color='E2EFDA')
doc.add_paragraph()

add_bullet(doc, 'Take the tablet IMMEDIATELY on waking - before doing anything else.', 'IMMEDIATELY on waking')
add_bullet(doc, 'صبح اٹھتے ہی، ناشتے سے پہلے، سب سے پہلے یہ گولی لیں')

doc.add_paragraph()
add_shaded_para(doc, 'STEP 2: Take with plain water ONLY  |  صرف سادہ پانی کے ساتھ لیں', shade_color='E2EFDA')
doc.add_paragraph()

add_bullet(doc, 'Use a FULL glass of plain water (250 ml).', 'FULL glass of plain water')
add_bullet(doc, 'صرف سادہ پانی - ایک پورا گلاس پیئیں')

doc.add_paragraph()
add_shaded_para(doc, 'STEP 3: Wait 30-60 minutes before eating or drinking', shade_color='E2EFDA')
doc.add_paragraph()

add_bullet(doc, 'Wait at least 30 to 60 minutes before breakfast, tea, or milk.', '30 to 60 minutes')
add_bullet(doc, 'گولی لینے کے بعد 30 سے 60 منٹ تک کچھ نہ کھائیں پئیں')
add_bullet(doc, 'Then eat breakfast, drink tea - after waiting.', 'after waiting')
add_bullet(doc, 'پھر ناشتہ کریں، چائے پیئیں - لیکن پہلے انتظار کریں')

add_divider(doc)

# ═══════════════════════════════════════════════════════════════
#  SECTION 2: WHAT TO AVOID
# ═══════════════════════════════════════════════════════════════
add_colored_heading(doc, '2.  WHAT TO AVOID  /  کیا نہ کریں', level=2, r=180, g=0, b=0)

table = doc.add_table(rows=1, cols=2)
table.style = 'Table Grid'
hdr = table.rows[0].cells
hdr[0].text = 'Do NOT take tablet with... / ان کے ساتھ گولی نہ لیں'
hdr[1].text = 'Why / وجہ'
hdr[0].paragraphs[0].runs[0].bold = True
hdr[1].paragraphs[0].runs[0].bold = True

rows_data = [
    ('Chai / Tea / چائے', 'Milk in tea blocks absorption / دودھ جذب ہونے سے روکتا ہے'),
    ('Milk / Doodh / دودھ', 'Calcium blocks the tablet / کیلشیم دوائی کو روکتا ہے'),
    ('Iron tablets / آئرن کی گولیاں', 'Blocks absorption completely / مکمل جذب نہیں ہوتی'),
    ('Calcium tablets / کیلشیم گولیاں', 'Blocks absorption / جذب روک دیتا ہے'),
    ('Antacids / تیزابیت کی دوائی', 'Reduces effect of tablet / اثر کم کر دیتا ہے'),
    ('Food / Khana / کھانا', 'Reduces absorption by 40% / 40% کم جذب ہوتی ہے'),
]
for eng, ur in rows_data:
    row = table.add_row().cells
    row[0].text = eng
    row[1].text = ur

doc.add_paragraph()
add_divider(doc)

# ═══════════════════════════════════════════════════════════════
#  SECTION 3: NEVER SKIP / NEVER STOP
# ═══════════════════════════════════════════════════════════════
add_colored_heading(doc, '3.  NEVER SKIP - NEVER STOP  /  گولی کبھی نہ چھوڑیں', level=2, r=180, g=0, b=0)

add_bullet(doc, 'Take this tablet EVERY DAY - including weekends and holidays.', 'EVERY DAY')
add_bullet(doc, 'یہ گولی ہر روز لیں - چھٹی ہو یا عید ہو - روز لیں')
add_bullet(doc, 'Never stop on your own - even if you feel better.', 'Never stop on your own')
add_bullet(doc, 'اپنے آپ سے بند نہ کریں - چاہے بہتر محسوس ہو')
add_bullet(doc, 'If you miss ONE dose - take it as soon as you remember (same day).', 'miss ONE dose')
add_bullet(doc, 'اگر ایک دن بھول جائیں - اسی دن یاد آتے ہی لے لیں')
add_bullet(doc, 'If you miss a full day - do NOT take double next day. Just continue normally.', 'do NOT take double')
add_bullet(doc, 'اگر پورا دن چھوٹ گیا - اگلے دن دو گولیاں نہ لیں، معمول کے مطابق چلتے رہیں')

add_divider(doc)

# ═══════════════════════════════════════════════════════════════
#  SECTION 4: WHY THIS MATTERS (5-DAY CONTEXT)
# ═══════════════════════════════════════════════════════════════
add_colored_heading(doc, '4.  WHY TIMING MATTERS  /  وقت کیوں ضروری ہے', level=2, r=0, g=100, b=0)

p_why = doc.add_paragraph()
r_why = p_why.add_run(
    'Levothyroxine has a LONG half-life of 7 days. '
    'This means it takes 6-8 WEEKS for any dose change to fully show its effect in your blood test. '
    'Taking the tablet at the wrong time or skipping doses is the most common reason TSH remains high.'
)
r_why.font.size = Pt(11)

p_why_ur = doc.add_paragraph()
r_why_ur = p_why_ur.add_run(
    'اس دوائی کا اثر 7 دن تک رہتا ہے۔ '
    'خون کے ٹیسٹ میں بہتری دیکھنے کے لیے صحیح طریقے سے 6 سے 8 ہفتے لگتے ہیں۔ '
    'غلط وقت پر لینا یا بھول جانا - یہی وجہ ہے کہ TSH بلند رہتا ہے۔'
)
r_why_ur.font.size = Pt(11)

add_divider(doc)

# ═══════════════════════════════════════════════════════════════
#  SECTION 5: DOSE CHANGE - 5 DAYS
# ═══════════════════════════════════════════════════════════════
add_colored_heading(doc, '5.  ABOUT YOUR DOSE  /  آپ کی خوراک کے بارے میں', level=2, r=0, g=80, b=160)

box = doc.add_paragraph()
r_box = box.add_run(
    'Your current dose of 100 mcg is correct for your body weight (54 kg). '
    'DO NOT change this dose on your own. '
    'After only 5 days of taking the tablet, it is TOO EARLY to judge whether the dose is working. '
    'The body needs 6-8 full weeks of correct, daily intake before a blood test can show the result.'
)
r_box.font.size = Pt(11)
r_box.bold = True
r_box.font.color.rgb = RGBColor(0, 80, 160)

box_ur = doc.add_paragraph()
r_box_ur = box_ur.add_run(
    'آپ کی موجودہ خوراک 100 mcg آپ کے وزن (54 کلو) کے مطابق درست ہے۔ '
    'خود سے خوراک تبدیل نہ کریں۔ '
    'صرف 5 دن میں یہ اندازہ نہیں لگایا جا سکتا کہ دوائی کام کر رہی ہے یا نہیں۔ '
    'خون کے ٹیسٹ میں نتیجہ دیکھنے کے لیے 6 سے 8 ہفتے درکار ہیں۔'
)
r_box_ur.font.size = Pt(11)

add_divider(doc)

# ═══════════════════════════════════════════════════════════════
#  SECTION 6: STORAGE
# ═══════════════════════════════════════════════════════════════
add_colored_heading(doc, '6.  HOW TO STORE  /  دوائی کیسے رکھیں', level=2, r=0, g=100, b=0)

add_bullet(doc, 'Keep at room temperature - away from heat and moisture.', 'room temperature')
add_bullet(doc, 'کمرے کے درجہ حرارت پر رکھیں - گرمی اور نمی سے دور')
add_bullet(doc, 'Do NOT store in bathroom or kitchen near the stove.', 'NOT store in bathroom')
add_bullet(doc, 'باتھ روم یا چولہے کے قریب نہ رکھیں')
add_bullet(doc, 'Keep in original packaging until use.', 'original packaging')
add_bullet(doc, 'استعمال تک اصلی ڈبے میں رکھیں')

add_divider(doc)

# ═══════════════════════════════════════════════════════════════
#  SECTION 7: FOLLOW-UP
# ═══════════════════════════════════════════════════════════════
add_colored_heading(doc, '7.  YOUR FOLLOW-UP PLAN  /  آپ کا فالو اپ', level=2, r=0, g=80, b=160)

fu_table = doc.add_table(rows=1, cols=3)
fu_table.style = 'Table Grid'
fh = fu_table.rows[0].cells
fh[0].text = 'When / کب'
fh[1].text = 'What / کیا'
fh[2].text = 'Why / کیوں'
for c in fh:
    c.paragraphs[0].runs[0].bold = True

fu_data = [
    ('6-8 weeks / 6-8 ہفتے بعد', 'TSH blood test / TSH خون ٹیسٹ', 'Check if dose is working / خوراک کا اثر دیکھیں'),
    ('6-8 weeks / 6-8 ہفتے بعد', 'Cholesterol test / کولیسٹرول ٹیسٹ', 'Should improve with thyroid / تھائیرائڈ ٹھیک ہونے سے بہتر ہو گا'),
    ('Every 6 months / ہر 6 ماہ', 'TSH check / TSH چیک', 'Routine monitoring / معمول کی نگرانی'),
]
for w, wh, wy in fu_data:
    row = fu_table.add_row().cells
    row[0].text = w
    row[1].text = wh
    row[2].text = wy

doc.add_paragraph()
add_divider(doc)

# ═══════════════════════════════════════════════════════════════
#  QUICK REMINDER BOX
# ═══════════════════════════════════════════════════════════════
reminder = doc.add_paragraph()
reminder.alignment = WD_ALIGN_PARAGRAPH.CENTER
r_rem = reminder.add_run(
    'REMEMBER  /  یاد رکھیں\n'
    'Wake Up  →  Take tablet with water  →  Wait 30-60 min  →  Breakfast / Tea\n'
    'اٹھیں  →  پانی سے گولی لیں  →  30-60 منٹ انتظار کریں  →  ناشتہ / چائے'
)
r_rem.bold = True
r_rem.font.size = Pt(12)
r_rem.font.color.rgb = RGBColor(0, 80, 160)

add_divider(doc)

# ── Footer note ──────────────────────────────────────────────
footer_p = doc.add_paragraph()
footer_p.alignment = WD_ALIGN_PARAGRAPH.CENTER
r_f = footer_p.add_run(
    'This handout is for educational purposes only. '
    'Always follow your doctor\'s specific instructions.\n'
    'یہ ہدایات صرف معلومات کے لیے ہیں۔ ہمیشہ اپنے ڈاکٹر کی ہدایات پر عمل کریں۔'
)
r_f.font.size = Pt(9)
r_f.italic = True
r_f.font.color.rgb = RGBColor(100, 100, 100)

# ── Save ─────────────────────────────────────────────────────
out_path = '/tmp/workspace/thyroid-handout/Levothyroxine_Patient_Handout.docx'
doc.save(out_path)
print(f'Saved: {out_path}')