import pandas as pd

from google_images_download import google_images_download

df = pd.read_csv("pokemon.csv")

response = google_images_download.googleimagesdownload()

for pokemon in df["identifier"][:251]:
    absolute_image_paths = response.download(
        {
            "keywords": pokemon,
            "limit": 250,
            "chromedriver": "/usr/lib/chromium-browser/chromedriver",
            # This needs to be changed based on the computer trying to download the images
            "format": "jpg"
        }
    )

# TODO: Need to clean data up here.... really should be added to another class as well you lazy asshole