try:
    from IPython.core.magic import needs_local_scope, register_cell_magic
except ImportError:
    pass

import warnings

import gradio


def load_ipython_extension(ipython):
    __demo = gradio.Blocks()

    @register_cell_magic
    @needs_local_scope
    def blocks(line, cell, local_ns=None):
        if "gr.Interface" in cell:
            warnings.warn("Usage of gr.Interface with %%blocks may result in errors.")
        with __demo.clear():
            exec(cell, None, local_ns)
            __demo.launch(quiet=True)