add download to pre-loaded results

app.py

@@ -160,6 +160,10 @@ 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.', convert_df(results.head(selected_k)), file_name='retrieved_drugs.csv')
+
+        st.markdown('## TMP title')
                     
     elif query_embedding is not None:
         st.markdown('### Inference')