|
import argparse |
|
import logging |
|
|
|
from fastapi import FastAPI |
|
|
|
from modules import config |
|
from modules.api.Api import APIManager |
|
from modules.api.impl import ( |
|
google_api, |
|
models_api, |
|
openai_api, |
|
ping_api, |
|
refiner_api, |
|
speaker_api, |
|
ssml_api, |
|
style_api, |
|
tts_api, |
|
xtts_v2_api, |
|
) |
|
from modules.utils import env |
|
|
|
logger = logging.getLogger(__name__) |
|
|
|
|
|
def create_api(app: FastAPI, exclude=[]): |
|
app_mgr = APIManager(app=app, exclude_patterns=exclude) |
|
|
|
ping_api.setup(app_mgr) |
|
models_api.setup(app_mgr) |
|
style_api.setup(app_mgr) |
|
speaker_api.setup(app_mgr) |
|
tts_api.setup(app_mgr) |
|
ssml_api.setup(app_mgr) |
|
google_api.setup(app_mgr) |
|
openai_api.setup(app_mgr) |
|
refiner_api.setup(app_mgr) |
|
xtts_v2_api.setup(app_mgr) |
|
|
|
return app_mgr |
|
|
|
|
|
def setup_api_args(parser: argparse.ArgumentParser): |
|
parser.add_argument( |
|
"--cors_origin", |
|
type=str, |
|
help="Allowed CORS origins. Use '*' to allow all origins.", |
|
) |
|
parser.add_argument( |
|
"--no_playground", |
|
action="store_true", |
|
help="Disable the playground entry", |
|
) |
|
parser.add_argument( |
|
"--no_docs", |
|
action="store_true", |
|
help="Disable the documentation entry", |
|
) |
|
|
|
parser.add_argument( |
|
"--exclude", |
|
type=str, |
|
help="Exclude the specified API from the server", |
|
) |
|
|
|
|
|
def process_api_args(args: argparse.Namespace, app: FastAPI): |
|
cors_origin = env.get_and_update_env(args, "cors_origin", "*", str) |
|
no_playground = env.get_and_update_env(args, "no_playground", False, bool) |
|
no_docs = env.get_and_update_env(args, "no_docs", False, bool) |
|
exclude = env.get_and_update_env(args, "exclude", "", str) |
|
|
|
api = create_api(app=app, exclude=exclude.split(",")) |
|
config.api = api |
|
|
|
if cors_origin: |
|
api.set_cors(allow_origins=[cors_origin]) |
|
|
|
if not no_playground: |
|
api.setup_playground() |
|
|
|
if compile: |
|
logger.info("Model compile is enabled") |
|
|
