tools/record_interestingness.py

"""Template for custom function or Pydantic model."""
import csv
from pathlib import Path

def record_interestingness(titles: list[str], uninteresting: list[bool]) -> None:
    """Label the interestingness of the entries.

    Args:
        titles (list[str]): The titles of the entries to label.
        uninteresting (list[bool]): The list of booleans indicating whether the
        entry is uninteresting.
    """
    assert len(titles) == len(uninteresting)

    # Create result.csv if it doesn't exist
    result_file = Path("result.csv")
    file_exists = result_file.exists()

    with open(result_file, "a", newline="") as f:
        writer = csv.writer(f)

        # Write header if file is new
        if not file_exists:
            writer.writerow(["title", "uninteresting"])

        # Write results
        for title, is_uninteresting in zip(titles, uninteresting, strict=True):
            writer.writerow([title, int(is_uninteresting)])