Update utils.py

utils.py

@@ -170,11 +170,11 @@ def extract_mermaid_and_generate_graph(state: GraphState):
     # Save the mermaid code in a file with the current timestamp
     # print(response)
     file_name = f"mermaid/{int(time.time())}.png"
-    render_mermaid(response, file_name)
+    # render_mermaid(response, file_name)
 
-    # Read image to return as output
-    img = Image.open(file_name)
-    return {"messages": [AIMessage(response)], "final_graph": img}
+    # # Read image to return as output
+    # img = Image.open(file_name)
+    return {"messages": [AIMessage(response)], "final_graph": None}
 
 
 def need_to_update_working_knowledge(state: GraphState):