Update app

app.py

@@ -394,7 +394,10 @@ def molecules_from_image():
             mol_imgs = draw_molecules(smiles)
 
         with st.container():
-            #st.write("Ranking of most similar molecules")
+            if uploaded_file is not None:
+                st.write("Retrieved molecules")
+            else:
+                st.write("Retrieved molecules from the Cell Painting database")
             columns = st.columns(len(top_probs))
             for i, col in enumerate(columns):
                 if predefined_features:
@@ -408,7 +411,8 @@ def molecules_from_image():
         print((top_probs, top_labels))
 
 def images_from_molecule():
-    smiles = st.text_input("Enter a query molecule in [SMILES](https://en.wikipedia.org/wiki/Simplified_molecular-input_line-entry_system) format", value="CC(=O)OC1=CC=CC=C1C(=O)O", placeholder="CC(=O)OC1=CC=CC=C1C(=O)O")
+    st.markdown("Enter a query molecule in [SMILES](https://en.wikipedia.org/wiki/Simplified_molecular-input_line-entry_system) format",)
+    smiles = st.text_input("", value="CC(=O)OC1=CC=CC=C1C(=O)O", placeholder="CC(=O)OC1=CC=CC=C1C(=O)O")
     if smiles:
         smiles = [smiles]
         morgan = [morgan_from_smiles(s) for s in smiles]
@@ -450,6 +454,7 @@ def images_from_molecule():
 
         images_dict = np.load(images_arr, allow_pickle = True)
 
+        st.write("Retrieved images from the Cell Painting database")
         with st.container():
             columns = st.columns(len(top_probs))
             for i, col in enumerate(columns):