-
Notifications
You must be signed in to change notification settings - Fork 0
/
NFTCombiner.py
36 lines (28 loc) · 1.11 KB
/
NFTCombiner.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import os
from ast import For
from multiprocessing.dummy import Array
from PIL import Image
skintoneAttributes = ["shirt", "mouth", "accessory", "eyebrows", "eyes" ]
skintoneNames = ["skin-tone-1"]
attributes = ["background", "skin-tone-1", "eyewear", "hair-hat"]
attributesCount = len(attributes)
image = Image.new('RGB', (1200, 1200))
image.save('combo.png', "PNG")
def process_folder(directory, prevImage):
print('Getting items in ' + directory)
for root, dirs, files in os.walk(directory):
for file in files:
if file.endswith('.png'):
# For now, we'll just go 1 level deep
print(os.getcwd())
newImage = Image.open('./' + directory + '/' + file)
prevImage.paste(newImage, (0,0), newImage)
prevImage.save('combo.png', "PNG")
break
for folder in attributes:
directory = 'images/' + folder
if folder in skintoneNames:
for skintoneFolder in skintoneAttributes:
process_folder(directory + '/' + skintoneFolder, image)
else:
process_folder(directory, image)