| | 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") |
| |
|
