from pyvis.network import Network


def main():
    net = Network(notebook=False, cdn_resources="local", neighborhood_highlight=True, bgcolor="#222222", font_color="#FAFAFA", layout="hierarchical")

    net.add_nodes([0,1,2,3,4], label=['1','2','3','4','5'],color=['#E0E0FF','#E0FFE0','#E0E0FF','#E0E0FF','#E0E0FF'])
    net.add_edge(0, 1, length=500)
    net.add_edge(1, 2, length=750)
    net.add_edge(2, 3, length=500)
    net.add_edge(3, 4, length=250)
    net.add_edge(2, 4, length=250)
    
    
    net.write_html("network.html", notebook=False)


if __name__ == "__main__":
    main()