Function prepare_ppt_folder

utils/src/experiment/preprocess.py:162–202

(ppt_folder: str, text_model: BGEM3FlagModel, image_model)

def prepare_ppt_folder(ppt_folder: str, text_model: BGEM3FlagModel, image_model):
    if pexists(ppt_folder + "/source.pptx") or not older_than(
        ppt_folder + "/original.pptx"
    ):
        return
    config = Config(rundir=ppt_folder, debug=False)
    presentation = Presentation.from_file(ppt_folder + "/original.pptx", config=config)
    if not os.path.exists(pjoin(ppt_folder, "original_slides")):
        ppt_to_images(presentation.source_file, pjoin(ppt_folder, "original_slides"))
    ppt_image_folder = pjoin(ppt_folder, "source_slides")
    shutil.rmtree(ppt_image_folder, ignore_errors=True)
    shutil.copytree(pjoin(ppt_folder, "original_slides"), ppt_image_folder)

    removed_slides = prs_dedup(presentation, text_model)
    for slide in [slide for slide in presentation.slides if I_dont_want_to_filter_slide(slide)]:
        removed_slides.append(slide)
        presentation.slides.remove(slide)

    for slide in removed_slides:
        os.remove(pjoin(ppt_image_folder, f"slide_{slide.real_idx:04d}.jpg"))
    for err_idx, _ in presentation.error_history:
        os.remove(pjoin(ppt_image_folder, f"slide_{err_idx:04d}.jpg"))
    assert len(presentation) == len(
        [i for i in os.listdir(ppt_image_folder) if i.endswith(".jpg")]
    )
    for i, slide in enumerate(presentation.slides, 1):
        slide.slide_idx = i
        os.rename(
            pjoin(ppt_image_folder, f"slide_{slide.real_idx:04d}.jpg"),
            pjoin(ppt_image_folder, f"slide_{slide.slide_idx:04d}.jpg"),
        )

    check_consistency(presentation.slides, ppt_folder, image_model)
    ImageLabler(presentation, config).caption_images()
    presentation.save(pjoin(ppt_folder, "source.pptx"))
    presentation.save(pjoin(ppt_folder, "template.pptx"), layout_only=True)
    ppt_to_images(
        pjoin(ppt_folder, "template.pptx"),
        pjoin(ppt_folder, "template_images"),
    )
    os.remove(pjoin(ppt_folder, "template.pptx"))


def prepare_induction(induct_id: int, wait: bool = False):