mirror of
https://github.com/lobehub/sd-webui-lobe-theme.git
synced 2026-01-09 06:23:44 +08:00
30 lines
727 B
Python
30 lines
727 B
Python
from fastapi import FastAPI
|
|
|
|
import modules.scripts as scripts
|
|
import os
|
|
from typing import Any
|
|
|
|
from modules import shared
|
|
from modules import script_callbacks
|
|
|
|
from scripts.lib.lobe_log import LobeLog
|
|
from scripts.lib.api import LobeApi
|
|
from scripts.lib.config import LobeConfig
|
|
from scripts.lib.package import LobePackage
|
|
from scripts.lib.prompt import LobePrompt
|
|
from scripts.lib.locale import LobeLocale
|
|
|
|
def init_lobe(_: Any, app: FastAPI, **kwargs):
|
|
LobeLog.info("Initializing...")
|
|
|
|
package = LobePackage()
|
|
prompt = LobePrompt()
|
|
locale = LobeLocale()
|
|
config = LobeConfig()
|
|
api = LobeApi(config, package, prompt, locale)
|
|
api.create_api_route(app)
|
|
|
|
|
|
|
|
script_callbacks.on_app_started(init_lobe)
|