Prevent duplicate titles

lecture2notes/end_to_end/summarization_approaches.py

@@ -1,7 +1,7 @@
 import json
 import logging
 import math
-from collections import OrderedDict, namedtuple
+from collections import OrderedDict, defaultdict, namedtuple
 from functools import partial
 from operator import attrgetter
 from time import time, sleep
@@ -963,6 +963,7 @@ def structured_joined_sum(
  transcript_before_slides = transcript_before_slides.strip()
  final_dict = OrderedDict({"Preface": {"transcript": transcript_before_slides}})
 
+ all_titles = defaultdict(int)
  no_conclusion = False
  for idx, slide in tqdm(
  enumerate(ssa), total=len(ssa), desc="Grouping Slides and Transcript"
@@ -1012,6 +1013,12 @@ def structured_joined_sum(
  title = " ".join([slide["text"][line] for line in title_lines]).strip()
  if not title:
  title = "Slide {}".format(idx + 1)
+ else:
+ if title in all_titles:
+ all_titles[title] += 1
+ title = f"{title} ({all_titles[title]})"
+ else:
+ all_titles.add(title)
 
  current_slide_timestamp_seconds = ssa[idx]["frame_number"] * frame_every_x