def process_images()

in distilvit/pexels.py [0:0]

• 21 lines of code
• 5 McCabe index (conditional complexity)

def process_images(image_ids, generator, image_by_id, pbar):
    cache = OpenaiCache()
    cached_images = cache.get(image_ids)
    new_images = []
    new_images_ids = []

    for image_id in image_ids:
        image = Image.open(os.path.join("images", image_id)).convert("RGB")
        byte_arr = BytesIO()
        image.save(byte_arr, format="JPEG")
        image_by_id[image_id] = byte_arr.getvalue()

        if image_id not in cached_images:
            new_images.append(image)
            new_images_ids.append(image_id)

    if new_images_ids:
        generated = generator.generate(new_images, new_images_ids)
        generated = generated["images"]
        cache.set(new_images_ids, generated)

        for id, payload in zip(new_images_ids, generated):
            cached_images[id] = payload

    pbar.update(5)
    return cached_images