Update app.py

app.py

@@ -114,20 +114,23 @@ loaded.empty()
 st.markdown("Examples of rare diseases include [**Fellman syndrome**](https://rarediseases.info.nih.gov/diseases/1/gracile-syndrome), [**Classic Homocystinuria**](https://rarediseases.info.nih.gov/diseases/6667/classic-homocystinuria), [**7383**](https://rarediseases.info.nih.gov/diseases/7383/phenylketonuria), and [**GARD:0009941**](https://rarediseases.info.nih.gov/diseases/9941/fshmd1a). A full list of rare diseases tracked by the NIH Genetic and Rare Diseases Information Center (GARD) can be found [here](https://rarediseases.info.nih.gov/diseases/browse-by-first-letter).")
 
 if disease_or_gard_id:
-    df, sankey_data = extract_abs.streamlit_extraction(disease_or_gard_id, max_results, filtering,
+    df, sankey_data, name_gardID = extract_abs.streamlit_extraction(disease_or_gard_id, max_results, filtering,
                                 NER_pipeline, entity_classes,
                                 extract_diseases,GARD_dict, max_length,
                                 classify_model_vars)
     df.replace(to_replace='None', value="None")
     st.dataframe(df, height=200)
     csv = convert_df(df)
+    disease, gardID = name_gardID
     st.download_button(
-        label="Download epidemiology results for "+disease_or_gard_id+" as CSV",
+        label="Download epidemiology results for "+disease+" as CSV",
         data = csv,
-        file_name=disease_or_gard_id+'.csv',
+        file_name=disease+'.csv',
         mime='text/csv',
         )
     
+    st.markdown('Search for ['+disease+'](https://rarediseases.info.nih.gov/diseases/'+str(re.sub('GARD:|0','',gardID))+'/'+str('-'.join(disease.split())))
+    
     fig = epi_sankey(sankey_data,disease_or_gard_id)
     st.plotly_chart(fig, use_container_width=True)