extract.py

import argparse
from pathlib import Path

from pdf2image import convert_from_path
from PIL import Image, ImageDraw, ImageFont


def pdf_page_to_image(
    pdf_path: Path,
    page_num: int,
    output_path: Path,
    dpi: int = 200,
    scale: float = 0.5,
    font_path: str = "DejaVuSans.ttf",
    font_size: int = 20,
) -> None:
    """
    Convert a specific PDF page into a smaller image with text annotation.
    """

    pages = convert_from_path(
        str(pdf_path), first_page=page_num, last_page=page_num, dpi=dpi
    )
    image = pages[0].convert("RGBA")

    if scale != 1.0:
        image = image.resize(
            (int(image.width * scale), int(image.height * scale)),
            Image.LANCZOS,
        )

    pdf_name = pdf_path.name
    text = f"{pdf_name} - Page {page_num}"

    draw = ImageDraw.Draw(image)
    try:
        font = ImageFont.truetype(font_path, font_size)
    except Exception:
        font = ImageFont.load_default()

    bbox = draw.textbbox((0, 0), text, font=font)
    text_w, text_h = bbox[2] - bbox[0], bbox[3] - bbox[1]

    margin = 10
    x = image.width - text_w - margin
    y = image.height - text_h - margin

    # background rectangle for readability
    draw.rectangle(
        [(x - 5, y - 5), (x + text_w + 5, y + text_h + 5)],
        fill=(200, 200, 200, 255),
    )

    draw.text((x, y), text, font=font, fill="black")

    output_path.parent.mkdir(parents=True, exist_ok=True)
    image.save(output_path, "PNG")


def main() -> None:
    parser = argparse.ArgumentParser()
    parser.add_argument("pdf", type=Path, help="Path to PDF file")
    parser.add_argument("page", type=int, help="1-based page number")
    parser.add_argument(
        "--outdir",
        type=Path,
        default=Path("."),
        help="Output directory for PNG (default: current directory)",
    )
    parser.add_argument(
        "--label",
        type=str,
        default=None,
        help="Custom link text for the markdown image (default: PDF file name)",
    )
    args = parser.parse_args()

    pdf_input: Path = args.pdf.resolve()
    page: int = args.page
    outdir: Path = args.outdir.resolve()

    png_output = outdir / f"{pdf_input.stem}_page_{page}.png"
    pdf_page_to_image(pdf_input, page, png_output)

    # Safer relative path: make relative to cwd if possible
    try:
        rel_path = png_output.relative_to(Path.cwd())
    except ValueError:
        # fallback to absolute path if not inside cwd
        rel_path = png_output

    label = args.label if args.label else pdf_input.name
    enc_path = markdown_encode(rel_path.as_posix())
    print(f"![{label} - Page {page}]({enc_path})")


def markdown_encode(path: str):
    path = path.replace(' ', '%20')
    path = path.replace('(', '%28')
    path = path.replace(')', '%29')
    return path


if __name__ == "__main__":
    main()