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

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

# --- Helper: set paragraph spacing ---
def set_spacing(para, before=0, after=4, line=None):
    pPr = para._p.get_or_add_pPr()
    spacing = OxmlElement('w:spacing')
    spacing.set(qn('w:before'), str(before))
    spacing.set(qn('w:after'), str(after))
    if line:
        spacing.set(qn('w:line'), str(line))
        spacing.set(qn('w:lineRule'), 'auto')
    pPr.append(spacing)

# ─── TITLE ───────────────────────────────────────────────────────────────────
title = doc.add_paragraph()
title.alignment = WD_ALIGN_PARAGRAPH.CENTER
run = title.add_run('MCQ Practice Sheet')
run.bold = True
run.font.size = Pt(16)
run.font.color.rgb = RGBColor(0x1F, 0x49, 0x7D)
set_spacing(title, before=0, after=6)

subtitle = doc.add_paragraph()
subtitle.alignment = WD_ALIGN_PARAGRAPH.CENTER
run2 = subtitle.add_run('Chapter 1: Basics of Electricity')
run2.bold = True
run2.font.size = Pt(13)
run2.font.color.rgb = RGBColor(0x1F, 0x49, 0x7D)
set_spacing(subtitle, before=0, after=4)

ref = doc.add_paragraph()
ref.alignment = WD_ALIGN_PARAGRAPH.CENTER
run3 = ref.add_run('Reference: Textbook of Electrotherapy — Jagmohan Singh (Jaypee Brothers, 3rd Ed.)')
run3.italic = True
run3.font.size = Pt(10)
run3.font.color.rgb = RGBColor(0x60, 0x60, 0x60)
set_spacing(ref, before=0, after=12)

# horizontal rule via border
def add_horizontal_rule(doc):
    p = doc.add_paragraph()
    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'), '1F497D')
    pBdr.append(bottom)
    pPr.append(pBdr)
    set_spacing(p, before=0, after=8)
    return p

add_horizontal_rule(doc)

