Draw 3d border in plotly like persp in R

I am trying to draw a border on all sides in a 3d surface plot like in the image below. This is an example in R using persp function:

How can I get similar borders using Plotly?

Thank you.

Maybe try using something like this around each axis:

fig.add_shape(
        # Rectangle with reference to the plot
            type="rect",
            xref="paper",
            yref="paper",
            x0=0,
            y0=0,
            x1=1.0,
            y1=1.0,
            line=dict(
                color="black",
                width=1,
            )
        )