Merge pull request #3982 from MaikoTan/on-started-callback

from collections import namedtuple

import inspect

from fastapi import FastAPI

from gradio import Blocks

def report_exception(c, job):

    print(f"Error executing callback {job} for {c.script}", file=sys.stderr)

ScriptCallback = namedtuple("ScriptCallback", ["script", "callback"])

callbacks_app_started = []

callbacks_model_loaded = []

callbacks_ui_tabs = []

callbacks_ui_settings = []

    callbacks_image_saved.clear()

def app_started_callback(demo: Blocks, app: FastAPI):

    for c in callbacks_app_started:

        try:

            c.callback(demo, app)

        except Exception:

            report_exception(c, 'app_started_callback')

def model_loaded_callback(sd_model):

    for c in callbacks_model_loaded:

        try:

    callbacks.append(ScriptCallback(filename, fun))

def on_app_started(callback):

    """register a function to be called when the webui started, the gradio `Block` component and

    fastapi `FastAPI` object are passed as the arguments"""

    add_callback(callbacks_app_started, 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"""

import modules.sd_models

import modules.shared as shared

import modules.txt2img

import modules.script_callbacks

import modules.ui

from modules import devices

        if launch_api:

            create_api(app)

        modules.script_callbacks.app_started_callback(demo, app)

        wait_on_server(demo)

        sd_samplers.set_samplers()