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

api_app: rename web_config to app_config for consistency

Browse Source
This commit is contained in:
Eugene
2023-05-17 19:42:13 -04:00
parent 9e4e386c9b
commit 8a8b09a953
1 changed files with 9 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
invokeai/app/api_app.py
View File

@ -35,24 +35,23 @@ app.add_middleware(
socket_io = SocketIO(app) socket_io = SocketIO(app)
# parse command-line settings, environment and the init file # initialize config
# (this is a module global) # this is a module global
global web_config app_config = InvokeAIAppConfig()
web_config = InvokeAIAppConfig()
# 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( app.add_middleware(
CORSMiddleware, CORSMiddleware,
allow_origins=web_config.allow_origins, allow_origins=app_config.allow_origins,
allow_credentials=web_config.allow_credentials, allow_credentials=app_config.allow_credentials,
allow_methods=web_config.allow_methods, allow_methods=app_config.allow_methods,
allow_headers=web_config.allow_headers, allow_headers=app_config.allow_headers,
) )
ApiDependencies.initialize( ApiDependencies.initialize(
config=web_config, event_handler_id=event_handler_id, logger=logger config=app_config, event_handler_id=event_handler_id, logger=logger
) )
@ -143,10 +142,9 @@ def overridden_redoc():
app.mount("/", StaticFiles(directory="invokeai/frontend/web/dist", html=True), name="ui") app.mount("/", StaticFiles(directory="invokeai/frontend/web/dist", html=True), name="ui")
def invoke_api(): def invoke_api():
global web_config
# 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=app_config.host, port=app_config.port, loop=loop)
# Use access_log to turn off logging # Use access_log to turn off logging
server = uvicorn.Server(config) server = uvicorn.Server(config)
loop.run_until_complete(server.serve()) loop.run_until_complete(server.serve())