from PIL import Image
import numpy as np
import requests
from bs4 import BeautifulSoup as bs
from io import BytesIO, StringIO


image_list = []
thumbnail_list = []

def get_image():

    source = requests.get('https://www.pexels.com/search/dunes/')
    data = source.text
    soup = bs(data, 'html.parser')

    # img = Image.open(StringIO(data.content))
    # img.show()


    for image in soup.find_all('img'):
        image_source = image.get('src')
        image_list.append(image_source)


    print(image_list)
    thumbnail_size = 100
    grid = int(np.ceil(np.sqrt(len(image_list)))) * thumbnail_size
    new_image = Image.new('RGB', (grid, grid))

    for file in range(len(image_list)):
        r = requests.get(image_list[file])

        with open(new_image, 'wb') as f:

            index = -1

            # for image in thumbnail_list:
            for i in range(0, grid, 100):
                for j in range(0, grid, 100):
                    if index < len(image_list) - 1:
                        index += 1
                        new_image.paste(r.content, (j, i))
           # f.write(r.content)
                file += 1

    # for image in image_list:
    #     try:
    #         url = image['src']
    #         im = Image.open(url)
    #         im.show()
    #     except:
    #         TypeError

    #print(image_list)