from reportlab.lib.pagesizes import A4
from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
from reportlab.lib.units import cm
from reportlab.lib import colors
from reportlab.platypus import (
    SimpleDocTemplate, Paragraph, Spacer, Image, Table, TableStyle,
    HRFlowable, PageBreak, KeepTogether
)
from reportlab.lib.enums import TA_CENTER, TA_LEFT, TA_JUSTIFY
from reportlab.pdfbase import pdfmetrics
from reportlab.pdfbase.ttfonts import TTFont
import os

OUTPUT = "/tmp/workspace/sticky-notes-pdf/Urogenital_Sticky_Notes.pdf"

doc = SimpleDocTemplate(
    OUTPUT,
    pagesize=A4,
    rightMargin=2*cm, leftMargin=2*cm,
    topMargin=2*cm, bottomMargin=2*cm
)

W, H = A4
CONTENT_W = W - 4*cm

styles = getSampleStyleSheet()

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

subtitle_style = ParagraphStyle(
    'Subtitle',
    parent=styles['Normal'],
    fontSize=11,
    textColor=colors.HexColor('#5c6bc0'),
    spaceAfter=20,
    fontName='Helvetica',
    alignment=TA_CENTER
)

section_heading = ParagraphStyle(
    'SectionHeading',
    parent=styles['Heading1'],
    fontSize=15,
    textColor=colors.white,
    backColor=colors.HexColor('#c62828'),
    spaceBefore=16,
    spaceAfter=8,
    fontName='Helvetica-Bold',
    leftIndent=8,
    rightIndent=8,
    borderPad=6
)

sub_heading = ParagraphStyle(
    'SubHeading',
    parent=styles['Heading2'],
    fontSize=12,
    textColor=colors.HexColor('#b71c1c'),
    spaceBefore=10,
    spaceAfter=4,
    fontName='Helvetica-Bold',
)

body_style = ParagraphStyle(
    'Body',
    parent=styles['Normal'],
    fontSize=10,
    textColor=colors.HexColor('#212121'),
    spaceAfter=4,
    leading=15,
    fontName='Helvetica',
    leftIndent=12
)

bullet_style = ParagraphStyle(
    'Bullet',
    parent=styles['Normal'],
    fontSize=10,
    textColor=colors.HexColor('#212121'),
    spaceAfter=3,
    leading=14,
    fontName='Helvetica',
    leftIndent=24,
    bulletIndent=12
)

arrow_style = ParagraphStyle(
    'Arrow',
    parent=styles['Normal'],
    fontSize=10,
    textColor=colors.HexColor('#b71c1c'),
    spaceAfter=2,
    fontName='Helvetica-Bold',
    leftIndent=20,
    alignment=TA_CENTER
)

note_caption = ParagraphStyle(
    'NoteCaption',
    parent=styles['Normal'],
    fontSize=8,
    textColor=colors.HexColor('#616161'),
    spaceAfter=6,
    fontName='Helvetica-Oblique',
    alignment=TA_CENTER
)

image_section_heading = ParagraphStyle(
    'ImageSectionHeading',
    parent=styles['Normal'],
    fontSize=13,
    textColor=colors.white,
    backColor=colors.HexColor('#1565c0'),
    spaceBefore=16,
    spaceAfter=8,
    fontName='Helvetica-Bold',
    leftIndent=8,
    borderPad=5
)

def arrow(text="↓"):
    return Paragraph(text, arrow_style)

def body(text):
    return Paragraph(text, body_style)

def bullet(text):
    return Paragraph(f"• {text}", bullet_style)

def spacer(h=0.3):
    return Spacer(1, h*cm)

def hr():
    return HRFlowable(width="100%", thickness=0.5, color=colors.HexColor('#bdbdbd'), spaceAfter=6, spaceBefore=6)

# ---- Build content ----
story = []

