import os
import scandir 
import json
import random


data = dict()
data['database'] = dict()

categories = list()

print("{} validation instances".format(len(data['database'])))
c = 0
with open("moments_train.csv", 'w') as f:
    for root, dirs, files in scandir.walk("./training"):
        print(c)
        c += 1
        label = root.strip().split('/')[-1]
        if files:
            categories.append(label)
        for fi in files:
            f.write("{},{}\n".format(os.path.join(root, fi), label))
            data['database'][fi] = {'subset': 'training', 'annotations': {'label': label}}

with open("categories.txt", 'w') as f:
    categories = sorted(categories)
    for i, label in enumerate(categories):
        f.write("{},{}\n".format(label, i))

with open("annotation.json", 'w') as f:
    json.dump(data, f)