Update app.py

app.py

@@ -76,9 +76,9 @@ def epi_sankey(sankey_data):
         node = dict(
           pad = 15,
           thickness = 20,
-          line = dict(color = "black", width = 0.5),
+          line = dict(color = "white", width = 0.5),
           label = ["PubMed IDs Gathered", "Irrelevant Abstracts","Relevant Abstracts Gathered","Epidemiologic Abstracts","Not Epidemiologic"],
-          color = "blue"
+          color = "violet"
         ),
         #label = ["A1", "A2", "B1", "B2", "C1", "C2"]
         link = dict(
@@ -86,6 +86,11 @@ def epi_sankey(sankey_data):
           target = [2, 1, 3, 4],
           value = [relevant, gathered-relevant, epidemiologic, relevant-epidemiologic]
       ))])
+    fig.update_layout(
+    hovermode = 'x',
+    title="Search for the Epidemiology of "+disease_or_gard_id,
+    font=dict(size = 10, color = 'black'),
+)
     
     return fig
 
@@ -119,8 +124,8 @@ if disease_or_gard_id:
         mime='text/csv',
         )
     #st.dataframe(data=None, width=None, height=None)
-    fig = epi_sankey(sankey_data)
+    fig = epi_sankey(sankey_data,disease_or_gard_id)
         
-    if st.button('Display Sankey Diagram'):
+    if st.button('Display Sankey Diagram of Automated Search'):
         st.plotly_chart(fig, use_container_width=True)
 # st.code(body, language="python")