mirror of https://github.com/coqui-ai/TTS.git
102 lines
4.1 KiB
Python
102 lines
4.1 KiB
Python
#!flask/bin/python
|
|
import argparse
|
|
import os
|
|
import sys
|
|
import io
|
|
from pathlib import Path
|
|
|
|
from flask import Flask, render_template, request, send_file
|
|
from TTS.utils.synthesizer import Synthesizer
|
|
from TTS.utils.manage import ModelManager
|
|
from TTS.utils.io import load_config
|
|
|
|
|
|
def create_argparser():
|
|
def convert_boolean(x):
|
|
return x.lower() in ['true', '1', 'yes']
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--list_models', type=convert_boolean, nargs='?', const=True, default=False, help='list available pre-trained tts and vocoder models.')
|
|
parser.add_argument('--model_name', type=str, default="tts_models/en/ljspeech/speedy-speech-wn", help='name of one of the released tts models.')
|
|
parser.add_argument('--vocoder_name', type=str, default=None, help='name of one of the released vocoder models.')
|
|
parser.add_argument('--tts_checkpoint', type=str, help='path to custom tts checkpoint file')
|
|
parser.add_argument('--tts_config', type=str, help='path to custom tts config.json file')
|
|
parser.add_argument('--tts_speakers', type=str, help='path to JSON file containing speaker ids, if speaker ids are used in the model')
|
|
parser.add_argument('--vocoder_config', type=str, default=None, help='path to vocoder config file.')
|
|
parser.add_argument('--vocoder_checkpoint', type=str, default=None, help='path to vocoder checkpoint file.')
|
|
parser.add_argument('--port', type=int, default=5002, help='port to listen on.')
|
|
parser.add_argument('--use_cuda', type=convert_boolean, default=False, help='true to use CUDA.')
|
|
parser.add_argument('--debug', type=convert_boolean, default=False, help='true to enable Flask debug mode.')
|
|
parser.add_argument('--show_details', type=convert_boolean, default=False, help='Generate model detail page.')
|
|
return parser
|
|
|
|
# parse the args
|
|
args = create_argparser().parse_args()
|
|
|
|
path = Path(__file__).parent / "../.models.json"
|
|
manager = ModelManager(path)
|
|
|
|
if args.list_models:
|
|
manager.list_models()
|
|
sys.exit()
|
|
|
|
# update in-use models to the specified released models.
|
|
if args.model_name is not None:
|
|
tts_checkpoint_file, tts_config_file, tts_json_dict = manager.download_model(args.model_name)
|
|
args.vocoder_name = tts_json_dict['default_vocoder'] if args.vocoder_name is None else args.vocoder_name
|
|
|
|
if args.vocoder_name is not None:
|
|
vocoder_checkpoint_file, vocoder_config_file, vocoder_json_dict = manager.download_model(args.vocoder_name)
|
|
|
|
# If these were not specified in the CLI args, use default values with embedded model files
|
|
if not args.tts_checkpoint and os.path.isfile(tts_checkpoint_file):
|
|
args.tts_checkpoint = tts_checkpoint_file
|
|
if not args.tts_config and os.path.isfile(tts_config_file):
|
|
args.tts_config = tts_config_file
|
|
|
|
if not args.vocoder_checkpoint and os.path.isfile(vocoder_checkpoint_file):
|
|
args.vocoder_checkpoint = vocoder_checkpoint_file
|
|
if not args.vocoder_config and os.path.isfile(vocoder_config_file):
|
|
args.vocoder_config = vocoder_config_file
|
|
|
|
synthesizer = Synthesizer(args.tts_checkpoint, args.tts_config, args.vocoder_checkpoint, args.vocoder_config, args.use_cuda)
|
|
|
|
app = Flask(__name__)
|
|
|
|
|
|
@app.route('/')
|
|
def index():
|
|
return render_template('index.html', show_details=args.show_details)
|
|
|
|
@app.route('/details')
|
|
def details():
|
|
model_config = load_config(args.tts_config)
|
|
if args.vocoder_config is not None and os.path.isfile(args.vocoder_config):
|
|
vocoder_config = load_config(args.vocoder_config)
|
|
else:
|
|
vocoder_config = None
|
|
|
|
return render_template('details.html',
|
|
show_details=args.show_details
|
|
, model_config=model_config
|
|
, vocoder_config=vocoder_config
|
|
, args=args.__dict__
|
|
)
|
|
|
|
@app.route('/api/tts', methods=['GET'])
|
|
def tts():
|
|
text = request.args.get('text')
|
|
print(" > Model input: {}".format(text))
|
|
wavs = synthesizer.tts(text)
|
|
out = io.BytesIO()
|
|
synthesizer.save_wav(wavs, out)
|
|
return send_file(out, mimetype='audio/wav')
|
|
|
|
|
|
def main():
|
|
app.run(debug=args.debug, host='0.0.0.0', port=args.port)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|