Update app.py

app.py

@@ -105,7 +105,7 @@ def smiles_node(state: State) -> State:
     smiles_string = f'The SMILES string for the molecule is: {smiles}\n'
   except:
     smiles = "unknown"
-    smiles_string = 'Failed loop. Look further.\n'
+    smiles_string = 'Look further.\n'
 
   state["query_smiles"] = smiles
 
@@ -166,7 +166,7 @@ def similars_node(state: State) -> State:
     #     outf.write(pic)
     img.save(filename)
   except:
-    props_string = 'Failed loop. Look further.\n'
+    props_string = 'Look further.\n'
     filename = None
 
   current_props_string += props_string
@@ -238,7 +238,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':