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

Fixes CORS handling

Browse Source
This commit is contained in:
psychedelicious 2022-10-08 07:12:51 +08:00 committed by Lincoln Stein
parent 47a97f7e97
commit f5dfd5b0dc
1 changed files with 11 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
backend/invoke_ai_web_server.py
View File

@ -49,21 +49,16 @@ class InvokeAIWebServer:
engineio_logger = True if args.web_verbose else False engineio_logger = True if args.web_verbose else False
max_http_buffer_size = 10000000 max_http_buffer_size = 10000000
# CORS Allowed Setup socketio_args = {
cors_allowed_origins = [ 'logger': logger,
'http://127.0.0.1:5173', 'engineio_logger': engineio_logger,
'http://localhost:5173', 'max_http_buffer_size': max_http_buffer_size,
'http://localhost:9090' 'ping_interval': (50, 50),
] 'ping_timeout': 60,
additional_allowed_origins = ( }
opt.cors if opt.cors else []
) # additional CORS allowed origins if opt.cors:
cors_allowed_origins.append(f'http://{self.host}:{self.port}') socketio_args['cors_allowed_origins'] = opt.cors
if self.host == '127.0.0.1' or self.host == '0.0.0.0':
cors_allowed_origins.append(f'http://localhost:{self.port}')
cors_allowed_origins = (
cors_allowed_origins + additional_allowed_origins
)
self.app = Flask( self.app = Flask(
__name__, static_url_path='', static_folder='../frontend/dist/' __name__, static_url_path='', static_folder='../frontend/dist/'
@ -71,12 +66,7 @@ class InvokeAIWebServer:
self.socketio = SocketIO( self.socketio = SocketIO(
self.app, self.app,
logger=logger, **socketio_args
engineio_logger=engineio_logger,
max_http_buffer_size=max_http_buffer_size,
cors_allowed_origins=cors_allowed_origins,
ping_interval=(50, 50),
ping_timeout=60,
) )
# Keep Server Alive Route # Keep Server Alive Route