#!/usr/bin/env python3
"""
Da Vinci Surgical Robot Generations - PDF Comparison Guide
"""

import json
import subprocess
import base64
import os
from reportlab.lib.pagesizes import A4
from reportlab.lib import colors
from reportlab.lib.units import cm, mm
from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
from reportlab.lib.enums import TA_CENTER, TA_LEFT, TA_JUSTIFY
from reportlab.platypus import (
    SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle,
    PageBreak, HRFlowable, Image, KeepTogether
)
from reportlab.platypus.flowables import Flowable
from reportlab.pdfgen import canvas
from io import BytesIO

# ── Color Palette ──────────────────────────────────────────────────────────────
DARK_BLUE   = colors.HexColor("#0B2545")   # header / title bg
MID_BLUE    = colors.HexColor("#1B4F8A")   # section heading bg
LIGHT_BLUE  = colors.HexColor("#D6E4F0")   # alternating table row
ACCENT      = colors.HexColor("#2980B9")   # thin rule / bullet accent
WHITE       = colors.white
LIGHT_GREY  = colors.HexColor("#F4F6F9")
TEXT_DARK   = colors.HexColor("#1A1A2E")
GOLD        = colors.HexColor("#F0A500")

OUTPUT_PATH = "/home/daytona/workspace/davinci-guide/DaVinci_Generations_Guide.pdf"
PAGE_W, PAGE_H = A4

# ── Image downloading helper ───────────────────────────────────────────────────
def fetch_images(urls):
    try:
        result = subprocess.check_output(
            ["node", "/home/daytona/skills/shared/scripts/fetch_images.js"] + urls,
            timeout=30
        )
        return json.loads(result)
    except Exception as e:
        print(f"Image fetch error: {e}")
        return [{"url": u, "base64": None, "error": str(e)} for u in urls]

def b64_to_image_flowable(b64_str, max_width, max_height):
    """Convert base64 image string to a ReportLab Image flowable."""
    try:
        raw = base64.b64decode(b64_str.split(",")[1])
        buf = BytesIO(raw)
        img = Image(buf, width=max_width, height=max_height, kind='proportional')
        return img
    except Exception as e:
        print(f"Image conversion error: {e}")
        return None

# ── Canvas background for header pages ────────────────────────────────────────
class HeaderCanvas(canvas.Canvas):
    def __init__(self, *args, **kwargs):
        canvas.Canvas.__init__(self, *args, **kwargs)
        self._saved_page_states = []

    def showPage(self):
        self._saved_page_states.append(dict(self.__dict__))
        self._startPage()

    def save(self):
        num_pages = len(self._saved_page_states)
        for state in self._saved_page_states:
            self.__dict__.update(state)
            self.draw_page_decorations(num_pages)
            canvas.Canvas.showPage(self)
        canvas.Canvas.save(self)

    def draw_page_decorations(self, total_pages):
        page_num = self._saved_page_states.index(dict(self.__dict__)) + 1 if dict(self.__dict__) in self._saved_page_states else 1
        # Bottom footer bar
        self.setFillColor(DARK_BLUE)
        self.rect(0, 0, PAGE_W, 1.2*cm, fill=1, stroke=0)
        self.setFillColor(WHITE)
        self.setFont("Helvetica", 8)
        self.drawString(2*cm, 0.4*cm, "Da Vinci Surgical Robot Generations | Intuitive Surgical")
        self.drawRightString(PAGE_W - 2*cm, 0.4*cm, f"Page {self._pageNumber} of {total_pages}")
        # Top thin accent line
        self.setStrokeColor(ACCENT)
        self.setLineWidth(2)
        self.line(0, PAGE_H - 0.5*cm, PAGE_W, PAGE_H - 0.5*cm)


# ── Styles ─────────────────────────────────────────────────────────────────────
def make_styles():
    base = getSampleStyleSheet()
    styles = {}

    styles['cover_title'] = ParagraphStyle(
        'cover_title', fontName='Helvetica-Bold', fontSize=34,
        textColor=WHITE, alignment=TA_CENTER, leading=42, spaceAfter=8
    )
    styles['cover_sub'] = ParagraphStyle(
        'cover_sub', fontName='Helvetica', fontSize=14,
        textColor=LIGHT_BLUE, alignment=TA_CENTER, leading=20, spaceAfter=4
    )
    styles['cover_year'] = ParagraphStyle(
        'cover_year', fontName='Helvetica-Bold', fontSize=12,
        textColor=GOLD, alignment=TA_CENTER, leading=16
    )
    styles['section_title'] = ParagraphStyle(
        'section_title', fontName='Helvetica-Bold', fontSize=17,
        textColor=WHITE, alignment=TA_LEFT, leading=22, leftIndent=0
    )
    styles['gen_header'] = ParagraphStyle(
        'gen_header', fontName='Helvetica-Bold', fontSize=13,
        textColor=WHITE, alignment=TA_LEFT, leading=18
    )
    styles['body'] = ParagraphStyle(
        'body', fontName='Helvetica', fontSize=10,
        textColor=TEXT_DARK, alignment=TA_JUSTIFY, leading=15, spaceAfter=4
    )
    styles['bullet'] = ParagraphStyle(
        'bullet', fontName='Helvetica', fontSize=9.5,
        textColor=TEXT_DARK, leftIndent=14, bulletIndent=4,
        leading=14, spaceAfter=2
    )
    styles['bold_label'] = ParagraphStyle(
        'bold_label', fontName='Helvetica-Bold', fontSize=10,
        textColor=MID_BLUE, leading=14, spaceAfter=2
    )
    styles['table_header'] = ParagraphStyle(
        'table_header', fontName='Helvetica-Bold', fontSize=9,
        textColor=WHITE, alignment=TA_CENTER, leading=12
    )
    styles['table_cell'] = ParagraphStyle(
        'table_cell', fontName='Helvetica', fontSize=8.5,
        textColor=TEXT_DARK, alignment=TA_CENTER, leading=12
    )
    styles['table_cell_left'] = ParagraphStyle(
        'table_cell_left', fontName='Helvetica', fontSize=8.5,
        textColor=TEXT_DARK, alignment=TA_LEFT, leading=12
    )
    styles['caption'] = ParagraphStyle(
        'caption', fontName='Helvetica-Oblique', fontSize=8,
        textColor=colors.grey, alignment=TA_CENTER, leading=11
    )
    styles['toc_entry'] = ParagraphStyle(
        'toc_entry', fontName='Helvetica', fontSize=11,
        textColor=TEXT_DARK, leading=20, leftIndent=20
    )
    return styles


