gradio and FastAPI

class Api:

    def __init__(self, app):

    def __init__(self, app, queue_lock):

        self.router = APIRouter()

        app.add_api_route("/sdapi/v1/txt2img", self.text2imgapi, methods=["POST"])

        self.app = app

        self.queue_lock = queue_lock

        self.app.add_api_route("/sdapi/v1/txt2img", self.text2imgapi, methods=["POST"])

    def text2imgapi(self, txt2imgreq: StableDiffusionProcessingAPI ):

        populate = txt2imgreq.copy(update={ # Override __init__ params

        )

        p = StableDiffusionProcessingTxt2Img(**vars(populate))

        # Override object param

        with self.queue_lock:

            processed = process_images(p)

        b64images = []

        raise NotImplementedError

    def launch(self, server_name, port):

        app.include_router(self.router)

        uvicorn.run(app, host=server_name, port=port)

        self.app.include_router(self.router)

        uvicorn.run(self.app, host=server_name, port=port)

import importlib

import signal

import threading

from fastapi import FastAPI

from fastapi.middleware.gzip import GZipMiddleware

from modules.paths import script_path

from modules.shared import cmd_opts

import modules.hypernetworks.hypernetwork

queue_lock = threading.Lock()

def create_api(app):

    from modules.api.api import Api

    api = Api(app)

    api = Api(app, queue_lock)

    return api

def wait_on_server(demo=None):

        print('Restarting Gradio')

task = []

if __name__ == "__main__":

    if not cmd_opts.nowebui:

    if cmd_opts.nowebui:

        api_only()

    if cmd_opts.api:

        webui(True)

    else:

        webui(False)

        webui(cmd_opts.api)

 No newline at end of file