from reportlab.lib.pagesizes import A4
from reportlab.lib import colors
from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
from reportlab.lib.units import cm
from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle, HRFlowable
from reportlab.lib.enums import TA_LEFT, TA_CENTER

doc = SimpleDocTemplate(
    "/tmp/workspace/pharma-pyq/Pharma_PYQ_Unit_Wise.pdf",
    pagesize=A4,
    rightMargin=2*cm,
    leftMargin=2*cm,
    topMargin=2*cm,
    bottomMargin=2*cm,
)

styles = getSampleStyleSheet()

# Custom styles
title_style = ParagraphStyle(
    'Title', parent=styles['Title'],
    fontSize=20, textColor=colors.HexColor('#1a237e'),
    spaceAfter=6, spaceBefore=6, alignment=TA_CENTER,
    fontName='Helvetica-Bold'
)

subtitle_style = ParagraphStyle(
    'Subtitle', parent=styles['Normal'],
    fontSize=10, textColor=colors.HexColor('#555555'),
    spaceAfter=12, alignment=TA_CENTER
)

unit_heading_style = ParagraphStyle(
    'UnitHeading',
    fontSize=14, textColor=colors.white,
    spaceAfter=4, spaceBefore=10,
    fontName='Helvetica-Bold', alignment=TA_LEFT,
    leftIndent=6
)

long_q_style = ParagraphStyle(
    'LongQ',
    fontSize=11, textColor=colors.HexColor('#b71c1c'),
    spaceAfter=4, spaceBefore=2,
    fontName='Helvetica-Bold', leftIndent=10
)

five_mark_style = ParagraphStyle(
    'FiveMark',
    fontSize=11, textColor=colors.HexColor('#0d47a1'),
    spaceAfter=4, spaceBefore=2,
    fontName='Helvetica-Bold', leftIndent=10
)

short_q_style = ParagraphStyle(
    'ShortQ',
    fontSize=10, textColor=colors.HexColor('#1b5e20'),
    spaceAfter=3, spaceBefore=2,
    fontName='Helvetica', leftIndent=10
)

black_q_style = ParagraphStyle(
    'BlackQ',
    fontSize=10, textColor=colors.black,
    spaceAfter=3, spaceBefore=2,
    fontName='Helvetica', leftIndent=10
)

legend_style = ParagraphStyle(
    'Legend',
    fontSize=9, textColor=colors.HexColor('#333333'),
    spaceAfter=3, spaceBefore=2,
    fontName='Helvetica', leftIndent=4
)

# -------------------------------------------------------
# DATA STRUCTURE
# type: 'long' (red) | '5mark' (blue) | 'short' (green) | 'black'
# -------------------------------------------------------

