parse_inews

topic_segmentation.news_dataset.parse_inews

Parse saved pages in data/pages/inews_subheadings/ into data/outputs/inews_segments.json.

is_noise_heading(text: str) -> bool

Source code in src/topic_segmentation/news_dataset/parse_inews.py

def is_noise_heading(text: str) -> bool:
    t = text.lower().strip()
    return len(t) < 3 or any(pat in t for pat in NOISE_PATTERNS)

is_skip_container(el) -> bool

Source code in src/topic_segmentation/news_dataset/parse_inews.py

def is_skip_container(el) -> bool:
    el_classes = set(el.get("class") or [])
    if el_classes & SKIP_CLASSES:
        return True
    if el.name in ("aside", "form", "nav"):
        return True
    return False

is_link_only_list(el) -> bool

Source code in src/topic_segmentation/news_dataset/parse_inews.py

def is_link_only_list(el) -> bool:
    if el.name not in ("ul", "ol"):
        return False
    items = el.find_all("li", recursive=False)
    if not items:
        return False
    for li in items:
        li_text = li.get_text(" ", strip=True)
        a_text = "".join(a.get_text(" ", strip=True) for a in li.find_all("a"))
        if not li_text or not a_text:
            return False
        if li_text == a_text:
            continue
        if len(a_text) / len(li_text) >= 0.6:
            continue
        return False
    return True

is_trailing_element(el) -> bool

Source code in src/topic_segmentation/news_dataset/parse_inews.py

def is_trailing_element(el) -> bool:
    if el.name in ("h2", "h3", "h4"):
        t = el.get_text(" ", strip=True).lower().rstrip(":")
        if t in TRAILING_HEADINGS:
            return True
    text_lower = el.get_text(" ", strip=True).lower()
    if any(text_lower.startswith(p) for p in TRAILING_TEXT_PREFIXES):
        return True
    return False

clean_inner_html(el) -> str

Source code in src/topic_segmentation/news_dataset/parse_inews.py

def clean_inner_html(el) -> str:
    el_copy = copy.deepcopy(el)
    for tag in el_copy.find_all(True):
        if tag.name == "script":
            tag.decompose()
            continue
        for attr in STRIP_ATTRS:
            tag.attrs.pop(attr, None)
        if tag.name == "a" and (tag.get("href") or "").startswith("/"):
            tag["href"] = "https://inews.co.uk" + tag["href"]
            tag["target"] = "_blank"
            tag["rel"] = "noopener"
    return el_copy.decode_contents().strip()

collect_elements(container) -> list

Source code in src/topic_segmentation/news_dataset/parse_inews.py

def collect_elements(container) -> list:
    seen_ids: set = set()
    elements: list = []

    def walk(node):
        for el in node.children:
            if not hasattr(el, "name") or el.name is None:
                continue
            if is_skip_container(el):
                continue
            eid = id(el)
            if eid in seen_ids:
                continue
            seen_ids.add(eid)
            if el.name in BLOCK_TAGS:
                # Recurse into blockquote and figure wrappers containing headings.
                if el.name in ("blockquote", "figure") and el.find(["h2", "h3", "h4"]):
                    walk(el)
                elif not is_link_only_list(el):
                    elements.append(el)
            else:
                walk(el)

    walk(container)
    return elements

process_article(aid: str, raw_html: str, meta: dict)

Source code in src/topic_segmentation/news_dataset/parse_inews.py

def process_article(aid: str, raw_html: str, meta: dict):
    soup = BeautifulSoup(ftfy.fix_text(raw_html), "html.parser")

    article = soup.find("article")
    if not article:
        return None

    content = article.find("div", class_="article-content")
    if not content:
        return None

    # Title
    h1 = article.find("h1")
    title = h1.get_text(" ", strip=True) if h1 else meta.get("title", "")

    # Standfirst from header excerpt
    standfirst_text = None
    header = article.find("header")
    if header:
        excerpt = header.find(class_="inews__post-excerpt")
        if excerpt:
            standfirst_text = excerpt.get_text(" ", strip=True)

    # Hero image from header figure
    hero_img_src = None
    if header:
        fig = header.find("figure", class_="inews__post-header__media")
        if fig:
            img = fig.find("img")
            if img:
                hero_img_src = img.get("src", "")

    sentences, html_parts, labels, title_labels = [], [], [], []

    if standfirst_text:
        sentences.append(standfirst_text)
        html_parts.append({"tag": "p", "html": standfirst_text})
        labels.append(0)
        title_labels.append(1)

    if hero_img_src:
        sentences.append("")
        html_parts.append({"tag": "figure", "html": f'<img src="{hero_img_src}" style="max-width:100%">'})
        labels.append(0)
        title_labels.append(0)

    first_elem = not (standfirst_text or hero_img_src)

    for el in collect_elements(content):
        if is_trailing_element(el):
            break

        # Figures with images
        if el.name == "figure":
            img = el.find("img")
            if not img:
                continue
            src = img.get("src", "")
            if not src:
                continue
            figcap = el.find("figcaption")
            cap_text = figcap.get_text(" ", strip=True) if figcap else ""
            cap_html = f"<figcaption>{cap_text}</figcaption>" if cap_text else ""
            sentences.append(cap_text)
            html_parts.append({"tag": "figure", "html": f'<img src="{src}" style="max-width:100%">{cap_html}'})
            labels.append(0)
            title_labels.append(0)
            first_elem = False
            continue

        text = el.get_text(" ", strip=True)
        if not text:
            continue

        if el.name in ("h2", "h3", "h4") and is_noise_heading(text):
            continue
        if el.name == "p" and any(text.lower().startswith(p) for p in SKIP_PARA_PREFIXES):
            continue

        is_heading = el.name in ("h2", "h3", "h4") and len(text) <= 200

        sentences.append(text)
        html_parts.append({"tag": el.name, "html": clean_inner_html(el)})
        if is_heading:
            labels.append(0 if first_elem else 1)
            title_labels.append(1)
        else:
            labels.append(0)
            title_labels.append(0)
        first_elem = False

    # Remove trailing headings without following content.
    while sentences and title_labels[-1] == 1:
        sentences.pop()
        html_parts.pop()
        labels.pop()
        title_labels.pop()

    if not sentences:
        return None

    return {
        "articleId": aid,
        "outlet": "INews",
        "title": title,
        "url": meta.get("url", ""),
        "sentences": sentences,
        "html_parts": html_parts,
        "labels": labels,
        "title_labels": title_labels,
        "num_sections": labels.count(1) + 1,
    }

main()

Source code in src/topic_segmentation/news_dataset/parse_inews.py

def main():
    with open(ROOT / "data/articles.json", encoding="latin-1") as f:
        all_meta = {a["articleId"]: a for a in json.load(f)}

    results = []
    seen_content: set = set()
    for html_file in sorted((ROOT / "data/pages/inews_subheadings").glob("*.html")):
        aid = html_file.stem
        raw = html_file.read_text(encoding="utf-8")
        rec = process_article(aid, raw, all_meta.get(aid, {}))
        if not rec:
            continue
        content = (rec.get("title", "") + "".join(rec["sentences"]))
        if content in seen_content:
            continue
        seen_content.add(content)
        results.append(rec)

    results = [r for r in results if r["num_sections"] >= 2]
    print(f"Processed {len(results)} articles")

    (ROOT / "data/outputs").mkdir(parents=True, exist_ok=True)
    seg_path = ROOT / "data/outputs/inews_segments.json"
    with open(seg_path, "w", encoding="utf-8") as f:
        json.dump(results, f, ensure_ascii=False, indent=2)
    print(f"Saved {seg_path}")