# TITLE PAGE BLOCK
story.append(spacer(1))
story.append(Paragraph("Urogenital System – Embryology", title_style))
story.append(Paragraph("Visual Sticky Notes Reference Guide", subtitle_style))
story.append(HRFlowable(width="80%", thickness=2, color=colors.HexColor('#c62828'), spaceAfter=6, spaceBefore=2))
story.append(spacer(0.5))
story.append(Paragraph(
    "This document compiles handwritten sticky-note summaries from an embryology textbook, "
    "covering the development of female genital ducts, the vagina, the ovary, and remnants of genital ducts. "
    "Original visual notes are reproduced at the end.",
    ParagraphStyle('Intro', parent=styles['Normal'], fontSize=10, textColor=colors.HexColor('#424242'),
                   leading=16, alignment=TA_JUSTIFY, spaceAfter=16)
))
story.append(hr())
story.append(spacer(0.5))

# ===== SECTION 1: Development of Female Genital Ducts =====
story.append(Paragraph("1. Development of Female Genital Ducts", section_heading))
story.append(spacer(0.2))

steps_fgd = [
    ("Hormonal trigger", "Absence of testosterone and AMH"),
    ("Duct changes", "Mesonephric duct regresses; paramesonephric ducts develop"),
    ("3 parts of each duct",
     "(i) Cranial vertical portion – opens into peritoneal cavity\n"
     "(ii) Horizontal part – crosses mesonephric duct\n"
     "(iii) Caudal vertical part – fuses with its partner on the opposite side"),
    ("Descent of ovary", "First two parts → uterine tube\nCaudal parts fuse → uterine canal"),
    ("Second part",
     "Mesonephric duct moves mediocaudally"),
    ("Urogenital ridge", "Comes to lie in transverse plane"),
    ("Broad ligament",
     "Fusion of fused paramesonephric duct\n→ Broad transverse pelvic fold\n→ Broad ligament of uterus is formed"),
    ("Pelvic pouches",
     "Broad ligament + uterus divide pelvic cavity into 2 parts:\n"
     "(i) Rectouterine pouch\n(ii) Vesicouterine pouch"),
]

for title, detail in steps_fgd:
    story.append(Paragraph(f'<b>{title}:</b>', sub_heading))
    for line in detail.split('\n'):
        story.append(bullet(line.strip()))
    story.append(spacer(0.15))
    story.append(arrow())

story.pop()  # remove last arrow
story.append(spacer(0.4))

# Sub-section: Uterus
story.append(Paragraph("Uterus Formation", sub_heading))
story.append(bullet("Fused paramesonephric ducts → cervix + body of uterus"))
story.append(bullet("Surrounded by layer of mesenchyme → Myometrium + Perimetrium (peritoneal covering)"))
story.append(bullet("Two layers of broad ligament along sides of uterus → mesenchyme forms loose CT → called Parametrium"))
story.append(spacer(0.4))

# ===== SECTION 2: Development of Vagina =====
story.append(Paragraph("2. Development of the Vagina", section_heading))
story.append(spacer(0.2))

vagina_steps = [
    "Caudal tip of uterovaginal canal contacts with urogenital sinus → forming sinus tubercle",
    "Induction of formation of paired outgrowths called sinovaginal bulb",
    "Endodermal cells of bulb proliferate and form solid plate — Vaginal Plate (by fusion of bulbs)",
    "Cells from tip of uterovaginal canal also add to vaginal plate by proliferation",
    "Vaginal plate is now interposed between uterovaginal canal and urogenital sinus",
    "Upper part of plate expands & surrounds the caudal end of canal",
    "Central cells of vaginal plate break down → forming lumen of vagina",
    "Wing-like expansion of vagina around cervix → forms the Fornices of vagina",
    "Vagina is separated from urogenital sinus by a membrane — Hymen",
    "Hymen consists of thin layer of vaginal cells",
]