# ── Cover page ─────────────────────────────────────────────────────────────────
def build_cover(styles):
    story = []

    # Dark header band
    cover_data = [[Paragraph("DA VINCI<br/>SURGICAL ROBOT", styles['cover_title'])]]
    cover_table = Table(cover_data, colWidths=[PAGE_W - 4*cm])
    cover_table.setStyle(TableStyle([
        ('BACKGROUND', (0,0), (-1,-1), DARK_BLUE),
        ('TOPPADDING', (0,0), (-1,-1), 2*cm),
        ('BOTTOMPADDING', (0,0), (-1,-1), 1.5*cm),
        ('LEFTPADDING', (0,0), (-1,-1), 1*cm),
        ('RIGHTPADDING', (0,0), (-1,-1), 1*cm),
    ]))
    story.append(cover_table)
    story.append(Spacer(1, 0.5*cm))

    # Subtitle band
    sub_data = [[Paragraph("GENERATIONS COMPARISON GUIDE", styles['cover_sub'])]]
    sub_table = Table(sub_data, colWidths=[PAGE_W - 4*cm])
    sub_table.setStyle(TableStyle([
        ('BACKGROUND', (0,0), (-1,-1), MID_BLUE),
        ('TOPPADDING', (0,0), (-1,-1), 0.7*cm),
        ('BOTTOMPADDING', (0,0), (-1,-1), 0.7*cm),
    ]))
    story.append(sub_table)
    story.append(Spacer(1, 0.8*cm))

    # Year badge
    story.append(Paragraph("1999 - 2024", styles['cover_year']))
    story.append(Spacer(1, 0.4*cm))

    # Timeline visual as table
    timeline = [
        ["1999", "2006", "2009", "2014", "2017/18", "2024"],
        ["Standard", "S", "Si", "Xi", "X / SP", "5"],
        ["1st Gen", "2nd Gen", "3rd Gen", "4th Gen", "Variants", "5th Gen"],
    ]
    col_w = (PAGE_W - 4*cm) / 6
    tl_table = Table(timeline, colWidths=[col_w]*6, rowHeights=[1.0*cm, 0.9*cm, 0.7*cm])
    tl_table.setStyle(TableStyle([
        # Year row
        ('BACKGROUND', (0,0), (-1,0), MID_BLUE),
        ('TEXTCOLOR', (0,0), (-1,0), WHITE),
        ('FONTNAME', (0,0), (-1,0), 'Helvetica-Bold'),
        ('FONTSIZE', (0,0), (-1,0), 11),
        ('ALIGN', (0,0), (-1,0), 'CENTER'),
        ('VALIGN', (0,0), (-1,0), 'MIDDLE'),
        # Name row
        ('BACKGROUND', (0,1), (-1,1), LIGHT_BLUE),
        ('TEXTCOLOR', (0,1), (-1,1), DARK_BLUE),
        ('FONTNAME', (0,1), (-1,1), 'Helvetica-Bold'),
        ('FONTSIZE', (0,1), (-1,1), 13),
        ('ALIGN', (0,1), (-1,1), 'CENTER'),
        ('VALIGN', (0,1), (-1,1), 'MIDDLE'),
        # Gen row
        ('BACKGROUND', (0,2), (-1,2), LIGHT_GREY),
        ('TEXTCOLOR', (0,2), (-1,2), colors.grey),
        ('FONTNAME', (0,2), (-1,2), 'Helvetica-Oblique'),
        ('FONTSIZE', (0,2), (-1,2), 8),
        ('ALIGN', (0,2), (-1,2), 'CENTER'),
        ('VALIGN', (0,2), (-1,2), 'MIDDLE'),
        ('GRID', (0,0), (-1,-1), 0.5, WHITE),
        ('BOX', (0,0), (-1,-1), 1.5, ACCENT),
    ]))
    story.append(tl_table)
    story.append(Spacer(1, 1.0*cm))

    # Cover summary paragraph
    cover_text = (
        "Since its FDA clearance in July 2000, the da Vinci Surgical System by Intuitive Surgical "
        "has transformed minimally invasive surgery. Each generation has brought meaningful advances "
        "in vision, instrumentation, arm design, ergonomics, and data integration. This guide "
        "compares every platform - from the original da Vinci Standard to the latest da Vinci 5 - "
        "covering key specifications, clinical improvements, and the most important differences "
        "surgeons and hospitals need to know."
    )
    story.append(Paragraph(cover_text, styles['body']))
    story.append(Spacer(1, 0.8*cm))

    # Published by
    pub_data = [[
        Paragraph("Compiled from Intuitive Surgical data, peer-reviewed literature, and clinical sources.", styles['caption']),
    ]]
    pub_table = Table(pub_data, colWidths=[PAGE_W - 4*cm])
    pub_table.setStyle(TableStyle([
        ('BACKGROUND', (0,0), (-1,-1), LIGHT_GREY),
        ('TOPPADDING', (0,0), (-1,-1), 0.5*cm),
        ('BOTTOMPADDING', (0,0), (-1,-1), 0.5*cm),
        ('BOX', (0,0), (-1,-1), 1, ACCENT),
    ]))
    story.append(pub_table)

    story.append(PageBreak())
    return story


