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")
    net.repulsion(spring_length=300)

    net.add_nodes([1,2,3,4,5,6,7,8,9,10], label=["1","2","3","4","5","6","7","8","9","10"],color=[random_colour() for _ in range(10)])
    net.add_edges([(1, 2, 10), (1, 3, 5), (1, 4, 6), (2, 3, 10), (2, 5, 1), (5, 6, 9), (7, 8, 7), (8, 9, 7), (9, 7, 8), (1, 7, 7), (9, 10, 2)])
    
    net.show("network.html", notebook=False)


if __name__ == "__main__":
    main()