# ─── MCQ DATA ────────────────────────────────────────────────────────────────
sections_data = [
    {
        'title': 'Section A: Basic Definitions & Concepts',
        'questions': [
            {
                'q': '1. The SI unit of electric charge is:',
                'options': ['A) Ampere', 'B) Volt', 'C) Coulomb', 'D) Joule'],
                'answer': 'C'
            },
            {
                'q': '2. Electric current is defined as:',
                'options': ['A) Force per unit charge', 'B) Energy per unit time', 'C) Rate of flow of electric charge', 'D) Resistance offered to flow of electrons'],
                'answer': 'C'
            },
            {
                'q': '3. The unit of electric current is:',
                'options': ['A) Volt', 'B) Ohm', 'C) Watt', 'D) Ampere'],
                'answer': 'D'
            },
            {
                'q': '4. Which law states that V = I × R?',
                'options': ["A) Faraday's Law", "B) Ohm's Law", "C) Coulomb's Law", "D) Kirchhoff's Law"],
                'answer': 'B'
            },
            {
                'q': '5. The unit of electrical resistance is:',
                'options': ['A) Farad', 'B) Henry', 'C) Ohm', 'D) Siemen'],
                'answer': 'C'
            },
            {
                'q': '6. Impedance is the total opposition to flow of:',
                'options': ['A) Direct current only', 'B) Galvanic current only', 'C) Alternating current', 'D) Pulsed current only'],
                'answer': 'C'
            },
            {
                'q': '7. The unit of frequency is:',
                'options': ['A) Joule', 'B) Watt', 'C) Hertz', 'D) Pascal'],
                'answer': 'C'
            },
            {
                'q': '8. Electrical potential difference is measured in:',
                'options': ['A) Ampere', 'B) Ohm', 'C) Volt', 'D) Watt'],
                'answer': 'C'
            },
            {
                'q': '9. Current that flows in one direction only is called:',
                'options': ['A) Alternating current', 'B) Faradic current', 'C) Sinusoidal current', 'D) Direct (Galvanic) current'],
                'answer': 'D'
            },
            {
                'q': '10. Current that periodically reverses its direction is called:',
                'options': ['A) Galvanic current', 'B) Alternating current', 'C) Unidirectional current', 'D) Diadynamic current'],
                'answer': 'B'
            },
            {
                'q': '11. The positive electrode in electrotherapy is called:',
                'options': ['A) Cathode', 'B) Anode', 'C) Dispersive electrode', 'D) Reference electrode'],
                'answer': 'B'
            },
            {
                'q': '12. The negative electrode is called:',
                'options': ['A) Anode', 'B) Cathode', 'C) Active electrode', 'D) Ground electrode'],
                'answer': 'B'
            },
            {
                'q': '13. The unit of electrical power is:',
                'options': ['A) Volt', 'B) Joule', 'C) Watt', 'D) Coulomb'],
                'answer': 'C'
            },
            {
                'q': '14. Capacitance is measured in:',
                'options': ['A) Henry', 'B) Ohm', 'C) Farad', 'D) Tesla'],
                'answer': 'C'
            },
            {
                'q': '15. Inductance is measured in:',
                'options': ['A) Farad', 'B) Ohm', 'C) Volt', 'D) Henry'],
                'answer': 'D'
            },
            {
                'q': '16. The duty cycle is expressed as:',
                'options': ['A) Pulse amplitude / frequency', 'B) On-time × Off-time', 'C) On-time / Total cycle time × 100', 'D) Frequency / Wavelength'],
                'answer': 'C'
            },
            {
                'q': '17. The relationship between frequency and wavelength is:',
                'options': ['A) f = λ × t', 'B) λ = c / f  (where c = speed of light)', 'C) λ = f × R', 'D) f = λ / c'],
                'answer': 'B'
            },
            {
                'q': '18. The delivery of ions into tissue using direct current is called:',
                'options': ['A) Phonophoresis', 'B) Diathermy', 'C) Iontophoresis', 'D) Electrophoresis'],
                'answer': 'C'
            },
            {
                'q': '19. The number of complete cycles per second is called:',
                'options': ['A) Amplitude', 'B) Pulse width', 'C) Duty cycle', 'D) Frequency'],
                'answer': 'D'
            },
            {
                'q': '20. Pulse duration is typically measured in:',
                'options': ['A) Seconds', 'B) Milliseconds only', 'C) Microseconds (μs) or milliseconds (ms)', 'D) Hertz'],
                'answer': 'C'
            },
        ]
    },
    {
        'title': 'Section B: Frequency Categories & Modality Names',
        'questions': [
            {
                'q': '21. Low frequency currents are defined as currents with frequency:',
                'options': ['A) 1 kHz - 100 kHz', 'B) Above 100 kHz', 'C) Up to 1,000 Hz', 'D) 100 Hz - 10 kHz'],
                'answer': 'C'
            },
            {
                'q': '22. Medium frequency currents fall in the range of:',
                'options': ['A) 0 - 1,000 Hz', 'B) 1,000 Hz - 100,000 Hz', 'C) Above 1 MHz', 'D) Below 500 Hz'],
                'answer': 'B'
            },
            {
                'q': '23. High frequency currents are those with a frequency:',
                'options': ['A) 1,000 Hz - 100,000 Hz', 'B) Up to 1,000 Hz', 'C) Above 100,000 Hz (100 kHz)', 'D) Exactly 2,500 Hz'],
                'answer': 'C'
            },
            {
                'q': '24. Which of the following is a LOW frequency modality?',
                'options': ['A) Interferential Current', 'B) Shortwave Diathermy', 'C) Russian Current', 'D) TENS'],
                'answer': 'D'
            },
            {
                'q': '25. Interferential Current (IFC) belongs to which category?',
                'options': ['A) Low frequency', 'B) Medium frequency', 'C) High frequency', 'D) Radiation therapy'],
                'answer': 'B'
            },
            {
                'q': '26. Shortwave Diathermy (SWD) belongs to which frequency category?',
                'options': ['A) Low frequency', 'B) Medium frequency', 'C) High frequency', 'D) Ultrasonic'],
                'answer': 'C'
            },
            {
                'q': '27. Russian current operates at a frequency of:',
                'options': ['A) 50 Hz', 'B) 4,000 Hz', 'C) 2,500 Hz', 'D) 10,000 Hz'],
                'answer': 'C'
            },
            {
                'q': '28. Russian current is classified under:',
                'options': ['A) Low frequency', 'B) Medium frequency', 'C) High frequency', 'D) Galvanic'],
                'answer': 'B'
            },
            {
                'q': '29. Microwave Diathermy is classified as:',
                'options': ['A) Low frequency', 'B) Medium frequency', 'C) High frequency', 'D) Ultrasonic therapy'],
                'answer': 'C'
            },
            {
                'q': '30. Which of the following is NOT a low frequency modality?',
                'options': ['A) TENS', 'B) Faradic current', 'C) Galvanic current', 'D) Shortwave Diathermy'],
                'answer': 'D'
            },
            {
                'q': '31. NMES (Neuromuscular Electrical Stimulation) belongs to which category?',
                'options': ['A) High frequency', 'B) Low frequency', 'C) Medium frequency', 'D) Radiation therapy'],
                'answer': 'B'
            },
            {
                'q': '32. Diadynamic currents (Bernard currents) are classified as:',
                'options': ['A) Medium frequency', 'B) High frequency', 'C) Low frequency', 'D) Ultrasonic'],
                'answer': 'C'
            },
            {
                'q': '33. Which frequency category does premodulated current fall under?',
                'options': ['A) Low frequency', 'B) Medium frequency', 'C) High frequency', 'D) Galvanic'],
                'answer': 'B'
            },
            {
                'q': '34. FES (Functional Electrical Stimulation) is a:',
                'options': ['A) High frequency modality', 'B) Medium frequency modality', 'C) Low frequency modality', 'D) Radiation modality'],
                'answer': 'C'
            },
            {
                'q': '35. HVPC (High Voltage Pulsed Current) belongs to:',
                'options': ['A) Medium frequency', 'B) High frequency', 'C) Low frequency', 'D) Diathermy group'],
                'answer': 'C'
            },
        ]
    }
]

