from icrawler.builtin import BingImageCrawler
from icrawler.builtin import GoogleImageCrawler

# we are building cats image detection that's why we put cat here
# if you want some other images then put that name in classes list
classes = [
    "horse standing in white background",
    "horse running in white background",
    "horse sleeping in white background",
    "horse eating in white background",
    "horse shitting in white background",
    "horse sitting in white background",
    "horse racing in white background",
    "horse head in white background",
    "horse body in white background",
    "horse legs in white background",
    "horse ears in white background",
    "horse eyes in white background",
    "horse tail in white background",
    "any horse in white background",
]
number = 99999
# here root directory is find your root directory there u will find
# new file name data in which all images are saved.
for c in classes:
    # bing_crawler = BingImageCrawler(storage={"root_dir": f'p/{c.replace(" ",".")}'})
    # bing_crawler.crawl(keyword=c, filters=None, max_num=number, offset=0)

    google_crawler = GoogleImageCrawler(storage={"root_dir": f'p/{c.replace(" ",".")}'})
    google_crawler.crawl(
        keyword=c,
        offset=0,
        max_num=99999,
        min_size=(50, 50),
        max_size=None,
    )