# ── TOC ────────────────────────────────────────────────────────────────────────
def build_toc(styles):
    story = []
    toc_header = [[Paragraph("TABLE OF CONTENTS", styles['section_title'])]]
    toc_table = Table(toc_header, colWidths=[PAGE_W - 4*cm])
    toc_table.setStyle(TableStyle([
        ('BACKGROUND', (0,0), (-1,-1), DARK_BLUE),
        ('TOPPADDING', (0,0), (-1,-1), 0.6*cm),
        ('BOTTOMPADDING', (0,0), (-1,-1), 0.6*cm),
        ('LEFTPADDING', (0,0), (-1,-1), 0.8*cm),
    ]))
    story.append(toc_table)
    story.append(Spacer(1, 0.6*cm))

    entries = [
        ("1.", "Da Vinci Standard (1999) - 1st Generation"),
        ("2.", "Da Vinci S (2006) - 2nd Generation"),
        ("3.", "Da Vinci Si (2009) - 3rd Generation"),
        ("4.", "Da Vinci Xi (2014) - 4th Generation"),
        ("5.", "Da Vinci X (2017) - Budget Hybrid"),
        ("6.", "Da Vinci SP (2018) - Single Port"),
        ("7.", "Da Vinci 5 (2024) - 5th Generation"),
        ("8.", "Full Comparison Table - All Generations"),
        ("9.", "Summary & Key Takeaways"),
    ]

    for num, title in entries:
        row = [[
            Paragraph(f"<b>{num}</b>", styles['bold_label']),
            Paragraph(title, styles['toc_entry']),
        ]]
        row_table = Table(row, colWidths=[1.2*cm, PAGE_W - 5.2*cm])
        row_table.setStyle(TableStyle([
            ('VALIGN', (0,0), (-1,-1), 'MIDDLE'),
            ('LINEBELOW', (0,0), (-1,-1), 0.3, colors.HexColor("#CCCCCC")),
            ('TOPPADDING', (0,0), (-1,-1), 4),
            ('BOTTOMPADDING', (0,0), (-1,-1), 4),
        ]))
        story.append(row_table)

    story.append(PageBreak())
    return story


# ── Section header band ────────────────────────────────────────────────────────
def section_header(title, year, gen_label, styles):
    """Returns a styled generation header table."""
    header_data = [[
        Paragraph(f"{title}", styles['section_title']),
        Paragraph(f"<b>{year}</b><br/><font size='9'>{gen_label}</font>", ParagraphStyle(
            'year_label', fontName='Helvetica-Bold', fontSize=13,
            textColor=GOLD, alignment=TA_CENTER, leading=16
        )),
    ]]
    ht = Table(header_data, colWidths=[PAGE_W - 8*cm, 4*cm])
    ht.setStyle(TableStyle([
        ('BACKGROUND', (0,0), (-1,-1), DARK_BLUE),
        ('LEFTPADDING', (0,0), (0,-1), 0.8*cm),
        ('RIGHTPADDING', (-1,0), (-1,-1), 0.6*cm),
        ('TOPPADDING', (0,0), (-1,-1), 0.6*cm),
        ('BOTTOMPADDING', (0,0), (-1,-1), 0.6*cm),
        ('VALIGN', (0,0), (-1,-1), 'MIDDLE'),
    ]))
    return ht


def specs_table(rows, styles):
    """Build a two-column spec table (Label | Value)."""
    table_data = []
    for label, value in rows:
        table_data.append([
            Paragraph(f"<b>{label}</b>", styles['bold_label']),
            Paragraph(value, styles['body']),
        ])
    t = Table(table_data, colWidths=[5.5*cm, PAGE_W - 9.5*cm])
    t.setStyle(TableStyle([
        ('BACKGROUND', (0,0), (-1,-1), WHITE),
        ('ROWBACKGROUNDS', (0,0), (-1,-1), [WHITE, LIGHT_GREY]),
        ('VALIGN', (0,0), (-1,-1), 'TOP'),
        ('TOPPADDING', (0,0), (-1,-1), 4),
        ('BOTTOMPADDING', (0,0), (-1,-1), 4),
        ('LEFTPADDING', (0,0), (-1,-1), 8),
        ('RIGHTPADDING', (0,0), (-1,-1), 8),
        ('LINEBELOW', (0,0), (-1,-1), 0.3, colors.HexColor("#DDDDDD")),
        ('BOX', (0,0), (-1,-1), 0.8, ACCENT),
    ]))
    return t


def bullet_list(items, styles):
    out = []
    for item in items:
        out.append(Paragraph(f"<bullet color='#{ACCENT.hexval()[2:]}'>&#x25CF;</bullet> {item}", styles['bullet']))
    return out


# ── Generation pages ───────────────────────────────────────────────────────────

