Generate figures on clicking a button with code hidden.

[atharvaaalok]

Hello,
I am writing a scientific article about my research.

I give a first code block to explain the method.
This generates a figure.

I then want to run the same code again in a different cell to generate figures many times.
Basically, some parameter is randomly initialized and a figure is generated.
This should help my reader get an idea of the types of outputs that can be generated for different parameter values.

Here is what I want to achieve.
I want my second code block to be hidden. Since I have already explained the working of the code in the first block I don't need to show it again in a loop. This just adds to clutter.
Then I want there to be a button (say an ipywidget) that when clicked generates a new figure.
So basically I want to hide my code to run behind a button that my reader can click to generate figures.

Example code:

from ipywidgets import Button, VBox, Output
from IPython.display import display

def my_func(t):
    return t

# Create an output widget to display the plot
output = Output()

# Function to execute when the button is clicked
def generate_plot(_):
    with output:
        output.clear_output(wait = True)  # Clear previous output for fresh plot
        num_pts = 1000
        t = np.linspace(0, 1, num_pts).reshape(-1, 2)

        X = my_func(t)

        # Plot the result
        plt.plot(X[:, 0], X[:, 1])
        plt.show()

# Create a button widget
button = Button(description = "Generate Plot")
button.on_click(generate_plot)

# Display the button and output together
display(VBox([button, output]))

I want only the "Generate Plot" button to be visible. The code should somehow be hidden in the background.
For example hidden in a dropdown which the reader can click to see the code if they want to.
How can I achieve functionality like this?