units = [
    {
        "title": "UNIT I – General Pharmacology & Pharmacokinetics",
        "color": "#1a237e",
        "questions": [
            # Red (Long)
            ("long",  "Antipyretic (Paracetamol) – MOA, uses, adverse effects"),
            ("long",  "Antitubercular (LA / COMA / sidecarms) – MOA, types, classification"),
            ("long",  "NSAIDs – Classify, give pharmacology of Aspirin / Ciprofloxacin"),
            ("long",  "Antimicrobial drugs – MOA, classification, B-Blockers, Penicillin, dose"),
            ("long",  "Anti-coagulant (Cardiovascular / Heparin Vs Warfarin)"),
            ("long",  "Oral antidiabetic (Sulfonyl urea)"),
            ("long",  "Antihypertensives – Classification, pharmacology (Antihypertensives)"),
            ("long",  "OTC adrenergic / anticholinergic drugs (Atropine)"),
            # Blue (5-mark)
            ("5mark", "Bioavailability"),
            ("5mark", "Nomenclature of drug"),
            ("5mark", "Pharmacodynamics"),
            ("5mark", "Pharmacogenetics"),
            ("5mark", "Adverse drug reaction (ADR)"),
            ("5mark", "Adverse drug effects"),
            ("5mark", "G-Protein coupled receptor"),
            ("5mark", "Zero-order kinetics – List 4"),
            ("5mark", "Types of receptor"),
            # Short/Green
            ("short", "Pharmacokinetics – definition & phases"),
            ("short", "Drug Antagonism"),
            # Black
            ("black", "Teratogenic drugs"),
        ]
    },
    {
        "title": "UNIT II – Drugs Acting on ANS / CVS / Respiratory",
        "color": "#1b5e20",
        "questions": [
            # Long (Red)
            ("long",  "Beta Blockers – MOA, uses, C/I in Asthma"),
            ("long",  "Calcium Channel Blockers in Angina"),
            ("long",  "Glyceryl Trinitrate / Nitrates in Angina"),
            ("long",  "Loop Diuretics"),
            ("long",  "Antiarrhythmics – classification, pharmacology"),
            ("long",  "Antiplatelet drugs"),
            ("long",  "Glucocorticoids – pharmacology, A/E & C/I of Corticosteroids"),
            ("long",  "Use of Prostaglandins"),
            ("long",  "HAART (HIV treatment)"),
            # Blue (5-mark)
            ("5mark", "Furosemide in oedema"),
            ("5mark", "Low Molecular Weight Heparin (LMWH)"),
            ("5mark", "Adrenaline in Anaphylactic Shock"),
            ("5mark", "Salbutamol in Asthma"),
            ("5mark", "Sodium Cromoglycate (Asthma)"),
            ("5mark", "Terbutaline (Asthma)"),
            ("5mark", "Management of Status Asthma"),
            ("5mark", "Pantoprazole in Peptic Ulcers"),
            ("5mark", "Omeprazole (Peptic Ulcers)"),
            ("5mark", "Amlodipine with Atenolol"),
            ("5mark", "Spironolactone with Furosemide (Treatment of Oedema)"),
            # Short / Green
            ("short", "DMARDS"),
            ("short", "Inhalation Steroids"),
            ("short", "SSRIs"),
            ("short", "ACE Inhibitor"),
            ("short", "Furosemide (oedema)"),
            ("short", "Nitrates (angina pectoris)"),
            # Black
            ("black", "Aminophylline (uses)"),
            ("black", "Amitriptyline (depression)"),
        ]
    },
    {
        "title": "UNIT III – Antimicrobials & Antifungals",
        "color": "#4a148c",
        "questions": [
            # Long (Red)
            ("long",  "Antibiotics – Classify, MOA, classification of antimicrobials"),
            ("long",  "Beta-Lactam antibiotics – MOA, classification"),
            ("long",  "Tetracycline – pharmacology, uses"),
            ("long",  "Fluoride / Fluoride-containing drugs"),
            ("long",  "Isoniazid (Antitubercular) – MOA, uses, ADR"),
            ("long",  "Rifampicin (Tuberculosis)"),
            ("long",  "Antiseptic & Disinfectants – classification"),
            # Blue (5-mark)
            ("5mark", "Antiseptic – EDTA in Dentistry"),
            ("5mark", "Chlorhexidine Mouthwash"),
            ("5mark", "Antifungal drugs"),
            ("5mark", "Antiviral – Acyclovir"),
            ("5mark", "Ketoconazole (oral candidiasis)"),
            ("5mark", "Fluconazole (oral candidiasis)"),
            ("5mark", "Metronidazole (amoebiasis)"),
            ("5mark", "Antimicrobial drugs – general"),
            # Short / Green
            ("short", "Antiseptics"),
            ("short", "Styptics"),
            ("short", "Mouth Washes"),
            ("short", "Tooth Powder"),
            ("short", "Bleaching Agent"),
            ("short", "Mummifying Agent"),
            ("short", "Dentifrices"),
            ("short", "DOXIN / Doxycycline in CHF"),
            # Black
            ("black", "Ciprofloxacin – uses"),
            ("black", "New Antiepileptic drugs"),
        ]
    },
    {
        "title": "UNIT IV – CNS Drugs, Analgesics & Anaesthetics",
        "color": "#bf360c",
        "questions": [
            # Long (Red)
            ("long",  "Morphine – pharmacology / Dental Analgesia"),
            ("long",  "Opioid analgesics – classification, MOA"),
            ("long",  "Opioid antagonists"),
            ("long",  "NSAIDs as analgesics – Diclofenac (toothache)"),
            ("long",  "Pre-anaesthetic medications"),
            ("long",  "Lidocaine / Lignocaine – pharmacology, local anaesthesia"),
            ("long",  "Lidocaine with Adrenaline"),
            ("long",  "Spinal anaesthesia"),
            ("long",  "Antiepileptics – Phenytoin (AE), classification"),
            ("long",  "OP Poisoning treatment – Oximes"),
            # Blue (5-mark)
            ("5mark", "Ketorolac / Ibuprofen"),
            ("5mark", "Metoclopramide"),
            ("5mark", "Promethazine"),
            ("5mark", "Non-depolarising neuromuscular blockers"),
            ("5mark", "IV Route of drug administration"),
            ("5mark", "Transdermal drug delivery"),
            ("5mark", "Betamethasone – Dental"),
            ("5mark", "Alprazolam (anxiety)"),
            ("5mark", "Antianxiety drugs"),
            ("5mark", "Antihistamines / Levocetirizine"),
            # Short / Green
            ("short", "Obtundents"),
            ("short", "Anti-emetic"),
            ("short", "Disulfiram"),
            ("short", "Lithium in Manic Patient"),
            ("short", "Ethanol in Alcohol Poisoning"),
            ("short", "Ondansetron in Vomit"),
            # Black
            ("black", "Opioid (Dental Analgesia)"),
            ("black", "Physostigmine (Belladonna Poisoning)"),
            ("black", "Cinnarizine (Vertigo)"),
        ]
    },
    {
        "title": "UNIT V – Endocrine, Antidiabetics & Miscellaneous",
        "color": "#006064",
        "questions": [
            # Long (Red)
            ("long",  "Insulin analogues – classification, uses"),
            ("long",  "Oral Hypoglycaemics (OHA/DM) – Metformin (DM), Glicazide"),
            ("long",  "Antithyroid drugs – Propylthiouracil in Graves disease / Thyrotoxicosis"),
            ("long",  "Corticosteroids – Uses, A/E & C/I of Corticosteroids (Dental)"),
            ("long",  "Glucocorticoids – pharmacology"),
            ("long",  "OCPs (Oral Contraceptive Pills)"),
            ("long",  "Bisphosphonates (Osteoporosis)"),
            ("long",  "Chelating agents"),
            # Blue (5-mark)
            ("5mark", "Hydrochlorothiazide (Hypertension)"),
            ("5mark", "Methotrexate (Anticancer)"),
            ("5mark", "Tamoxifen (Prostate cancer)"),
            ("5mark", "PPI – Dental"),
            ("5mark", "Insulin in Type II DM"),
            ("5mark", "Ranitidine in Peptic Ulcers"),
            ("5mark", "Drug treatment in Recurrent Apthous Ulcer"),
            ("5mark", "Chelating agents / EDTA"),
            ("5mark", "Pharmacovigilance"),
            # Short / Green
            ("short", "Disclosing agents"),
            ("short", "Styptics (dental)"),
            ("short", "Allopurinol (Gout) / Colchicine"),
            ("short", "Chronic Gout"),
            ("short", "Levodopa in Parkinson's"),
            ("short", "COMT inhibitor in Parkinsonism"),
            ("short", "Clomiphene citrate in Infertility"),
            ("short", "Astringents"),
            ("short", "Gonadal toxicity & Cytotoxic drugs"),
            # Black
            ("black", "Radioactive Iodine (use)"),
            ("black", "Propranolol in Thyrotoxicosis"),
            ("black", "Cimetidine – drug interaction"),
            ("black", "Nitrates in Angina (drug interaction)"),
            ("black", "NSAIDs as analgesics – note"),
            ("black", "Cetirizine as antihistamine"),
            ("black", "Beta Blocker (Dental)"),
        ]
    },
]