def gen_standard(styles):
    story = []
    story.append(section_header("DA VINCI STANDARD", "1999 / 2000", "1st Generation", styles))
    story.append(Spacer(1, 0.5*cm))

    story.append(Paragraph(
        "The original da Vinci Surgical System received FDA clearance in July 2000, marking the "
        "beginning of commercial robotic surgery. It was the first system to translate a surgeon's "
        "hand movements into precise micro-movements of instruments inside the patient, through a "
        "stereoscopic 3D viewer at the surgeon's console.",
        styles['body']
    ))
    story.append(Spacer(1, 0.4*cm))

    story.append(specs_table([
        ("FDA Clearance", "July 2000"),
        ("Arms", "3 (4th instrument arm added 2003)"),
        ("Camera", "Fixed dedicated camera arm"),
        ("Vision", "Standard 3D stereoscopic (non-HD)"),
        ("Degrees of Freedom", "7 DoF + 2 axial rotation per instrument"),
        ("Console", "Binocular viewer; computing housed in console"),
        ("Dual Console", "No"),
        ("Force Feedback", "No"),
        ("Fluorescence Imaging", "No"),
        ("HD Vision", "No"),
        ("Table Motion", "No"),
        ("Docking", "Slow; complex setup"),
    ], styles))
    story.append(Spacer(1, 0.4*cm))

    story.append(Paragraph("<b>Key Features &amp; Innovations:</b>", styles['bold_label']))
    story.extend(bullet_list([
        "First commercially available robotic surgery platform (Intuitive Surgical)",
        "EndoWrist instruments with 7 degrees of freedom - mimicking human wrist motion and eliminating tremor",
        "Immersive stereoscopic 3D binocular viewer at surgeon console (not a flat monitor)",
        "All robotic arms originating from a single patient cart (no per-arm table mounting)",
        "Surgeon seated at console, fully decoupled from the patient",
        "Heavy camera arm requiring counterweights due to large dual-optic endoscopes",
        "Long central pedal on foot board for manual camera focus",
    ], styles))
    story.append(Spacer(1, 0.4*cm))

    story.append(Paragraph("<b>Limitations:</b>", styles['bold_label']))
    story.extend(bullet_list([
        "No high-definition imaging",
        "Bulky footprint and long OR setup times",
        "Limited arm flexibility; arms required wide spacing",
        "No dual-console capability (no teaching/proctoring setup)",
        "Single console only; all computing in console",
        "No fluorescence or near-infrared imaging",
    ], styles))

    story.append(PageBreak())
    return story


def gen_s(styles):
    story = []
    story.append(section_header("DA VINCI S", "2006", "2nd Generation", styles))
    story.append(Spacer(1, 0.5*cm))

    story.append(Paragraph(
        "Launched in 2006, the da Vinci S was a significant hardware and software upgrade. "
        "The most impactful improvement was the introduction of 3D high-definition vision, which "
        "dramatically improved the surgeon's ability to see fine tissue planes and structures. "
        "The patient-side cart was redesigned to be lighter and more streamlined.",
        styles['body']
    ))
    story.append(Spacer(1, 0.4*cm))

    story.append(specs_table([
        ("FDA Clearance", "2006"),
        ("Arms", "4 (column-mounted)"),
        ("Camera", "Fixed dedicated camera arm"),
        ("Vision", "3D High-Definition (HD) - major upgrade"),
        ("Console", "Binocular viewer; similar pedal board to Standard"),
        ("Dual Console", "No"),
        ("Force Feedback", "No"),
        ("Fluorescence Imaging", "No"),
        ("Table Motion", "No"),
        ("Docking Time", "Moderate; improved over Standard"),
        ("Setup", "Interactive touchscreen added"),
    ], styles))
    story.append(Spacer(1, 0.4*cm))

    story.append(Paragraph("<b>Key Improvements over Standard:</b>", styles['bold_label']))
    story.extend(bullet_list([
        "3D High-Definition (HD) camera - the single most important visual leap in the platform's early history",
        "Lighter, more streamlined patient-side cart with improved aesthetics",
        "Updated software architecture with faster processing",
        "Autofocus camera system introduced (eliminating the manual focus pedal)",
        "Simplified system setup with interactive touchscreen interface",
        "Improved instrument compatibility and EndoWrist range",
        "da Vinci S HD variant later released with further image quality improvements",
    ], styles))
    story.append(Spacer(1, 0.4*cm))

    story.append(Paragraph("<b>Remaining Limitations:</b>", styles['bold_label']))
    story.extend(bullet_list([
        "Still uses column-based arm design; arms must be widely spaced",
        "No dual-console capability",
        "No fluorescence or near-infrared imaging",
        "Operation times significantly longer than Xi (studies show ~85 min longer)",
        "Docking time still ~16 min vs later generations",
    ], styles))

    story.append(PageBreak())
    return story


