Add polling callback

    callbacks_infotext_pasted=[],

    callbacks_script_unloaded=[],

    callbacks_before_ui=[],

    callbacks_on_polling=[],

)

        except Exception:

            report_exception(c, 'app_started_callback')

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

    for c in callback_map['callbacks_on_polling']:

        try:

            c.callback()

        except Exception:

            report_exception(c, 'callbacks_on_polling')

def model_loaded_callback(sd_model):

    for c in callback_map['callbacks_model_loaded']:

    add_callback(callback_map['callbacks_app_started'], callback)

def on_polling(callback):

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

    add_callback(callback_map['callbacks_on_polling'], callback)

def on_model_loaded(callback):

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

    passed as an argument; this function is also called when the script is reloaded. """

def wait_on_server(demo=None):

    while 1:

        time.sleep(0.5)

        modules.script_callbacks.app_polling_callback(None, demo)

        if shared.state.need_restart:

            shared.state.need_restart = False

            time.sleep(0.5)