utils.py

from IPython.display import display, Markdown


def build_buttons(link, github, colab = True, citation=False):
    """
    Used to generate github, collab, and website-linked buttons
    """
    if citation:
        citation_str = f'<a type="button" class="btn btn-outline-primary btn-sm" target="_self" href="#citation">🏛 Citation</a>'
    else:
        citation_str = ''

    if colab:
        colab_str = f'\n<a type="button" class="btn btn-outline-primary btn-sm" target="_blank" href="https://colab.research.google.com/github/{github.split(".com/")[-1]}">🖥️  Interactive version</a>'
    else:
        colab_str = ''
    display(Markdown(f"""
<a type="button" class="btn btn-outline-primary btn-sm" target="_blank" href="https://inspectelement.org/{link}">📖 Read online</a>{colab_str}
<a type="button" class="btn btn-outline-primary btn-sm" target="_blank" href="{github}">⚙️ GitHub</a>
{citation_str}
<br>
"""))