def gen_si(styles):
    story = []
    story.append(section_header("DA VINCI Si", "2009", "3rd Generation", styles))
    story.append(Spacer(1, 0.5*cm))

    story.append(Paragraph(
        "The da Vinci Si (2009) became the most widely disseminated robotic surgery platform globally. "
        "It introduced the transformative dual-console capability, enabling proctored surgery and "
        "resident/fellow training. The computing brain was moved from the surgeon's console to the "
        "vision cart - a strategic architecture shift that enabled all future improvements.",
        styles['body']
    ))
    story.append(Spacer(1, 0.4*cm))

    story.append(specs_table([
        ("FDA Clearance", "2009"),
        ("Arms", "4 (column-mounted)"),
        ("Camera", "Fixed dedicated camera arm"),
        ("Vision", "HD 3D + TilePro multi-input display"),
        ("Dual Console", "YES - first dual-console system"),
        ("Force Feedback", "No"),
        ("Fluorescence Imaging", "YES - real-time near-infrared"),
        ("Table Motion", "No"),
        ("Docking Time", "~19 min (longer than Xi)"),
        ("Computing", "Moved to vision cart (architectural shift)"),
        ("Communication", "Two fiber-optic cables between components"),
    ], styles))
    story.append(Spacer(1, 0.4*cm))

    story.append(Paragraph("<b>Key Improvements over S:</b>", styles['bold_label']))
    story.extend(bullet_list([
        "Dual-console surgery: a second surgeon console connects to the same patient cart - enabling proctored training and hand-off during complex procedures",
        "System computing moved from surgeon's console to the vision cart - sleeker console design and modular future upgrades",
        "Communication simplified to just two optic fiber cables between system components",
        "TilePro software: multiple video inputs (e.g., ultrasound + endoscope) displayed simultaneously on console",
        "Real-time fluorescence/near-infrared imaging (e.g., for sentinel node mapping, blood flow assessment)",
        "Touchscreen added to arm rest for intraoperative system configuration",
        "Assistant touchscreen on vision cart for parallel setup",
    ], styles))
    story.append(Spacer(1, 0.4*cm))

    story.append(Paragraph("<b>Remaining Limitations:</b>", styles['bold_label']))
    story.extend(bullet_list([
        "Patient cart design unchanged from S - arms still on central column",
        "Longer docking time (~19 min) vs Xi (~9.5 min)",
        "No overhead boom architecture; limited multi-quadrant repositioning",
        "Accuracy: FLE ~1.64 mm vs ~0.97 mm for Xi",
    ], styles))

    story.append(PageBreak())
    return story


def gen_xi(styles):
    story = []
    story.append(section_header("DA VINCI Xi", "2014", "4th Generation", styles))
    story.append(Spacer(1, 0.5*cm))

    story.append(Paragraph(
        "The da Vinci Xi (2014) was the most radical redesign in the platform's history. "
        "The entire patient-side cart was rebuilt from scratch with an overhead boom architecture "
        "that suspended the arms above the patient. This unlocked multi-quadrant surgery without "
        "re-docking and made the Xi the gold standard for complex minimally invasive procedures. "
        "It remains the most widely used multiport robotic system in the world.",
        styles['body']
    ))
    story.append(Spacer(1, 0.4*cm))

    story.append(specs_table([
        ("FDA Clearance", "2014"),
        ("Arms", "4 (overhead boom-mounted; any arm = camera)"),
        ("Instrument Size", "All instruments and camera: 8 mm"),
        ("Camera", "Any of the 4 arms can hold the camera"),
        ("Vision", "HD 3D + Firefly fluorescence imaging integrated"),
        ("Dual Console", "YES"),
        ("Force Feedback", "No"),
        ("Table Motion", "YES - integrated table motion (patient repositioning without undocking)"),
        ("Docking Time", "~9.5 min (vs ~19 min for Si)"),
        ("Accuracy (FLE)", "0.97 mm (vs 1.64 mm for Si)"),
        ("Port Setup", "Laser-guided port positioning system"),
        ("Instrument Reach", "Expanded workspace vs Si"),
        ("Vessel Sealer", "Vessel Sealer Extend: seals/cuts up to 7 mm"),
        ("Cannulas", "Fully reusable stainless steel; bladeless/blunt tip options"),
    ], styles))
    story.append(Spacer(1, 0.4*cm))

    story.append(Paragraph("<b>Landmark Improvements over Si:</b>", styles['bold_label']))
    story.extend(bullet_list([
        "Overhead boom architecture: arms suspended from horizontal boom above surgical field - no central column",
        "Arms numbered 1-4; ANY arm can hold the camera (no fixed camera arm)",
        "360-degree approach: dock from any angle, access any abdominal quadrant without re-docking",
        "Smaller arm spacing (one fist-width vs widely spaced) reduces external collisions",
        "Thinner arms with redesigned joints: greater internal range of motion and improved patient clearance",
        "Laser-guided port positioning for optimal trocar placement",
        "Integrated table motion: patient repositioning intraoperatively without breaking the sterile field",
        "Submillimetric accuracy: FLE 0.97 mm vs 1.64 mm on Si",
        "Shorter docking time: ~9.5 min (50% faster than Si)",
        "Firefly near-infrared fluorescence imaging fully integrated",
        "Second-generation Vessel Sealer Extend (up to 7 mm vessels)",
        "Improved robotic stapling with articulation and intelligent feedback",
    ], styles))

    story.append(PageBreak())
    return story


