State-dependent Action After Click

I’m new to Dash and I made a segmented funnel like this one
Now if the user clicks either segment of the graph, I have a callback that redraws the graph with just the chosen segment. What I want to do now to return to the original graph on another click.
Therefore, a click on the basic graph should get the user to a specialised graph; a click on a specialised graph should get the user back. The function drawing all the graphs is the same.
Is there a way to achieve this functionality?