# ─── WRITE SECTIONS & QUESTIONS ──────────────────────────────────────────────
for sec in sections_data:
    # Section heading
    sec_heading = doc.add_paragraph()
    run = sec_heading.add_run(sec['title'])
    run.bold = True
    run.font.size = Pt(12)
    run.font.color.rgb = RGBColor(0x1F, 0x49, 0x7D)
    set_spacing(sec_heading, before=12, after=6)

    for item in sec['questions']:
        # Question
        q_para = doc.add_paragraph()
        q_run = q_para.add_run(item['q'])
        q_run.bold = True
        q_run.font.size = Pt(11)
        set_spacing(q_para, before=8, after=2)

        # Options — highlight the correct one in green
        answer_letter = item['answer']
        for opt in item['options']:
            opt_para = doc.add_paragraph(style='List Bullet')
            opt_letter = opt[0]  # 'A', 'B', 'C', 'D'
            opt_run = opt_para.add_run(opt)
            opt_run.font.size = Pt(11)
            if opt_letter == answer_letter:
                opt_run.bold = True
                opt_run.font.color.rgb = RGBColor(0x00, 0x70, 0x00)  # dark green
            set_spacing(opt_para, before=0, after=2)

        # Small blank gap after each question
        gap = doc.add_paragraph()
        set_spacing(gap, before=0, after=2)

# ─── ANSWER KEY PAGE ─────────────────────────────────────────────────────────
doc.add_page_break()

ak_title = doc.add_paragraph()
ak_title.alignment = WD_ALIGN_PARAGRAPH.CENTER
run = ak_title.add_run('Answer Key')
run.bold = True
run.font.size = Pt(14)
run.font.color.rgb = RGBColor(0x1F, 0x49, 0x7D)
set_spacing(ak_title, before=0, after=10)

# Build answer key as a table: 5 columns
all_questions = []
for sec in sections_data:
    all_questions.extend(sec['questions'])

# Extract Q number and answer
answers = []
for i, item in enumerate(all_questions):
    q_num = i + 1
    ans = item['answer']
    answers.append((q_num, ans))

# Table with 5 cols
table = doc.add_table(rows=1, cols=5)
table.style = 'Table Grid'

# Header row
hdr_cells = table.rows[0].cells
for i, hdr in enumerate(['Q No.', 'Answer', 'Q No.', 'Answer', 'Q No.']):
    hdr_cells[i].text = hdr
    for run in hdr_cells[i].paragraphs[0].runs:
        run.bold = True
        run.font.size = Pt(10)

# Fill answers in groups of 7 rows (35 questions / 5 cols per group)
# Layout: col 0 = Q1-Q7 numbers, col1 = answers, col2 = Q8-14, col3 = answers, col4 = Q15-21...
# Simpler: two Q+A column pairs + a leftover col
# Let's do a clean 3-pair layout: Q1-12 | Q13-24 | Q25-35
col_groups = [answers[0:12], answers[12:24], answers[24:35]]
max_rows = max(len(g) for g in col_groups)

for row_i in range(max_rows):
    row_cells = table.add_row().cells
    col_offset = 0
    for grp_i, grp in enumerate(col_groups):
        if row_i < len(grp):
            qn_val, ans_val = grp[row_i]
            row_cells[col_offset].text = f'Q{qn_val}'
            row_cells[col_offset + 1].text = ans_val
        col_offset += 2
        if col_offset >= 5:
            break

# Style table font
for row in table.rows:
    for cell in row.cells:
        for para in cell.paragraphs:
            for run in para.runs:
                run.font.size = Pt(10)
                run.font.name = 'Arial'

# ─── FOOTER NOTE ─────────────────────────────────────────────────────────────
footer_note = doc.add_paragraph()
footer_note.alignment = WD_ALIGN_PARAGRAPH.CENTER
fn_run = footer_note.add_run(
    '\nNote: Correct answers are shown in bold green in the question section above.\n'
    'Reference: Textbook of Electrotherapy — Jagmohan Singh, Jaypee Brothers Publishers, 3rd Edition.'
)
fn_run.italic = True
fn_run.font.size = Pt(9)
fn_run.font.color.rgb = RGBColor(0x60, 0x60, 0x60)
set_spacing(footer_note, before=16, after=0)

# ─── SAVE ────────────────────────────────────────────────────────────────────
out_path = '/tmp/workspace/electrotherapy-mcqs/Electrotherapy_Ch1_MCQs.docx'
doc.save(out_path)
print(f'Saved: {out_path}')

Successfully replaced 1 instance(s) of the string in '/tmp/workspace/electrotherapy-mcqs/create_mcq_doc.py'