from PIL import Image
from imgbeddings import imgbeddings
import sys, os
import json as j

b = imgbeddings()

paths = sys.argv[1:]
batch_size = 8

for i in range(0, len(paths), batch_size):
    fs = paths[i:i+batch_size]

    ims = [Image.open(open(f, "rb")) for f in fs]
    for emb in b.to_embeddings(ims).tolist():
        print(j.dumps(emb))

sys.stderr.write("\n")