import gradio as gr from pyvis.network import Network import networkx as nx nx_graph = nx.cycle_graph(10) nx_graph.nodes[1]['title'] = 'Number 1' nx_graph.nodes[1]['group'] = 1 nx_graph.nodes[3]['title'] = 'I belong to a different group!' nx_graph.nodes[3]['group'] = 10 nx_graph.add_node(20, size=20, title='couple', group=2) nx_graph.add_node(21, size=15, title='couple', group=2) nx_graph.add_edge(20, 21, weight=5) nx_graph.add_node(25, size=25, label='lonely', title='lonely node', group=3) # graph above refer to: https://pyvis.readthedocs.io/en/latest/tutorial.html#networkx-integration # nt_notebook = Network('500px', '500px', notebook=True, cdn_resources='remote') # populates the nodes and edges data structures # nt_notebook.from_nx(nx_graph) # nt_notebook.show('nx.html') def needs_analysis(): nt = Network() nt.from_nx(nx_graph) html = nt.generate_html() print(html) return html demo = gr.Interface( needs_analysis, inputs=None, outputs=gr.outputs.HTML(), title="pyvis_in_gradio", allow_flagging='never' ) demo.launch()