convert df to csv before download

app.py

@@ -52,6 +52,12 @@ def about_page():
     )
 
     st.image('figures/hyper-dti.png', caption='Overview of HyperPCM architecture.')
+
+
+@st.cache
+def convert_df(df):
+    # IMPORTANT: Cache the conversion to prevent computation on every rerun
+    return df.to_csv().encode('utf-8')
     
 
 def retrieval():
@@ -203,7 +209,9 @@ def retrieval():
                     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')
+        st.download_button('Download retrieved drug compounds.', convert_df(results.head(selected_k)), file_name='retrieved_drugs.csv')
+
+        st.markdown('## TMP title')
     
 page_names_to_func = {
     'Retrieval': retrieval,