Update epi_pipeline.py

epi_pipeline.py

@@ -874,7 +874,12 @@ def streamlit_extraction(search_term:Union[int,str], maxResults:int, filtering:s
         #Export the name and GARD ID to the ap for better integration on page.
         name = search_term_list[-1].capitalize()
         
-        return results.sort_values('PROB_OF_EPI', ascending=False), sankey_data, name
+        if search_term_list[-1] in GARD_Search.GARD_dict.keys():
+            disease_gardID = (name, GARD_Search.GARD_dict[search_term_list[-1]]
+        else:
+            disease_gardID = (name, None)
+        
+        return results.sort_values('PROB_OF_EPI', ascending=False), sankey_data, disease_gardID
 
 #Identical to search_term_extraction, except it returns a JSON object instead of a df
 def API_extraction(search_term:Union[int,str], maxResults:int, filtering:str, #for abstract search