Update app.py

app.py

@@ -85,7 +85,7 @@ if disease_or_gard_id:
                                 classify_model_vars)
     st.dataframe(df, height=100)
     csv = convert_df(df)
-        st.download_button(
+    st.download_button(
         label="Download epidemiology results for "+disease_or_gard_id+" as CSV",
         data=df.to_csv().encode('utf-8'),
         file_name=disease_or_gard_id+'.csv',
@@ -93,5 +93,7 @@ if disease_or_gard_id:
         )
     #st.dataframe(data=None, width=None, height=None)
     gathered, relevant, epidemiologic = sankey_data
-    
+    if st.button('Display Sankey Diagram'):
+        st.write('Sankey Diagram here')
+        pass
 # st.code(body, language="python")