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.top_margin = Inches(0.8)
section.bottom_margin = Inches(0.8)
section.left_margin = Inches(1.0)
section.right_margin = Inches(1.0)

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

# Helper: add coloured heading
def add_main_title(doc, text):
    p = doc.add_paragraph()
    p.alignment = WD_ALIGN_PARAGRAPH.CENTER
    run = p.add_run(text)
    run.bold = True
    run.font.size = Pt(16)
    run.font.color.rgb = RGBColor(0x1A, 0x53, 0x76)  # dark blue
    run.font.name = 'Arial'

def add_subtitle(doc, text):
    p = doc.add_paragraph()
    p.alignment = WD_ALIGN_PARAGRAPH.CENTER
    run = p.add_run(text)
    run.font.size = Pt(11)
    run.font.color.rgb = RGBColor(0x55, 0x55, 0x55)
    run.font.name = 'Arial'
    run.italic = True

def add_section_heading(doc, text, color=RGBColor(0x1A, 0x53, 0x76)):
    p = doc.add_paragraph()
    run = p.add_run(text)
    run.bold = True
    run.font.size = Pt(12)
    run.font.color.rgb = color
    run.font.name = 'Arial'
    # bottom border
    pPr = p._p.get_or_add_pPr()
    pBdr = OxmlElement('w:pBdr')
    bottom = OxmlElement('w:bottom')
    bottom.set(qn('w:val'), 'single')
    bottom.set(qn('w:sz'), '6')
    bottom.set(qn('w:space'), '1')
    bottom.set(qn('w:color'), '1A5376')
    pBdr.append(bottom)
    pPr.append(pBdr)

def add_bullet(doc, text, bold_part=None):
    p = doc.add_paragraph(style='List Bullet')
    p.paragraph_format.left_indent = Inches(0.3)
    p.paragraph_format.space_after = Pt(2)
    if bold_part and text.startswith(bold_part):
        rest = text[len(bold_part):]
        run1 = p.add_run(bold_part)
        run1.bold = True
        run1.font.name = 'Arial'
        run1.font.size = Pt(11)
        run2 = p.add_run(rest)
        run2.font.name = 'Arial'
        run2.font.size = Pt(11)
    else:
        run = p.add_run(text)
        run.font.name = 'Arial'
        run.font.size = Pt(11)

def add_body(doc, text):
    p = doc.add_paragraph()
    p.paragraph_format.space_after = Pt(3)
    run = p.add_run(text)
    run.font.name = 'Arial'
    run.font.size = Pt(11)

def add_note(doc, text):
    p = doc.add_paragraph()
    run = p.add_run(text)
    run.font.name = 'Arial'
    run.font.size = Pt(10)
    run.italic = True
    run.font.color.rgb = RGBColor(0x80, 0x00, 0x00)

def add_table_row(table, col1, col2, col3, header=False):
    row = table.add_row()
    cells = row.cells
    for i, text in enumerate([col1, col2, col3]):
        cells[i].text = text
        for run in cells[i].paragraphs[0].runs:
            run.font.name = 'Arial'
            run.font.size = Pt(10)
            if header:
                run.bold = True
                run.font.color.rgb = RGBColor(0xFF, 0xFF, 0xFF)
    if header:
        for cell in cells:
            tc = cell._tc
            tcPr = tc.get_or_add_tcPr()
            shd = OxmlElement('w:shd')
            shd.set(qn('w:val'), 'clear')
            shd.set(qn('w:color'), 'auto')
            shd.set(qn('w:fill'), '1A5376')
            tcPr.append(shd)
    return row

# ============================================================
# TITLE
# ============================================================
add_main_title(doc, 'FMT MBBS 3rd Year — 5-Mark Revision Summary')
add_subtitle(doc, 'K. Narayana Reddy | Essentials of Forensic Medicine & Toxicology')
doc.add_paragraph()

# ============================================================
# Q1: CAUSES OF MALE INFERTILITY
# ============================================================
add_section_heading(doc, 'Q1. Causes of Male Infertility & Methods of Overcoming Each Cause')
doc.add_paragraph()

add_note(doc, 'Definition: Sterility (Infertility) = inability of the male to beget children. ~10-15% of all married couples are involuntarily sterile.')
doc.add_paragraph()

# Table
table = doc.add_table(rows=0, cols=3)
table.style = 'Table Grid'
table.autofit = True

# Set column widths
from docx.shared import Cm
table.columns[0].width = Cm(0.8)
table.columns[1].width = Cm(7.5)
table.columns[2].width = Cm(7.5)

add_table_row(table, '#', 'Cause', 'Method of Overcoming', header=True)
add_table_row(table, '1', 'Developmental defects — hypospadias, epispadias, cryptorchidism, absent testis', 'Surgical correction; Orchidopexy; AIH (Artificial Insemination by Husband)')
add_table_row(table, '2', 'Obstruction of vas deferens / epididymis — due to gonorrhoea, mumps orchitis, trauma', 'Microsurgical repair (vasovasostomy / epididymovasostomy); AID if irreparable')
add_table_row(table, '3', 'Endocrine / Hormonal — testosterone deficiency, hypothyroidism, hyperprolactinaemia, LH/FSH abnormality', 'Hormone replacement therapy; Gonadotropin treatment')
add_table_row(table, '4', 'General / Systemic diseases — diabetes (neuropathy), lead exposure (occupational), radiation, syphilis (tertiary)', 'Treat underlying disease; Remove toxic exposure; AID if permanently damaged')
add_table_row(table, '5', 'Vasculogenic — arteriosclerosis causing poor arterial inflow; venous leakage from penis', 'Sildenafil (50 mg); Microvascular surgery; Ligation of deep dorsal vein of penis')
add_table_row(table, '6', 'Psychological — anxiety, fear, depression, guilt, dislike of partner', 'Psychotherapy / Sex counselling')
add_table_row(table, '7', 'Drug-induced — antihypertensives, opiates, psychotropics, tranquillisers', 'Stop / substitute offending drug')

# Style alternate rows
for i, row in enumerate(table.rows[1:], 1):
    if i % 2 == 0:
        for cell in row.cells:
            tc = cell._tc
            tcPr = tc.get_or_add_tcPr()
            shd = OxmlElement('w:shd')
            shd.set(qn('w:val'), 'clear')
            shd.set(qn('w:color'), 'auto')
            shd.set(qn('w:fill'), 'D6E4F0')
            tcPr.append(shd)

doc.add_paragraph()
add_note(doc, '★ General Rule: When husband is permanently sterile from ANY cause → AID (Artificial Insemination by Donor) or IVF / Test Tube Baby.')
doc.add_paragraph()

# ============================================================
# Q2: ARTIFICIAL INSEMINATION
# ============================================================
add_section_heading(doc, 'Q2. Artificial Insemination (AI) — Key Points')
doc.add_paragraph()

add_body(doc, 'Definition: Deposition of semen into the female genital tract by artificial means without sexual intercourse.')
doc.add_paragraph()

p = doc.add_paragraph()
r = p.add_run('Types:')
r.bold = True
r.font.name = 'Arial'
r.font.size = Pt(11)

add_bullet(doc, 'AIH (by Husband): Used when husband has ejaculatory dysfunction, hypospadias/epispadias, or low sperm count with viable sperm.')
add_bullet(doc, 'AID (by Donor): Used when husband is sterile, impotent, has hereditary disease, or Rh incompatibility.')
doc.add_paragraph()

p = doc.add_paragraph()
r = p.add_run('Indications for AID:')
r.bold = True
r.font.name = 'Arial'
r.font.size = Pt(11)

add_bullet(doc, 'Husband is impotent')
add_bullet(doc, 'Husband cannot deposit semen due to hypospadias/epispadias')
add_bullet(doc, 'Husband is sterile')
add_bullet(doc, 'Rh incompatibility between husband and wife')
add_bullet(doc, 'Husband has hereditary disease')
doc.add_paragraph()

add_note(doc, 'Success rate: 30-40% pregnancies within 3-4 months of treatment.')
doc.add_paragraph()

