add option to download results

app.py

@@ -202,8 +202,8 @@ def retrieval():
                     mol = Chem.MolFromSmiles(results.loc[j + 5*i, 'SMILES'])
                     mol_img = Chem.Draw.MolToImage(mol)
                     st.image(mol_img, caption=f"{results.loc[j + 5*i, 'Prediction']:.2f}")
-    
-    
+
+        st.download_button('Download Retrieved Drug Compounds', results.head(selected_k), 'retrieved_drugs.csv')
     
 page_names_to_func = {
     'Retrieval': retrieval,