save_and_display_fig.py

from pathlib import Path
from IPython.display import display, Markdown

def save_and_display_fig(fig, output_stem):
    output_stem = Path(output_stem)
    output_stem.parent.mkdir(parents=True, exist_ok=True)

    fig.savefig(output_stem.with_suffix(".pdf"), bbox_inches="tight")
    fig.savefig(output_stem.with_suffix(".svg"), bbox_inches="tight")
    plt.close(fig)

    display(Markdown(f"![{output_stem.name}]({output_stem.with_suffix('.svg')})"))