Update app.py

app.py

@@ -356,9 +356,7 @@ CONTEXT: {props_string}.\n \
 '
 
   res = chat_model.invoke(check_prompt)
-  print('========== parser node first chat ===========================')
-  print(res)
-  print('*'*50)
+
   if str(res).split('<|assistant|>')[1].split('#')[0].strip().lower() == "loop":
     state["loop_again"] = "loop_again"
     return state
@@ -375,8 +373,7 @@ CONTEXT: {props_string}.\n \
 QUERY_TASK: {query_task}.\n '
 
     res = chat_model.invoke(prompt)
-    print("======== parser node second chat =============")
-    print(res)
+
     return {"messages": res}
 
 def reflect_node(state: State) -> State:
@@ -397,8 +394,7 @@ PREVIOUS ANSWER: {previous_answer}.\n \
 TOOL RESULTS: {props_string}. '
 
   res = chat_model.invoke(prompt)
-  print("======== reflect node second chat =============")
-  print(res)
+
   return {"messages": res}
 
 def get_chemtool(state):
@@ -503,8 +499,6 @@ def MoleculeAgent(task, smiles, name):
     if len(m) != 0:
       try:
           reply = c[str(m[0])]['messages']
-          print("======== DEBUG ================")
-          print(c)
           if 'assistant' in str(reply):
             reply = str(reply).split("<|assistant|>")[-1].split('#')[0].strip()
             replies.append(reply)
@@ -524,7 +518,7 @@ def MoleculeAgent(task, smiles, name):
 with gr.Blocks(fill_height=True) as forest:
   gr.Markdown('''
               # Molecule Agent - calls the PubChem API to:
-              - fetche names
+              - fetch names
               - fetch SMILES
               - find similar molecules
               ''')