# ============================================================
# Q3: TEMPORARY IMPOTENCE IN MALE
# ============================================================
add_section_heading(doc, 'Q3. Causes of Temporary Impotence in Male')
doc.add_paragraph()

add_bullet(doc, 'First-night / honeymoon impotence — fear or anxiety in bridegroom (most common)')
add_bullet(doc, 'Anxiety, guilt, depression, excessive passion')
add_bullet(doc, 'Dislike of partner / disgust of sexual act')
add_bullet(doc, 'Sexual overindulgence')
add_bullet(doc, '"Quoad hoc" — impotent with one specific woman but not others')
add_note(doc, 'Psychological impotence is usually temporary and resolves with counselling.')
doc.add_paragraph()

# ============================================================
# Q4: STERILIZATION
# ============================================================
add_section_heading(doc, 'Q4. Methods of Sterilization')
doc.add_paragraph()

add_body(doc, 'Definition: A procedure to make a person sterile without interfering with potency.')
doc.add_paragraph()

p = doc.add_paragraph()
r = p.add_run('Permanent Methods:')
r.bold = True
r.font.name = 'Arial'
r.font.size = Pt(11)
add_bullet(doc, 'Vasectomy (male)')
add_bullet(doc, 'Tubectomy (female)')
add_bullet(doc, 'Deep X-ray exposure (both sexes)')

doc.add_paragraph()
p = doc.add_paragraph()
r = p.add_run('Temporary Methods:')
r.bold = True
r.font.name = 'Arial'
r.font.size = Pt(11)
add_bullet(doc, 'Coitus interruptus')
add_bullet(doc, 'Loop (IUCD)')
add_bullet(doc, 'Oral hormonal pills')
add_bullet(doc, 'Foam tablets')
add_bullet(doc, 'Diaphragm, spermicidal jellies, condoms')
doc.add_paragraph()

p = doc.add_paragraph()
r = p.add_run('Types by indication:')
r.bold = True
r.font.name = 'Arial'
r.font.size = Pt(11)
add_bullet(doc, 'Therapeutic — to prevent danger to life/health of woman in future pregnancy')
add_bullet(doc, 'Eugenic — to prevent transmission of hereditary defects')
add_bullet(doc, 'Contraceptive — family planning / family limitation')
doc.add_paragraph()

# ============================================================
# Q5: ART
# ============================================================
add_section_heading(doc, 'Q5. Assisted Reproductive Techniques (ART) — Quick List')
doc.add_paragraph()

art_items = [
    ('AIH', 'Artificial Insemination by Husband — husband\'s sperm + wife\'s uterus'),
    ('AID', 'Artificial Insemination by Donor — donor sperm'),
    ('IVF', 'In Vitro Fertilisation — "Test Tube Baby" — fertilisation outside body, embryo transferred to uterus'),
    ('ICSI', 'Intracytoplasmic Sperm Injection — single sperm injected into egg; for severe oligozoospermia'),
    ('Surrogate Mother', 'Another woman carries the baby when wife cannot bear pregnancy'),
]
for abbr, desc in art_items:
    p = doc.add_paragraph(style='List Bullet')
    p.paragraph_format.left_indent = Inches(0.3)
    p.paragraph_format.space_after = Pt(2)
    r1 = p.add_run(abbr + ': ')
    r1.bold = True
    r1.font.name = 'Arial'
    r1.font.size = Pt(11)
    r2 = p.add_run(desc)
    r2.font.name = 'Arial'
    r2.font.size = Pt(11)

doc.add_paragraph()

# ============================================================
# FOOTER NOTE
# ============================================================
p = doc.add_paragraph()
p.alignment = WD_ALIGN_PARAGRAPH.CENTER
run = p.add_run('Source: The Essentials of Forensic Medicine & Toxicology, 36th Edition — K. Narayana Reddy | Chapter 15: Impotence & Sterility')
run.font.size = Pt(9)
run.italic = True
run.font.color.rgb = RGBColor(0x88, 0x88, 0x88)
run.font.name = 'Arial'

# Save
out_path = '/tmp/workspace/fmt-revision/FMT_Male_Infertility_5Mark_Revision.docx'
doc.save(out_path)
print(f'Saved: {out_path}')