fixed extraction of examples

app.py

@@ -18,7 +18,7 @@ def render_svg(svg_string):
     """Renders the given svg string."""
     c = st.container()
     with c:
-        html(svg_string)
+        html(svg_string, height=300, scrolling=True)
 
 
 # define groups to filter
@@ -330,35 +330,36 @@ if submit:
 
     filtered_gen_stats = valid_gen_stats.iloc[filtered_indices]
 
-    st.subheader('Generation results', divider='rainbow')
-    st.dataframe(filtered_gen_stats)
-    st.download_button(
-        label="Download results as CSV",
-        data=convert_df(filtered_gen_stats),
-        file_name='vqgae_tubulin_inhibitors_valid.csv',
-        mime='text/csv',
-    )
-
-    st.subheader('Examples of generated molecules')
-    examples_smiles = filtered_gen_stats.sort_values(by=["similarity_score"], ascending=False).iloc[:6].smiles.to_list()
-    examples = []
-    for smi in examples_smiles:
-        mol = smiles(smi)
-        mol.clean2d()
-        examples.append(mol)
-    svg = grid_depict(examples, 2)
-    render_svg(svg)
-
-    show_full_stats = st.checkbox('Show full stats')
-    if show_full_stats:
-        st.dataframe(full_stats)
-
+    with st.form("results"):
+        st.subheader('Generation results', divider='rainbow')
+        st.dataframe(filtered_gen_stats)
         st.download_button(
-            label="Download full results as CSV",
-            data=convert_df(full_stats),
-            file_name='vqgae_tubulin_inhibitors_full.csv',
+            label="Download results as CSV",
+            data=convert_df(filtered_gen_stats),
+            file_name='vqgae_tubulin_inhibitors_valid.csv',
             mime='text/csv',
         )
 
-    if st.button("Restart"):
+        st.subheader('Examples of generated molecules')
+        examples_smiles = filtered_gen_stats.sort_values(by=["similarity_score"], ascending=False).iloc[:6].smiles.to_list()
+        examples = []
+        for smi in examples_smiles:
+            mol = smiles(smi)
+            mol.clean2d()
+            examples.append(mol)
+        svg = grid_depict(examples, 2)
+        render_svg(svg)
+
+        with st.expander("Show full stats"):
+            st.dataframe(full_stats)
+
+            st.download_button(
+                label="Download full results as CSV",
+                data=convert_df(full_stats),
+                file_name='vqgae_tubulin_inhibitors_full.csv',
+                mime='text/csv',
+            )
+        restart = st.form_submit_button('Restart')
+
+    if restart:
         st.rerun()