tool update

app.py

@@ -92,7 +92,11 @@ class BasicAgent:
         builder.add_edge(START, "agent")
         builder.add_conditional_edges("agent", tools_condition, ["agent", END])
         builder.add_edge("agent", END)
-        return builder.compile()
+        graph = builder.compile()
+        image = graph.get_graph().draw_mermaid_png()
+        with open("output_graph.png", "wb") as file:
+            file.write(image)
+        return graph
 
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):