Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

implement change to web_config suggested by ebr

Browse Source
This commit is contained in:
Lincoln Stein 2023-05-13 21:33:19 -04:00
parent f28632980d
commit 11b2076b46
1 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
invokeai/app/api_app.py
View File

@ -17,7 +17,7 @@ from .api.dependencies import ApiDependencies
from .api.routers import images, sessions, models from .api.routers import images, sessions, models
from .api.sockets import SocketIO from .api.sockets import SocketIO
from .invocations.baseinvocation import BaseInvocation from .invocations.baseinvocation import BaseInvocation
from .services.config import InvokeAIWebConfig, get_invokeai_config from .services.config import InvokeAIWebConfig
# Create the app # Create the app
# TODO: create this all in a method so configuration/etc. can be passed in? # TODO: create this all in a method so configuration/etc. can be passed in?
@ -34,11 +34,23 @@ app.add_middleware(
) )
socket_io = SocketIO(app) socket_io = SocketIO(app)
web_config = {}
# parse command-line settings, environment and the init file
# (this is a module global)
global web_config
web_config = InvokeAIWebConfig()
# Add startup event to load dependencies # Add startup event to load dependencies
@app.on_event("startup") @app.on_event("startup")
async def startup_event(): async def startup_event():
app.add_middleware(
CORSMiddleware,
allow_origins=web_config.allow_origins,
allow_credentials=web_config.allow_credentials,
allow_methods=web_config.allow_methods,
allow_headers=web_config.allow_headers,
)
ApiDependencies.initialize( ApiDependencies.initialize(
config=web_config, event_handler_id=event_handler_id, logger=logger config=web_config, event_handler_id=event_handler_id, logger=logger
) )
@ -130,17 +142,7 @@ def overridden_redoc():
def invoke_api(): def invoke_api():
# parse command-line settings, environment and the init file
# (this is a module global)
global web_config global web_config
web_config = get_invokeai_config(InvokeAIWebConfig)
app.add_middleware(
CORSMiddleware,
allow_origins=web_config.allow_origins,
allow_credentials=web_config.allow_credentials,
allow_methods=web_config.allow_methods,
allow_headers=web_config.allow_headers,
)
# Start our own event loop for eventing usage # Start our own event loop for eventing usage
loop = asyncio.new_event_loop() loop = asyncio.new_event_loop()
config = uvicorn.Config(app=app, host=web_config.host, port=web_config.port, loop=loop) config = uvicorn.Config(app=app, host=web_config.host, port=web_config.port, loop=loop)