improving layout

app.py

@@ -8,7 +8,7 @@ import numpy as np
 from AtomLenz import *
 #from utils_graph import *
 from Object_Smiles import Objects_Smiles 
-
+from rdkit import Chem
 #from robust_detection import wandb_config
 from robust_detection import utils
 from robust_detection.models.rcnn import RCNN
@@ -198,8 +198,16 @@ def atomlenz(modelfile):
             predictions_list.append([image_idx,pred_smiles,problematic])
        file_preds = open('preds_atomlenz','w')
        for pred in predictions_list:
+            smiles = pred[1]
+            m = Chem.MolFromSmiles(smiles)
+            if m is None:
+               st.write("No valid chemical structure recognized")
+            else:
+                img = Chem.Draw.MolToImage(m)
+                st.image(img, caption=f"{smiles}", use_column_width=True)
             print(pred)
 
+
 #### TRYOUT MENU #####
 
 page_to_funcs = {