pyvisdemo / app.py
simonduerr's picture
Update app.py
c7216c7
raw
history blame
1.53 kB
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()
#need to remove ' from HTML
html = html.replace("'", "\"")
return f"""<iframe style="width: 100%; height: 600px;margin:0 auto" name="result" allow="midi; geolocation; microphone; camera;
display-capture; encrypted-media;" sandbox="allow-modals allow-forms
allow-scripts allow-same-origin allow-popups
allow-top-navigation-by-user-activation allow-downloads" allowfullscreen=""
allowpaymentrequest="" frameborder="0" srcdoc='{html}'></iframe>"""
demo = gr.Interface(
needs_analysis,
inputs=None,
outputs=gr.outputs.HTML(),
title="pyvis_in_gradio",
allow_flagging='never'
)
demo.launch()