def fetch_images()

in distilvit/pexels.py [0:0]

• 12 lines of code
• 2 McCabe index (conditional complexity)

def fetch_images(query):
    index = 1
    for image in tqdm(
        fetch_images_from_pexels(PEXELS_API_KEY, per_page, query, start_page=1),
        total=1000,
        desc=f"Fetching images from Pexels for {query}",
    ):
        image_url = image["src"]["original"]
        response = requests.get(image_url)
        pil_image = resize_image(Image.open(BytesIO(response.content)).convert("RGB"))
        pil_image.save(os.path.join("images", f"{query}-{index}.jpg"), format="JPEG")
        index += 1