def gen_x(styles):
    story = []
    story.append(section_header("DA VINCI X", "2017", "Budget Hybrid (3rd/4th Gen Bridge)", styles))
    story.append(Spacer(1, 0.5*cm))

    story.append(Paragraph(
        "The da Vinci X is not a true new generation - it is a cost-optimized hybrid designed to "
        "give hospitals access to Xi-generation instruments at a lower acquisition price. It uses "
        "the Si platform's patient-side cart, modified to accept Xi EndoWrist instruments.",
        styles['body']
    ))
    story.append(Spacer(1, 0.4*cm))

    story.append(specs_table([
        ("Released", "2017"),
        ("Patient Cart", "Si-based cart (column design) modified for Xi instruments"),
        ("Instruments", "Xi-generation EndoWrist instruments (8 mm)"),
        ("Camera", "Fixed camera arm (Si-style)"),
        ("Vision", "HD 3D"),
        ("Dual Console", "YES (limited)"),
        ("Force Feedback", "No"),
        ("Table Motion", "No"),
        ("Purpose", "Cost reduction vs full Xi system"),
        ("Overhead Boom", "No - retains Si column design"),
    ], styles))
    story.append(Spacer(1, 0.4*cm))

    story.append(Paragraph("<b>Positioning:</b>", styles['bold_label']))
    story.extend(bullet_list([
        "Entry-level option for hospitals unable to afford the full Xi platform",
        "Bridges instrument compatibility between 3rd and 4th generation platforms",
        "Shares Xi-generation instrument inventory, useful for multi-robot institutions",
        "Not recommended for complex multi-quadrant cases requiring full Xi flexibility",
        "Has since been phased out in favor of Xi and da Vinci 5",
    ], styles))

    story.append(Spacer(1, 0.6*cm))

    story.append(section_header("DA VINCI SP", "2018 / 2019", "Single-Port Variant (Xi Platform)", styles))
    story.append(Spacer(1, 0.5*cm))

    story.append(Paragraph(
        "The da Vinci SP was originally designed on the Si platform but that version never reached "
        "the market. The released system is built on the Xi platform. It is uniquely designed for "
        "single-incision and natural orifice surgery, delivering all instruments and the camera "
        "through a single cannula.",
        styles['body']
    ))
    story.append(Spacer(1, 0.4*cm))

    story.append(specs_table([
        ("FDA Clearance", "2018 (urology); expanded 2019+"),
        ("Base Platform", "Xi"),
        ("Arms", "1 outer arm; 3 articulating instruments + camera inside"),
        ("Entry", "Single cannula (all instruments through one port)"),
        ("Camera", "Fully articulating 3D HD endoscope"),
        ("Vision", "3D HD articulating endoscope"),
        ("Access Route", "Single incision OR natural orifice (transoral, transvaginal, transanal)"),
        ("Instrument Behavior", "Instruments diverge inside patient, then converge at target"),
        ("Footprint", "Smaller than Xi"),
        ("Dual Console", "No"),
        ("Force Feedback", "No"),
    ], styles))
    story.append(Spacer(1, 0.4*cm))

    story.append(Paragraph("<b>Clinical Advantages of SP:</b>", styles['bold_label']))
    story.extend(bullet_list([
        "Ideal for confined anatomical spaces: retroauricular neck surgery, pelvis, urethra, vagina",
        "Only one skin incision required (or natural orifice entry)",
        "Fully articulating camera provides improved visualization in narrow spaces",
        "Enables procedures previously impossible with standard multiport robotic systems",
        "Used in transoral robotic surgery (TORS) for oropharynx and larynx",
    ], styles))

    story.append(PageBreak())
    return story


def gen_dv5(styles):
    story = []
    story.append(section_header("DA VINCI 5", "2024", "5th Generation", styles))
    story.append(Spacer(1, 0.5*cm))

    story.append(Paragraph(
        "The da Vinci 5 received FDA clearance in March 2024 and represents the most significant "
        "leap in the platform since the Xi in 2014. While it shares the Xi's proven boom architecture "
        "externally, nearly everything under the skin has been redesigned. It brings over 150 design "
        "innovations, 10,000x the computing power of the Xi, and introduces the first force feedback "
        "technology in the platform's history.",
        styles['body']
    ))
    story.append(Spacer(1, 0.4*cm))

    story.append(specs_table([
        ("FDA Clearance", "March 2024"),
        ("Arms", "4 (overhead boom - same architecture as Xi)"),
        ("Camera", "Any arm (same as Xi)"),
        ("Vision", "Next-generation 3D imaging (improved over Xi)"),
        ("Computing Power", "10,000x more than da Vinci Xi"),
        ("Force Feedback", "YES - first in da Vinci history"),
        ("Tremor/Vibration", "Best-ever filtration (improved input + output)"),
        ("Hand Controllers", "Redesigned curved haptic arms (more ergonomic)"),
        ("Docking Speed", "~2x faster than Xi"),
        ("Table Motion", "YES (integrated, same as Xi)"),
        ("Fluorescence", "YES (enhanced)"),
        ("OR Footprint", "Smaller than Xi"),
        ("Data Integration", "My Intuitive+, SimNow VR, Case Insights, Intuitive Hub"),
        ("Dual Console", "YES"),
        ("Indications", "Same as Xi EXCEPT cardiac and pediatric (as of 2024)"),
        ("Force Feedback Contraindication", "Hysterectomy and myomectomy for Force Feedback needle driver"),
    ], styles))
    story.append(Spacer(1, 0.4*cm))

    story.append(Paragraph("<b>Landmark Advances over Xi:</b>", styles['bold_label']))
    story.extend(bullet_list([
        "Force Feedback: sensors measure tissue interaction forces; haptic resistance felt in surgeon's hands - toggleable on/off. Most significant sensory advance in the platform's 25-year history",
        "10,000x computing power: enables AI-driven features, real-time analytics, and future autonomous capabilities",
        "Redesigned hand controllers with curved haptic arms; thicker for more electronics and better tremor filtration",
        "Additional arm sensors for improved precision and sub-millimetric accuracy",
        "~2x faster docking workflow with guided, task-automated setup",
        "Surgeons make fewer staff requests for settings changes (streamlined intraoperative workflow)",
        "Universal user interface across all three system components (cart, console, vision system)",
        "Integration with My Intuitive+ for objective surgical performance review and personalized training",
        "SimNow VR simulator tightly integrated for training and credentialing",
        "Case Insights: AI-driven computational observer flags key surgical events from recorded video",
        "Smaller OR footprint than Xi",
        "Cost: ~15% higher than full Xi stack (inclusive of table motion, hub, simulation items bundled)",
    ], styles))

    story.append(PageBreak())
    return story


