from PIL import Image, ImageFont, ImageDraw <br>
import emoji<br>
import random<br>
from csv import writer<br>
// Chosen 5 emojis - 5 classes<br>
emotions = {1:':grinning_face:',
            2:':loudly_crying_face:',3:':astonished_face:',
            4:':winking_face:',5:':face_with_medical_mask:'}<br>
// Different sizes<br>
f = [25,20,28,30,32]<br>
n = 0<br>
// Generation of 1000 Images<br>
for i in range(1000):<br>
    elem = [0,0,0,0,0]<br>
    lab_loc = {}<br>
    num = random.randint(1,3)<br>
    img = Image.new('RGB', (64, 64), color='white')<br>
    d = ImageDraw.Draw(img)<br>
// Different Locations<br>
    loc = [[(0,0),(30,0),(30,30)],[(0,10),(20,30),(40,20)]]<br>
    e = [1, 2, 3, 4, 5]<br>
    binary_emo = []<br>
    for j in range(num):<br>
        val = random.choice(f)<br>
        l = random.randint(0,1)<br>
        font = ImageFont.truetype("C:/Users/U S Matharu/Downloads/Symbola/Symbola_hint.ttf", val)<br>
        emo_num = random.choice(e)<br>
        e.remove(emo_num)<br>
        elem[emo_num-1] = 1<br>
        emo = str(emoji.emojize(emotions[emo_num]))<br>
        d.text(loc[l][j],emo,(0,0,0), font=font)<br>
        lab_loc[emotions[emo_num][1:-1]] = [loc[l][j],val,emo_num]<br>
        binary_emo.append(emotions[emo_num])<br>

    img.save('test'+str(i)+'.png','PNG')<br>

// To generate the CSV file for the model<br>
    l = 1<br>
    location_data = [0,0,0,0,0]<br>
    for k,v in lab_loc.items():<br>
        location_data[v[2]-1] = l<br>
        l += 1<br>

    labels = []<br>
    for b in binary_emo:<br>
        labels.append(b[1:-1])<br>

    with open('locations.csv', 'a+', newline='') as write_obj:<br>
            csv_writer = writer(write_obj)<br>
            list_elem = ['test'+str(i)+'.png',labels,location_data[0],location_data[1],location_data[2],
                         location_data[3],location_data[4]]<br>
            csv_writer.writerow(list_elem)<br>
    n += 1<br>