Update app.py

app.py

@@ -117,12 +117,16 @@ st.markdown("Examples of rare diseases include [**Fellman syndrome**](https://ra
 if disease_or_gard_id:
     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,
+                                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
+    #if the user input does not have a number in it (i.e. weak proxy for if it is a GARD ID), then preserve the user input as the disease term.
+    if not bool(re.search(r'\d', disease_or_gard_id)):
+        disease = disease_or_gard_id
+        
     st.download_button(
         label="Download epidemiology results for "+disease+" as CSV",
         data = csv,