# ── Comparison table ───────────────────────────────────────────────────────────
def build_comparison_table(styles):
    story = []

    header_data = [[Paragraph("FULL COMPARISON TABLE", styles['section_title'])]]
    ht = Table(header_data, colWidths=[PAGE_W - 4*cm])
    ht.setStyle(TableStyle([
        ('BACKGROUND', (0,0), (-1,-1), DARK_BLUE),
        ('TOPPADDING', (0,0), (-1,-1), 0.6*cm),
        ('BOTTOMPADDING', (0,0), (-1,-1), 0.6*cm),
        ('LEFTPADDING', (0,0), (-1,-1), 0.8*cm),
    ]))
    story.append(ht)
    story.append(Spacer(1, 0.5*cm))

    def H(text): return Paragraph(f"<b>{text}</b>", styles['table_header'])
    def C(text, bold=False):
        st = ParagraphStyle('tc', fontName='Helvetica-Bold' if bold else 'Helvetica',
                            fontSize=8, textColor=TEXT_DARK, alignment=TA_CENTER, leading=11)
        return Paragraph(text, st)
    def CL(text):
        st = ParagraphStyle('tcl', fontName='Helvetica-Bold', fontSize=8,
                            textColor=TEXT_DARK, alignment=TA_LEFT, leading=11)
        return Paragraph(text, st)

    YES  = lambda: Paragraph("<b><font color='#27AE60'>YES</font></b>", ParagraphStyle('y', fontName='Helvetica-Bold', fontSize=8, alignment=TA_CENTER))
    NO   = lambda: Paragraph("<font color='#C0392B'>No</font>", ParagraphStyle('n', fontName='Helvetica', fontSize=8, alignment=TA_CENTER))
    PART = lambda t: Paragraph(f"<font color='#E67E22'>{t}</font>", ParagraphStyle('p', fontName='Helvetica', fontSize=8, alignment=TA_CENTER))

    col_w = [3.8*cm, 2.2*cm, 2.2*cm, 2.2*cm, 2.5*cm, 2.0*cm, 2.0*cm, 2.5*cm]

    table_data = [
        # Header
        [H("Feature"), H("Standard\n1999"), H("S\n2006"), H("Si\n2009"), H("Xi\n2014"), H("X\n2017"), H("SP\n2018"), H("DV5\n2024")],
        # Rows
        [CL("Arms"), C("3→4"), C("4"), C("4"), C("4"), C("4"), C("1 (3 inst)"), C("4")],
        [CL("Fixed Camera Arm"), YES(), YES(), YES(), NO(), YES(), C("Integrated"), NO()],
        [CL("HD Vision"), NO(), YES(), YES(), YES(), YES(), YES(), C("Enhanced")],
        [CL("3D Vision"), YES(), YES(), YES(), YES(), YES(), YES(), YES()],
        [CL("Dual Console"), NO(), NO(), YES(), YES(), PART("Limited"), NO(), YES()],
        [CL("Fluorescence (NIR)"), NO(), NO(), YES(), YES(), YES(), YES(), C("Enhanced")],
        [CL("Force Feedback"), NO(), NO(), NO(), NO(), NO(), NO(), YES()],
        [CL("Table Motion"), NO(), NO(), NO(), YES(), NO(), NO(), YES()],
        [CL("Overhead Boom"), NO(), NO(), NO(), YES(), NO(), NO(), YES()],
        [CL("Single Port"), NO(), NO(), NO(), NO(), NO(), YES(), NO()],
        [CL("Docking Time"), C("Slow"), C("~16 min"), C("~19 min"), C("~9.5 min"), C("Moderate"), C("N/A"), C("~2x faster\nthan Xi")],
        [CL("Accuracy (FLE mm)"), C("~1.31"), C("~1.31"), C("~1.64"), C("0.97"), C("~1.0"), C("N/A"), C("<0.97")],
        [CL("AI / Data Integration"), NO(), NO(), NO(), PART("Basic"), NO(), NO(), YES()],
        [CL("Computing Power"), C("Basic"), C("Basic"), C("Moderate"), C("Moderate"), C("Moderate"), C("Moderate"), C("10,000x vs Xi")],
        [CL("Laser Port Guide"), NO(), NO(), NO(), YES(), NO(), NO(), YES()],
        [CL("OR Footprint"), C("Large"), C("Large"), C("Large"), C("Moderate"), C("Moderate"), C("Compact"), C("Smaller than Xi")],
    ]

    ct = Table(table_data, colWidths=col_w, repeatRows=1)
    ct.setStyle(TableStyle([
        # Header row
        ('BACKGROUND', (0,0), (-1,0), DARK_BLUE),
        ('TEXTCOLOR', (0,0), (-1,0), WHITE),
        ('ALIGN', (0,0), (-1,-1), 'CENTER'),
        ('VALIGN', (0,0), (-1,-1), 'MIDDLE'),
        # Alternating rows
        ('ROWBACKGROUNDS', (0,1), (-1,-1), [WHITE, LIGHT_GREY]),
        # Feature column left-align
        ('ALIGN', (0,1), (0,-1), 'LEFT'),
        ('LEFTPADDING', (0,0), (0,-1), 6),
        ('TOPPADDING', (0,0), (-1,-1), 5),
        ('BOTTOMPADDING', (0,0), (-1,-1), 5),
        ('FONTSIZE', (0,0), (-1,-1), 8),
        ('GRID', (0,0), (-1,-1), 0.4, colors.HexColor("#CCCCCC")),
        ('BOX', (0,0), (-1,-1), 1, ACCENT),
        # Highlight DV5 column
        ('BACKGROUND', (7,1), (7,-1), colors.HexColor("#EAF4FB")),
    ]))
    story.append(ct)
    story.append(PageBreak())
    return story


