Update app.py

app.py

@@ -142,7 +142,6 @@ def similars_node(state: State) -> State:
       print('Not enough information to run similars tool')
       return state
 
-    props_string = 'Found Similar compounds: \n'
     sub_smiles = []
 
     i = 0
@@ -295,7 +294,7 @@ Returns the names, SMILES strings, molecular weights and logP values for the sim
 
   res = chat_model.invoke(prompt)
 
-  tool_choices = str(res).split('<|assistant|>')[1].split('#')[0].strip()
+  tool_choices = str(res).replace('smilars', 'similars').split('<|assistant|>')[1].split('#')[0].strip()
   tool_choices = tool_choices.split(',')
   if len(tool_choices) == 1:
     if tool_choices[0].strip().lower() == 'none':
@@ -513,8 +512,10 @@ def MoleculeAgent(smiles, name, task):
 
 with gr.Blocks(fill_height=True) as forest:
   gr.Markdown('''
-              # Molecule Agent - fetches names and SMILES from PubChem or finds
-              similar molecules.
+              # Molecule Agent - calls the PubChem API to:
+              - fetche names
+              - fetch SMILES
+              - find similar molecules
               ''')
 
   name, smiles = None, None