story.append(Paragraph("<b>Vagina has 2 origins:</b>", body_style))
story.append(bullet("Upper part: from uterovaginal canal"))
story.append(bullet("Lower part: from sinovaginal bulbs"))
story.append(spacer(0.3))

for i, step in enumerate(vagina_steps):
    story.append(bullet(step))
    if i < len(vagina_steps) - 1:
        story.append(arrow())

story.append(spacer(0.4))

# Superority note
story.append(Paragraph("<b>Superiority and epithelial lining of sinus inferiorly:</b>", sub_heading))
story.append(bullet("Hymen usually ruptures during perinatal life"))
story.append(bullet("Leaving a small opening"))
story.append(bullet("Remains itself as mucous membrane within vaginal orifice"))
story.append(spacer(0.4))

# ===== SECTION 3: Development of Ovary =====
story.append(Paragraph("3. Development of the Ovary", section_heading))
story.append(spacer(0.2))

story.append(Paragraph("<b>Left Note – Sequential Development:</b>", sub_heading))
ovary_left = [
    "Absence of Y chromosome and presence of XX",
    "Primitive sex cords containing primordial germ cells extend into medulla",
    "Cords dissociate into irregular cell clusters called Rete Ovari",
    "Later they disappear and are replaced by vascular stroma",
    "Stroma forms ovarian medulla",
    "Surface epithelium of gonad proliferates",
    "Forms 2nd generation of sex cords (by 7th week)",
    "2nd generation sex cords contain primordial germ cells",
    "Don't extend to medulla — hence called Cortical Cord",
    "3rd month of IUL: cords get fragmented and form isolated cell clusters",
    "Each cluster has a PGC in centre surrounded by a layer of coelomic epithelial cells",
    "PGC form Oogonia; coelomic epithelial cells form Follicular cells",
    "Combination of two results in formation of Primordial Follicle",
]

for i, step in enumerate(ovary_left):
    story.append(bullet(step))
    if i < len(ovary_left) - 1:
        story.append(arrow())

story.append(spacer(0.4))

story.append(Paragraph("<b>Right Note – Key Features of Ovary:</b>", sub_heading))
ovary_right = [
    "All primordial follicles remain confined in cortex",
    "Mesoderm does NOT form Tunica Albuginea thick fibrous layer",
    "Ovarian follicles are not separated from surface epithelium",
    "Surface epithelium flattens to form a single layer of cells called Germinal Epithelium",
    "It is continuous with peritoneum at hilum of ovary",
    "Mesoderm forms thin CT covering of ovary and CT stroma of ovary",
    "The ovary is suspended by Mesovarium from posterior abdominal wall",
]

for step in ovary_right:
    story.append(bullet(step))
story.append(spacer(0.4))

# ===== SECTION 4: Remnants of Genital Ducts =====
story.append(Paragraph("4. Remnants of Genital Ducts", section_heading))
story.append(spacer(0.2))

# Table for remnants
remnant_data = [
    [Paragraph('<b>Structure</b>', ParagraphStyle('th', fontName='Helvetica-Bold', fontSize=10, textColor=colors.white)),
     Paragraph('<b>Remnants</b>', ParagraphStyle('th', fontName='Helvetica-Bold', fontSize=10, textColor=colors.white))],

    [Paragraph('Mesonephric remnant\nin Male', body_style),
     Paragraph('• Appendix of epididymis\n• Paradidymis', body_style)],

    [Paragraph('Mesonephric remnant\nin Female', body_style),
     Paragraph('• Appendix vesiculosa\n• Epoophoron\n• Parabophoron\n• Duct of Gartner', body_style)],

    [Paragraph('Paramesonephric remnant\nin Male', body_style),
     Paragraph('• Appendix of Testis\n• Prostatic utricle', body_style)],

    [Paragraph('Paramesonephric remnant\nin Female', body_style),
     Paragraph('• Hydatid of Morgagni', body_style)],
]

