Update app.py

app.py

@@ -9,6 +9,7 @@ import streamlit as st
 import spacy
 import tensorflow as tf
 import pickle
+import plotly.graph_objects as go
 
 ########## Title for the Web App ##########
 st.markdown('''<img src="https://huggingface.co/spaces/ncats/EpiPipeline4GARD/resolve/main/NCATS_logo.png" alt="National Center for Advancing Translational Sciences Logo" width=550>''',unsafe_allow_html=True)
@@ -22,11 +23,11 @@ st.markdown(
     """
     <style>
     [data-testid="stSidebar"][aria-expanded="true"] > div:first-child {
-        width: 200px;
+        width: 250px;
     }
     [data-testid="stSidebar"][aria-expanded="false"] > div:first-child {
-        width: 200px;
-        margin-left: -300px;
+        width: 250px;
+        margin-left: -350px;
     }
     </style>
     """,
@@ -111,8 +112,8 @@ if disease_or_gard_id:
         mime='text/csv',
         )
     #st.dataframe(data=None, width=None, height=None)
-    gathered, relevant, epidemiologic = sankey_data
+    fig = epi_sankey(sankey_data)
+        
     if st.button('Display Sankey Diagram'):
-        fig = epi_sankey(sankey_data)
         st.plotly_chart(fig, use_container_width=True)
 # st.code(body, language="python")