Add a callback called before reloading the server

    callbacks_before_component=[],

    callbacks_after_component=[],

    callbacks_on_polling=[],

    callbacks_on_reload=[],

)

    for callback_list in callback_map.values():

        callback_list.clear()

def app_started_callback(demo: Optional[Blocks], app: FastAPI):

    for c in callback_map['callbacks_app_started']:

        try:

        except Exception:

            report_exception(c, 'callbacks_on_polling')

def app_reload_callback(demo: Optional[Blocks], app: FastAPI):

    for c in callback_map['callbacks_on_reload']:

        try:

            c.callback()

        except Exception:

            report_exception(c, 'callbacks_on_reload')

def model_loaded_callback(sd_model):

    for c in callback_map['callbacks_model_loaded']:

        try:

    """register a function to be called on each polling of the server."""

    add_callback(callback_map['callbacks_on_polling'], callback)

def on_before_reload(callback):

    """register a function to be called just before the server reloads."""

    add_callback(callback_map['callbacks_on_reload'], callback)

def on_model_loaded(callback):

    """register a function to be called when the stable diffusion model is created; the model is

        time.sleep(0.5)

        modules.script_callbacks.app_polling_callback(None, demo)

        if shared.state.need_restart:

            modules.script_callbacks.app_reload_callback(None, demo)

            shared.state.need_restart = False

            time.sleep(0.5)

            demo.close()