| from __future__ import annotations |
| from pathlib import Path |
| import fitz |
| from src.core.models import PDFOperation |
|
|
| class OfflineAnnotationAgent: |
| name = "annotation_agent" |
|
|
| def run(self, input_pdf: Path, output_pdf: Path, operations: list[PDFOperation]) -> Path: |
| doc = fitz.open(input_pdf) |
| for op in operations: |
| if op.page is None or op.page >= len(doc): continue |
| page = doc[op.page] |
| rect = fitz.Rect(op.rect) if op.rect else None |
| color = tuple(op.color) if op.color else (1, 1, 0) |
| if op.type == "highlight" and rect: |
| annot = page.add_highlight_annot(rect) |
| annot.set_colors(stroke=color) |
| annot.update() |
| elif op.type == "comment" and op.position and op.text: |
| annot = page.add_text_annot(fitz.Point(op.position), op.text) |
| annot.set_info(title=op.author, content=op.text) |
| annot.update() |
| elif op.type == "rectangle" and rect: |
| annot = page.add_rect_annot(rect) |
| annot.set_colors(stroke=color) |
| annot.update() |
| elif op.type == "redact" and rect: |
| page.add_redact_annot(rect, text=op.text or "", fill=(0,0,0)) |
| for page in doc: |
| try: |
| page.apply_redactions(images=fitz.PDF_REDACT_IMAGE_REMOVE) |
| except Exception: |
| pass |
| output_pdf.parent.mkdir(parents=True, exist_ok=True) |
| doc.save(output_pdf, garbage=4, deflate=True, clean=True) |
| return output_pdf |
|
|
