added graph

app.py

@@ -191,3 +191,63 @@ with st.container():
         for i in range(0, len(tabs)):
             with tabs[i]:
                 render_tab(trials[i])
+
+
+from utils import get_all_diseases_name, get_most_similar_diseases_from_uri, get_uri_from_name
+
+username = 'demo'
+password = 'demo'
+hostname = os.getenv('IRIS_HOSTNAME', 'localhost')
+port = '1972' 
+namespace = 'USER'
+CONNECTION_STRING = f"iris://{username}:{password}@{hostname}:{port}/{namespace}"
+engine = create_engine(CONNECTION_STRING)
+
+chosen_disease_name = st.selectbox(
+    "Choose a disease",
+    get_all_diseases_name(engine))
+
+st.write("You selected:", chosen_disease_name)
+chosen_disease_uri = get_uri_from_name(engine, chosen_disease_name)
+
+nodes = []
+edges = []
+
+
+nodes.append( Node(id=chosen_disease_uri, 
+    label=chosen_disease_name, 
+    size=25, 
+    shape="circular") 
+)
+
+similar_diseases = get_most_similar_diseases_from_uri(engine, chosen_disease_uri, threshold=0.6)
+print(similar_diseases)
+for uri, name, weight in similar_diseases:
+    nodes.append( Node(id=uri, 
+        label=name, 
+        size=25, 
+        shape="circular") 
+    )
+
+    print(True if float(weight) > 0.7 else False)
+    edges.append( Edge(source=chosen_disease_uri, 
+                   target=uri,
+                   color="red" if float(weight) > 0.7 else "blue",
+                   weight=float(weight)**10,
+                   type="CURVE_SMOOTH"
+                #    type="STRAIGHT"
+                   ) 
+            ) 
+
+config = Config(width=750,
+                height=950,
+                directed=False, 
+                physics=True, 
+                hierarchical=False,
+                collapsible=False,
+                # **kwargs
+                )
+
+return_value = agraph(nodes=nodes, 
+                      edges=edges, 
+                      config=config)