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