Update extract_abs.py

extract_abs.py

@@ -328,9 +328,13 @@ def streamlit_extraction(search_term:Union[int,str], maxResults:int, filtering:s
             i+=1
             my_bar.progress(min(round(i*percent_at_step/100,1),1.0))
         
-        sankey_data = (found, relevant, epidemiologic)
         st.write(len(results),'abstracts classified as epidemiological.')
-        return results.sort_values('PROB_OF_EPI', ascending=False), sankey_data
+        
+        sankey_data = (found, relevant, epidemiologic)
+        
+        name_gardID = (search_term_list[-1],GARD_dict[search_term_list[-1]])
+        
+        return results.sort_values('PROB_OF_EPI', ascending=False), sankey_data, name_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