tbl = Table(remnant_data, colWidths=[CONTENT_W * 0.42, CONTENT_W * 0.58])
tbl.setStyle(TableStyle([
    ('BACKGROUND', (0, 0), (-1, 0), colors.HexColor('#c62828')),
    ('TEXTCOLOR', (0, 0), (-1, 0), colors.white),
    ('FONTNAME', (0, 0), (-1, 0), 'Helvetica-Bold'),
    ('FONTSIZE', (0, 0), (-1, -1), 10),
    ('ROWBACKGROUNDS', (0, 1), (-1, -1), [colors.HexColor('#fce4ec'), colors.HexColor('#fff8f8')]),
    ('GRID', (0, 0), (-1, -1), 0.5, colors.HexColor('#ef9a9a')),
    ('VALIGN', (0, 0), (-1, -1), 'TOP'),
    ('PADDING', (0, 0), (-1, -1), 7),
    ('ROWBACKGROUNDS', (0, 1), (-1, -1), [colors.HexColor('#fce4ec'), colors.white]),
]))
story.append(tbl)
story.append(spacer(0.5))

# Diagram description
story.append(Paragraph("<b>Diagram Notes (from visual):</b>", sub_heading))
diag_items = [
    "Uterine tube → Uterine septum → Urogenital septum → Sinovaginal bulb → Vaginal plate",
    "Cervix → Vaginal Fornix → Vagina → Hymen (Reference: Vishram)",
    "Red arrow indicates progression from upper structures to lower formed structures",
]
for d in diag_items:
    story.append(bullet(d))

story.append(spacer(0.5))
story.append(PageBreak())

# ===== VISUAL NOTES SECTION =====
story.append(Paragraph("Visual Notes – Original Sticky Notes (Photographs)", image_section_heading))
story.append(spacer(0.3))

img_dir = "/tmp/workspace/sticky-notes-pdf"
image_info = [
    ("img3.jpg", "Fig. 1 – Orange note: Development of Female Genital Ducts (broad ligament, uterine pouches) + Uterus note (parametrium)"),
    ("img4.jpg", "Fig. 2 – Green note: Development of Vagina (full sequence with sinovaginal bulbs, vaginal plate, hymen) + Epithelial lining note"),
    ("img1.jpg", "Fig. 3 – Yellow note (continuation): Vagina lumen formation, fornices, hymen; plus Remnants of Genital Ducts table with diagram"),
    ("img2.jpg", "Fig. 4 – Development of Ovary (left & right sticky notes: cortical cord, primordial follicles, germinal epithelium, mesovarium)"),
]

for fname, caption in image_info:
    fpath = os.path.join(img_dir, fname)
    if os.path.exists(fpath):
        from PIL import Image as PILImage
        with PILImage.open(fpath) as pimg:
            iw, ih = pimg.size
        # Scale to fit content width, max height ~15cm
        max_w = CONTENT_W
        max_h = 15 * cm
        scale = min(max_w / iw, max_h / ih)
        display_w = iw * scale
        display_h = ih * scale
        img = Image(fpath, width=display_w, height=display_h)
        img.hAlign = 'CENTER'
        story.append(KeepTogether([img, Spacer(1, 0.2*cm), Paragraph(caption, note_caption), Spacer(1, 0.5*cm)]))
    else:
        story.append(body(f"[Image not found: {fname}]"))

story.append(spacer(0.5))
story.append(hr())
story.append(Paragraph(
    "Notes compiled from handwritten sticky-note summaries on embryology textbook pages. "
    "Topic: Urogenital System – Development of Female Genital Ducts, Vagina, Ovary, and Genital Duct Remnants.",
    ParagraphStyle('Footer', parent=styles['Normal'], fontSize=8, textColor=colors.HexColor('#9e9e9e'),
                   alignment=TA_CENTER)
))

# Build
doc.build(story)
print(f"PDF generated: {OUTPUT}")