# -------------------------------------------------------
# BUILD PDF
# -------------------------------------------------------
story = []

# Title
story.append(Spacer(1, 0.3*cm))
story.append(Paragraph("PHARMACOLOGY – PREVIOUS YEAR QUESTIONS", title_style))
story.append(Paragraph("Unit-Wise Organized | BDS / MDS Pattern", subtitle_style))

# Legend box
legend_data = [
    [Paragraph("<font color='#b71c1c'><b>■ RED = Long Question (10 marks)</b></font>", legend_style),
     Paragraph("<font color='#0d47a1'><b>■ BLUE = 5-Mark Question</b></font>", legend_style),
     Paragraph("<font color='#1b5e20'>■ GREEN = Short Note (2-3 marks)</font>", legend_style),
     Paragraph("■ BLACK = Short Answer", legend_style)],
]
legend_table = Table(legend_data, colWidths=[4.2*cm, 4.2*cm, 4.2*cm, 4.2*cm])
legend_table.setStyle(TableStyle([
    ('BACKGROUND', (0,0), (-1,-1), colors.HexColor('#f5f5f5')),
    ('BOX', (0,0), (-1,-1), 0.5, colors.grey),
    ('INNERGRID', (0,0), (-1,-1), 0.25, colors.grey),
    ('VALIGN', (0,0), (-1,-1), 'MIDDLE'),
    ('TOPPADDING', (0,0), (-1,-1), 5),
    ('BOTTOMPADDING', (0,0), (-1,-1), 5),
]))
story.append(legend_table)
story.append(Spacer(1, 0.5*cm))

