| """ |
| PDF I/O utilities - Load and render PDFs |
| Placeholder for Task 2 |
| """ |
|
|
| from typing import List, Tuple |
| from PIL import Image |
| import io |
|
|
|
|
| def load_pdf_pages(pdf_bytes: bytes, dpi: int = 150, max_pages: int = 10) -> List[Image.Image]: |
| """ |
| Load PDF and render each page to PIL Image. |
| |
| Args: |
| pdf_bytes: PDF file as bytes |
| dpi: Resolution for rendering (default 150) |
| max_pages: Maximum pages to render (default 10) |
| |
| Returns: |
| List of PIL Images, one per page |
| """ |
| try: |
| import fitz |
|
|
| doc = fitz.open(stream=pdf_bytes, filetype="pdf") |
| pages = [] |
|
|
| for i in range(min(len(doc), max_pages)): |
| page = doc[i] |
| |
| pix = page.get_pixmap(matrix=fitz.Matrix(dpi/72, dpi/72)) |
| img = Image.frombytes("RGB", (pix.width, pix.height), pix.samples) |
| pages.append(img) |
|
|
| doc.close() |
| return pages |
|
|
| except ImportError: |
| |
| print("⚠️ PyMuPDF not available, returning placeholder") |
| img = Image.new("RGB", (800, 1000), color=(200, 200, 200)) |
| return [img] * 3 |
|
|
|
|
| def get_page_count(pdf_bytes: bytes) -> int: |
| """Get total page count of PDF.""" |
| try: |
| import fitz |
| doc = fitz.open(stream=pdf_bytes, filetype="pdf") |
| count = len(doc) |
| doc.close() |
| return count |
| except Exception: |
| return 0 |
|
|
