graph_intro/streamlit_app.py at main · CodingAndSharing/graph_intro · GitHub

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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
import streamlit as st
from st_link_analysis import st_link_analysis, NodeStyle, EdgeStyle

st.set_page_config(layout="wide")


st.title("🌍 Real World Summary Graph 🌐")
st.write(
    "Click on the nodes to know the information and the challenges."
)


# Sample Data
elements = {
    # "nodes": [
    #     {"data": {"id": 1, "label": "PERSON", "name": "Streamlit"}},
    #     {"data": {"id": 2, "label": "PERSON", "name": "Hello"}},
    #     {"data": {"id": 3, "label": "PERSON", "name": "World"}},
    #     {"data": {"id": 4, "label": "POST", "content": "x"}},
    #     {"data": {"id": 5, "label": "POST", "content": "y"}},
    # ],
    # "edges": [
    #     {"data": {"id": 6, "label": "FOLLOWS", "source": 1, "target": 2}},
    #     {"data": {"id": 7, "label": "FOLLOWS", "source": 2, "target": 3}},
    #     {"data": {"id": 8, "label": "POSTED", "source": 3, "target": 4}},
    #     {"data": {"id": 9, "label": "POSTED", "source": 1, "target": 5}},
    #     {"data": {"id": 10, "label": "QUOTES", "source": 5, "target": 4}},
    # ],

  "nodes": [
    # {"data": { "id": 1, "label": "CORE", "name": "data_harmonized" }}, # CAPITAL leters contains ICONS in ./venv/lib/site-packages/stk-analysis/frontend/build/icons..[TOFIX and push]
    # {"data": { "id": 2, "label": "HEALTH", "name":"patients" }},
    # {"data": { "id": 3, "label":"CURES", "name": "physicians" }},
    # {"data": { "id": 4, "label":"PREVENTION", "name": "institutions" }},
    # {"data": { "id": 5, "label":"PREVENTION", "name": "trial_registries" }},
    # {"data": { "id": 6, "label":"CURES", "name": "pharmacy" }},
    # {"data": { "id": 7, "label":"FINANCIAL", "name": "countries" }},
    # {"data": { "id": 8, "label":"FINANCIAL", "name": "claims" }},
    # {"data": { "id": 9, "label":"HEALTH", "name": "labbiomarkers" }},
    {"data": { "id": 1, "label": "POST", "name": "data_harmonized" }}, # CAPITAL leters contains ICONS in ./venv/lib/site-packages/stk-analysis/frontend/build/icons..[TOFIX and push]
    {"data": { "id": 2, "label": "PERSON", "name":"patients" }},
    {"data": { "id": 3, "label":"POST", "name": "physicians" }},
    {"data": { "id": 4, "label":"POST", "name": "institutions" }},
    {"data": { "id": 5, "label":"POST", "name": "trial_registries" }},
    {"data": { "id": 6, "label":"POST", "name": "pharmacy" }},
    {"data": { "id": 7, "label":"POST", "name": "countries" }},
    {"data": { "id": 8, "label":"POST", "name": "claims" }},
    {"data": { "id": 9, "label":"POST", "name": "labbiomarkers" }},
  ],
  "edges": [
    {"data": { "id": 10, "label": "INTERACT_WITH",  "source": 1, "target": 2 }},
    {"data": {  "id": 11, "label": "INTERACT_WITH", "source": 1, "target": 3}},
    {"data": {  "id": 12, "label": "INTERACT_WITH", "source": 1, "target": 4 }},
    {"data": {  "id": 13, "label": "INTERACT_WITH", "source": 1, "target": 5 }},
    {"data": {  "id": 14, "label": "INTERACT_WITH", "source": 2, "target": 6 }},
    {"data": {  "id": 15, "label": "INTERACT_WITH", "source": 3, "target": 6 }},
    {"data": {  "id": 16, "label": "INTERACT_WITH", "source": 3, "target": 7 }},
    {"data": { "id": 17, "label": "INTERACT_WITH",  "source": 5, "target": 7 }},
    {"data": { "id": 18, "label": "INTERACT_WITH",  "source": 5, "target": 8 }},
    {"data": { "id": 19, "label": "INTERACT_WITH",  "source": 4, "target": 8 }},
    {"data": { "id": 20, "label": "INTERACT_WITH",  "source": 4, "target": 9 }},
    {"data": { "id": 21, "label": "INTERACT_WITH",  "source": 2, "target": 9 }},
  ]


}

# Style node & edge groups
node_styles = [
    # NodeStyle("CORE", "#FF7F3E", "name", "core"),
    # NodeStyle("HEALTH", "#2A629A", "name", "health"),
    # NodeStyle("CURES", "#2A079A", "name", "cures"),
    # NodeStyle("PREVENTION", "#2A620A", "name", "prevention"),
    # NodeStyle("FINANCIAL", "#1A629A", "name", "financial")
    NodeStyle("PERSON", "#FF7F3E", "name", "person"),
    NodeStyle("POST", "#2A629A", "content", "description"),
]

edge_styles = [
    EdgeStyle("INTERACT_WITH", caption='label', directed=True),
    EdgeStyle("FOLLOWS", caption='label', directed=True),
    # EdgeStyle("INTERACT_WITH", caption='label', directed=True),
]

# Render the component
st.markdown("### st-link-analysis: Example")
st_link_analysis(elements, "cose", node_styles, edge_styles)