changed messages to the toast ones

app.py

@@ -325,6 +325,7 @@ if submit_side or submit_main:
         batch_size = 250
         random_seed = 44
 
+    st.toast('Optimisation is started. The progress will be printed below')
     X, Y, rf_model, vqgae_model, ordering_model = load_data(batch_size)
     assert X.shape == (603, 4096)
     start_time = time()
@@ -358,7 +359,9 @@ if submit_side or submit_main:
 
     with st.spinner('Getting unique solutions'):
         unique_solutions = list(set(tuple(s) for s in ga_instance.solutions))
-    st.success(f'{len(unique_solutions)} solutions were obtained')
+
+    st.toast(f'{len(unique_solutions)} latent vectors were obtained')
+    # st.success(f'{len(unique_solutions)} solutions were obtained')
 
     scores = {
         "rf_score": [],
@@ -399,7 +402,8 @@ if submit_side or submit_main:
     chosen_ids = chosen_gen.index.to_list()
     chosen_solutions = np.array([unique_solutions[ind] for ind in chosen_ids])
     gen_frag_inds = frag_counts_to_inds(chosen_solutions, max_atoms=51)
-    st.info(f'The number of chosen solutions is {gen_frag_inds.shape[0]}', icon="ℹ️")
+    st.toast(f'The number of chosen latent vectors is {gen_frag_inds.shape[0]}')
+    # st.info(f'The number of chosen solutions is {gen_frag_inds.shape[0]}', icon="ℹ️")
 
     gen_molecules = []
     results = {"smiles": [], "ordering_score": [], "validity": []}
@@ -451,6 +455,7 @@ if submit_side or submit_main:
     mins_used = int(round(time_used // 60, 0))
     seconds_used = int(round(time_used % 60, 0))
     st.success(f'{filtered_gen_stats.shape[0]} valid solutions were obtained in {mins_used} mins {seconds_used} secs')
+    st.balloons()
 
     st.subheader('Generation results', divider='rainbow')
     st.dataframe(filtered_gen_stats)