add --cors-allow-origins cmd opt

1 files changed, 9 insertions, 0 deletions

diff --git a/webui.py b/webui.py
index 81df09dd..3788af0b 100644
--- a/webui.py
+++ b/webui.py
@@ -5,6 +5,7 @@ import importlib
 import signal

 import threading

 from fastapi import FastAPI

+from fastapi.middleware.cors import CORSMiddleware

 from fastapi.middleware.gzip import GZipMiddleware

 

 from modules.paths import script_path

@@ -93,6 +94,11 @@ def initialize():
     signal.signal(signal.SIGINT, sigint_handler)

 

 

+def setup_cors(app):

+    if cmd_opts.cors_allow_origins:

+        app.add_middleware(CORSMiddleware, allow_origins=cmd_opts.cors_allow_origins.split(','), allow_methods=['*'])

+

+

 def create_api(app):

     from modules.api.api import Api

     api = Api(app, queue_lock)

@@ -114,6 +120,7 @@ def api_only():
     initialize()

 

     app = FastAPI()

+    setup_cors(app)

     app.add_middleware(GZipMiddleware, minimum_size=1000)

     api = create_api(app)

 

@@ -147,6 +154,8 @@ def webui():
         # runnnig its code. We disable this here. Suggested by RyotaK.

         app.user_middleware = [x for x in app.user_middleware if x.cls.__name__ != 'CORSMiddleware']

 

+        setup_cors(app)

+

         app.add_middleware(GZipMiddleware, minimum_size=1000)

 

         if launch_api: