import json

with open("../src/app/nodesDataNow.json", "r") as file:
    nodesData = json.load(file)

colors = ["#e75629", "#FD429C", "#ffaf06"]

genres_data_sorted = sorted(nodesData, key=lambda x: x["angle"])


def assign_colors(genres):
    last_color_index = -1
    for genre in genres:
        color_index = (last_color_index + 1) % len(colors)
        genre["color"] = colors[color_index]
        last_color_index = color_index


assign_colors(genres_data_sorted)

output_file_name = "finalGenresV4-5.json"

with open(output_file_name, "w") as outfile:
    json.dump(genres_data_sorted, outfile, indent=2)

print(f"Data has been successfully saved to {output_file_name}.")