import streamlit as st
import json 

with open("taxonomy_v3", "rb") as fp:
    taxonomy = json.load(fp)

taxonomy = taxonomy[:200]

names = [""]
parents = ["root"]

taxonomy_group_label_mapper = {el["group"]: [] for el in taxonomy}

for el in taxonomy:
    taxonomy_group_label_mapper[el["group"]].append(el["label"])

for group, labels in taxonomy_group_label_mapper.items():
    names.append(group)
    parents.append("root")
    for label in labels:
        names.append(label)
        parents.append(group)
        if "-" in label:
            names.append(label.split("-")[0])
            parents.append(label)
            names.append(label.split("-")[1])
            parents.append(label)

fig = px.treemap(
    names=names,
    parents=parents,
)
fig.update_traces(root_color="lightgrey")
fig.update_layout(margin=dict(t=100, l=50, r=50, b=50))


st.title('Musterdatenkatalog')


st.plotly_chart(fig)