import gradio as gr
import os
from PIL import Image


folder = "images"

img_list = os.listdir(folder)

#looks at the colors in an image and saves the findings to a file
with open("imageRGBs.txt", 'w') as color_file:
    for image_name in img_list:
        im = Image.open(folder+"/"+image_name)

        #Returns the three main colors of the image
        im2 = im.quantize(colors=3, method=2)
        rgb = im2.getpalette()[:9]
        b = list(map(str, rgb))
        color_file.write(' '.join(b) + ";" + image_name  + "\n")