Update app.py

app.py

@@ -65,7 +65,11 @@ react_graph = app_graph.compile()
 # Save graph visualization as an image
 with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmpfile:
     graph_image_path = tmpfile.name
-    react_graph.get_graph(xray=True).draw_mermaid_png(graph_image_path)
+    graph = react_graph.get_graph(xray=True)
+    graph.draw_mermaid_png()
+    with open(tmpfile.name, "wb") as f:
+        f.write(graph._repr_png_())  # Save the graph image
+
 
 # ------------------- Streamlit Interface -------------------
 st.title("ReAct Agent with Arithmetic and Search")