Update app.py

app.py

@@ -346,7 +346,7 @@ CONTEXT: {props_string}.\n \
 '
 
   res = chat_model.invoke(check_prompt)
-  print('*'*50)
+  print('========== parser node first chat ===========================')
   print(res)
   print('*'*50)
   if str(res).split('<|assistant|>')[1].split('#')[0].strip().lower() == "loop":
@@ -363,7 +363,7 @@ QUERY_TASK: {query_task}.\n \
 CONTEXT: {props_string}.\n '
 
     res = chat_model.invoke(prompt)
-    print("======== parser node=============")
+    print("======== parser node second chat =============")
     print(res)
     return {"messages": res}
 
@@ -385,6 +385,8 @@ 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):
@@ -412,7 +414,7 @@ def get_chemtool(state):
 def loop_or_not(state):
   '''
   '''
-  print(f"Loop? {state['loop_again']}")
+  print(f"(line 417) Loop? {state['loop_again']}")
   if state["loop_again"] == "loop_again":
     return True
   else: