from pyvis.network import Network
import pyvis
from random import randint


pyvis.__version__


def random_colour():
    return f"rgb({randint(0,255)},{randint(0,255)},{randint(0,255)})"

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

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


if __name__ == "__main__":
    main()