for unit in units:
    # Unit header bar
    unit_color = colors.HexColor(unit["color"])
    header_data = [[Paragraph(unit["title"], unit_heading_style)]]
    header_table = Table(header_data, colWidths=[17*cm])
    header_table.setStyle(TableStyle([
        ('BACKGROUND', (0,0), (-1,-1), unit_color),
        ('BOX', (0,0), (-1,-1), 0, colors.white),
        ('TOPPADDING', (0,0), (-1,-1), 7),
        ('BOTTOMPADDING', (0,0), (-1,-1), 7),
        ('LEFTPADDING', (0,0), (-1,-1), 8),
    ]))
    story.append(header_table)
    story.append(Spacer(1, 0.15*cm))

    # Separate by type
    long_qs   = [(q, t) for t, q in unit["questions"] if t == "long"]
    five_qs   = [(q, t) for t, q in unit["questions"] if t == "5mark"]
    short_qs  = [(q, t) for t, q in unit["questions"] if t == "short"]
    black_qs  = [(q, t) for t, q in unit["questions"] if t == "black"]

    if long_qs:
        story.append(Paragraph(
            "<font color='#b71c1c'><b>● LONG QUESTIONS (10 Marks)</b></font>",
            ParagraphStyle('sec', fontSize=10, leftIndent=4, spaceAfter=3, spaceBefore=4)
        ))
        for i, (q, _) in enumerate(long_qs, 1):
            story.append(Paragraph(f"<b>{i}. {q}</b>", long_q_style))
        story.append(Spacer(1, 0.1*cm))

    if five_qs:
        story.append(Paragraph(
            "<font color='#0d47a1'><b>● 5-MARK QUESTIONS</b></font>",
            ParagraphStyle('sec', fontSize=10, leftIndent=4, spaceAfter=3, spaceBefore=4)
        ))
        for i, (q, _) in enumerate(five_qs, 1):
            story.append(Paragraph(f"<b>{i}. {q}</b>", five_mark_style))
        story.append(Spacer(1, 0.1*cm))

    if short_qs:
        story.append(Paragraph(
            "<font color='#1b5e20'>● SHORT NOTES / GREEN (2-3 Marks)</font>",
            ParagraphStyle('sec', fontSize=10, leftIndent=4, spaceAfter=3, spaceBefore=4)
        ))
        for i, (q, _) in enumerate(short_qs, 1):
            story.append(Paragraph(f"{i}. {q}", short_q_style))
        story.append(Spacer(1, 0.1*cm))

    if black_qs:
        story.append(Paragraph(
            "● SHORT ANSWER / BLACK",
            ParagraphStyle('sec', fontSize=10, leftIndent=4, spaceAfter=3, spaceBefore=4)
        ))
        for i, (q, _) in enumerate(black_qs, 1):
            story.append(Paragraph(f"{i}. {q}", black_q_style))

    story.append(HRFlowable(width="100%", thickness=0.5, color=colors.HexColor('#cccccc'), spaceAfter=8, spaceBefore=6))

doc.build(story)
print("PDF created successfully!")