image_scraper.py

from selenium import webdriver
from selenium.webdriver.common.by import By
import requests
import io
from PIL import Image
import time

PATH = "C:\\Py Projects\\Machine Learning\\Functional Group Detector\\chromedriver.exe"

wd = webdriver.Chrome(PATH)

def get_images_from_google(wd, delay, max_images):
	def scroll_down(wd):
		wd.execute_script("window.scrollTo(0, document.body.scrollHeight);")
		time.sleep(delay)

	url = "https://www.google.com/search?q=randomorganic+compounds&tbm=isch&ved=2ahUKEwit2a2OlZ38AhX-CUQIHXZxD0MQ2-cCegQIABAA&oq=randomorganic+compounds&gs_lcp=CgNpbWcQAzoECAAQQzoFCAAQgARQ_QFYnhRg1xVoAHAAeAGAAdMNiAHKH5IBCTIuMi4xLjgtMpgBAKABAaoBC2d3cy13aXotaW1nwAEB&sclient=img&ei=vaisY-2GNv6TkPIP9uK9mAQ"
	wd.get(url)

	image_urls = set()
	skips = 0

	while len(image_urls) + skips < max_images:
		scroll_down(wd)


		thumbnails = wd.find_elements(By.CLASS_NAME, "Q4LuWd") 
        
		for img in thumbnails[len(image_urls) + skips:max_images]:
			try:
				img.click()
				time.sleep(delay)
			except:
				continue

			images = wd.find_elements(By.CLASS_NAME, "n3VNCb")
			for image in images:
				if image.get_attribute('src') in image_urls:
					max_images += 1
					skips += 1
					break

				if image.get_attribute('src') and 'http' in image.get_attribute('src'):
					image_urls.add(image.get_attribute('src'))
					print(f"Found {len(image_urls)}")

	return image_urls


def download_image(download_path, url, file_name):
	try:
		image_content = requests.get(url).content
		image_file = io.BytesIO(image_content)
		image = Image.open(image_file)
		file_path = download_path + file_name

		with open(file_path, "wb") as f:
			image.save(f, "PNG")

		print("Success")
	except Exception as e:
		print('FAILED -', e)

urls = get_images_from_google(wd, 1, 5)
print (urls)

for i, link in enumerate(urls):
	download_image("C:\\Py Projects\\Machine Learning\\test_images\\", link, str(i+5) +".png")

wd.quit()