# ── Summary / Takeaways ────────────────────────────────────────────────────────
def build_summary(styles):
    story = []
    header_data = [[Paragraph("SUMMARY &amp; KEY TAKEAWAYS", styles['section_title'])]]
    ht = Table(header_data, colWidths=[PAGE_W - 4*cm])
    ht.setStyle(TableStyle([
        ('BACKGROUND', (0,0), (-1,-1), DARK_BLUE),
        ('TOPPADDING', (0,0), (-1,-1), 0.6*cm),
        ('BOTTOMPADDING', (0,0), (-1,-1), 0.6*cm),
        ('LEFTPADDING', (0,0), (-1,-1), 0.8*cm),
    ]))
    story.append(ht)
    story.append(Spacer(1, 0.5*cm))

    milestones = [
        ("1999/2000 - Standard", "First commercial platform. Proved robotic surgery was viable. EndoWrist instruments and immersive 3D console established the template all successors follow."),
        ("2006 - da Vinci S", "HD vision transformed surgical detail. Autofocus and lighter cart improved usability. Established the platform in broader clinical use."),
        ("2009 - da Vinci Si", "Dual-console surgery changed surgical education. Moving the system brain to the vision cart was the architectural foundation for all future generations."),
        ("2014 - da Vinci Xi", "The most radical redesign. Overhead boom, any-arm camera, 360-degree access, laser port guidance, integrated table motion, and sub-mm accuracy. Set the global standard."),
        ("2017 - da Vinci X", "Cost-access bridge. Not a clinical leap but made Xi-generation instruments available at lower cost. Now largely superseded."),
        ("2018 - da Vinci SP", "Enabled single-port and natural orifice procedures previously impossible robotically. Key for urology, ENT, and colorectal."),
        ("2024 - da Vinci 5", "First force feedback in 25 years. 10,000x computing power unlocks AI features. Fastest docking, best ergonomics, and most data-rich platform ever built."),
    ]

    for name, desc in milestones:
        row_data = [[
            Paragraph(f"<b>{name}</b>", ParagraphStyle('ml', fontName='Helvetica-Bold', fontSize=10, textColor=WHITE, leading=14)),
            Paragraph(desc, styles['body']),
        ]]
        row_table = Table(row_data, colWidths=[4.5*cm, PAGE_W - 8.5*cm])
        row_table.setStyle(TableStyle([
            ('BACKGROUND', (0,0), (0,-1), MID_BLUE),
            ('BACKGROUND', (1,0), (1,-1), LIGHT_GREY),
            ('VALIGN', (0,0), (-1,-1), 'TOP'),
            ('TOPPADDING', (0,0), (-1,-1), 8),
            ('BOTTOMPADDING', (0,0), (-1,-1), 8),
            ('LEFTPADDING', (0,0), (0,-1), 8),
            ('LEFTPADDING', (1,0), (1,-1), 10),
            ('LINEBELOW', (0,0), (-1,-1), 0.5, WHITE),
        ]))
        story.append(row_table)

    story.append(Spacer(1, 0.8*cm))

    advice_data = [[
        Paragraph("<b>Choosing a Platform</b>", ParagraphStyle('adv', fontName='Helvetica-Bold', fontSize=11, textColor=WHITE, leading=16)),
    ]]
    adv_header = Table(advice_data, colWidths=[PAGE_W - 4*cm])
    adv_header.setStyle(TableStyle([
        ('BACKGROUND', (0,0), (-1,-1), ACCENT),
        ('TOPPADDING', (0,0), (-1,-1), 0.4*cm),
        ('BOTTOMPADDING', (0,0), (-1,-1), 0.4*cm),
        ('LEFTPADDING', (0,0), (-1,-1), 0.8*cm),
    ]))
    story.append(adv_header)
    story.append(Spacer(1, 0.3*cm))

    advice = [
        "For highest capability multiport surgery: da Vinci 5 (2024) is the top choice with force feedback and AI integration",
        "For most widely-deployed proven multiport platform: da Vinci Xi (2014) with 7+ million procedures recorded",
        "For single-incision or natural orifice approaches: da Vinci SP",
        "For cost-sensitive entry: da Vinci X (though Xi is strongly preferred for clinical versatility)",
        "The S and Si are largely phased out of new installations; Si may still be in active use at some centers",
        "The Standard (Classic) has been fully retired",
    ]
    story.extend(bullet_list(advice, styles))

    story.append(Spacer(1, 0.8*cm))
    story.append(HRFlowable(width="100%", thickness=1, color=ACCENT))
    story.append(Spacer(1, 0.3*cm))
    story.append(Paragraph(
        "Sources: Intuitive Surgical product data, PMC10683436 (History of Robotic Surgery), "
        "PMC8155181 (S/Si/Xi outcomes comparison), PMC9540612 (Xi vs Si accuracy), "
        "FDA 510(k) clearance records, and peer-reviewed surgical literature.",
        styles['caption']
    ))

    return story


# ── Main ───────────────────────────────────────────────────────────────────────
def main():
    styles = make_styles()

    doc = SimpleDocTemplate(
        OUTPUT_PATH,
        pagesize=A4,
        rightMargin=2*cm, leftMargin=2*cm,
        topMargin=1.8*cm, bottomMargin=2.0*cm,
        title="Da Vinci Surgical Robot Generations Comparison Guide",
        author="Orris Medical AI",
        subject="Robotic Surgery Platform Comparison",
    )

    story = []
    story += build_cover(styles)
    story += build_toc(styles)
    story += gen_standard(styles)
    story += gen_s(styles)
    story += gen_si(styles)
    story += gen_xi(styles)
    story += gen_x(styles)
    story += gen_dv5(styles)
    story += build_comparison_table(styles)
    story += build_summary(styles)

    doc.build(story, canvasmaker=HeaderCanvas)
    print(f"PDF saved to: {OUTPUT_PATH}")


if __name__ == "__main__":
    main()