forked from truckyducky/app
-
Notifications
You must be signed in to change notification settings - Fork 0
/
NIDAmodified.py
59 lines (46 loc) · 2.84 KB
/
NIDAmodified.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
import streamlit as st
import streamlit.components.v1 as components
import pandas as pd
import networkx as nx
from pyvis.network import Network
from IPython.core.display import display, HTML
net = Network(notebook = True, cdn_resources="remote",height="900px", width="90%", select_menu = True, filter_menu=True,font_color="white", bgcolor = "black")
net.add_node('Node 21 - NIDA Modified Assist Tool - 2 [NIDA Assist-2 Modified]',
title = 'NIDA Modified Assist Tool - 2 [NIDA Assist-2 Modified]', color = '#e07c3e')
net.add_nodes(['SPRINT: Signature for Pain Recovery IN Teens',
'SurgeryPal',
'Culturally adapted mobile treatment of chronic pain in adolescent survivors of pediatric bone sarcoma',
'Integrative Training Program for Pediatric Sickle Cell Pain',
'Latinx Children and Surgery'],
label = ['SPRINT: Signature for Pain Recovery IN Teens',
'SurgeryPal',
'Culturally adapted mobile treatment of chronic pain in adolescent survivors of pediatric bone sarcoma',
'Integrative Training Program for Pediatric Sickle Cell Pain',
'Latinx Children and Surgery'],
title = ['SPRINT: Signature for Pain Recovery IN Teens',
'SurgeryPal',
'Culturally adapted mobile treatment of chronic pain in adolescent survivors of pediatric bone sarcoma',
'Integrative Training Program for Pediatric Sickle Cell Pain',
'Latinx Children and Surgery'],
shape = ['text','text','text','text','text']
)
#'Node 21 - NIDA Modified Assist Tool - 2 [NIDA Assist-2 Modified]'
net.add_edge('Node 21 - NIDA Modified Assist Tool - 2 [NIDA Assist-2 Modified]','SPRINT: Signature for Pain Recovery IN Teens')
net.add_edge('Node 21 - NIDA Modified Assist Tool - 2 [NIDA Assist-2 Modified]','SurgeryPal')
net.add_edge('Node 21 - NIDA Modified Assist Tool - 2 [NIDA Assist-2 Modified]','Culturally adapted mobile treatment of chronic pain in adolescent survivors of pediatric bone sarcoma')
net.add_edge('Node 21 - NIDA Modified Assist Tool - 2 [NIDA Assist-2 Modified]','Integrative Training Program for Pediatric Sickle Cell Pain')
net.add_edge('Node 21 - NIDA Modified Assist Tool - 2 [NIDA Assist-2 Modified]','Latinx Children and Surgery')
net.repulsion(spring_strength = 0)
net.show("network.html")
# Save and read graph as HTML file (on Streamlit Sharing)
try:
path = '/tmp'
net.save_graph(f'{path}/pyvis_graph.html')
HtmlFile = open(f'{path}/pyvis_graph.html', 'r', encoding='utf-8')
# Save and read graph as HTML file (locally)
except:
path = '/html_files'
net.save_graph(f'{path}/pyvis_graph.html')
HtmlFile = open(f'{path}/pyvis_graph.html', 'r', encoding='utf-8')
# Load HTML file in HTML component for display on Streamlit page
components.html(HtmlFile.read(), height=1000, width = 1000)