Update app.py

app.py

@@ -121,13 +121,31 @@ if disease_or_gard_id:
                                 NER_pipeline, entity_classes,
                                 extract_diseases,GARD_dict, max_length,
                                 classify_model_vars)
-    st.dataframe(df, height=100)
-    st.markdown(df.columns)
-    st.markdown('''COLUMNS: 
-                   - PROB_OF_EPI: Probability that the paper is an epidemiologic study based on its abstract.
-                   - IsEpi: If it is an epidemiologic study (If PROB_OF_EPI >0.5)
+    st.dataframe(df, height=200)
+    csv = convert_df(df)
+    st.download_button(
+        label="Download epidemiology results for "+disease_or_gard_id+" as CSV",
+        data = csv,
+        file_name=disease_or_gard_id+'.csv',
+        mime='text/csv',
+        )
+    if 'IDS' in list(df.columns):
+        st.markdown('''COLUMNS: \\
+                   - PROB_OF_EPI: Probability that the paper is an epidemiologic study based on its abstract. \\
+                   - IsEpi: If it is an epidemiologic study (If PROB_OF_EPI >0.5) \\
                    - DIS: Rare disease terms or synonyms identified in the abstract from the GARD Dictionary
-                   - IDS: GARD IDs identified in the abstract from the GARD Dictionary 
+                   - IDS: GARD IDs identified in the abstract from the GARD Dictionary \\
+                   - EPI: Epidemiology Types are the metrics used to estimate disease burden such as "incidence", "prevalence rate", or "occurrence"
+                   - STAT: Epidemiology Rates describe how many people are afflicted by a disease.
+                   - DATE: The dates when the epidemiologic studies were conducted
+                   - LOC: Where the epidemiologic studies were conducted.
+                   - SEX: The biological sexes mentioned in the abstract. Useful for diseases that disproportionately affect one sex over the other or may provide context to composition of the study population
+                   - ETHN: Ethnicities, races, and nationalities of those represented in the epidemiologic study.
+                ''')
+    else:
+        st.markdown('''COLUMNS: \\
+                   - PROB_OF_EPI: Probability that the paper is an epidemiologic study based on its abstract. \\
+                   - IsEpi: If it is an epidemiologic study (If PROB_OF_EPI >0.5) \\
                    - EPI: Epidemiology Types are the metrics used to estimate disease burden such as "incidence", "prevalence rate", or "occurrence"
                    - STAT: Epidemiology Rates describe how many people are afflicted by a disease.
                    - DATE: The dates when the epidemiologic studies were conducted
@@ -135,16 +153,7 @@ if disease_or_gard_id:
                    - SEX: The biological sexes mentioned in the abstract. Useful for diseases that disproportionately affect one sex over the other or may provide context to composition of the study population
                    - ETHN: Ethnicities, races, and nationalities of those represented in the epidemiologic study.
                 ''')
-    csv = convert_df(df)
-    st.download_button(
-        label="Download epidemiology results for "+disease_or_gard_id+" as CSV",
-        data = csv,
-        file_name=disease_or_gard_id+'.csv',
-        mime='text/csv',
-        )
     #st.dataframe(data=None, width=None, height=None)
     fig = epi_sankey(sankey_data,disease_or_gard_id)
 
-    #if st.button('Display Sankey Diagram of Automated Search'):
-    st.plotly_chart(fig, use_container_width=True)
-# st.code(body, language="python")
+    st.plotly_chart(fig, use_container_width=True)