From e414582be658778861b37f55f9c1bdc57d7a8651 Mon Sep 17 00:00:00 2001 From: Thorsten Mueller Date: Wed, 20 Jan 2021 21:56:40 +0100 Subject: [PATCH 1/2] Added option for server ui details page. --- TTS/server/server.py | 18 ++++ TTS/server/templates/details.html | 131 ++++++++++++++++++++++++++++++ 2 files changed, 149 insertions(+) create mode 100644 TTS/server/templates/details.html diff --git a/TTS/server/server.py b/TTS/server/server.py index bd23ea9c..f2412bb8 100644 --- a/TTS/server/server.py +++ b/TTS/server/server.py @@ -4,6 +4,7 @@ import os from flask import Flask, request, render_template, send_file from TTS.server.synthesizer import Synthesizer +from TTS.utils.io import load_config def create_argparser(): @@ -23,6 +24,7 @@ def create_argparser(): 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 @@ -69,6 +71,22 @@ app = Flask(__name__) def index(): return render_template('index.html') +@app.route('/details') +def details(): + if args.tts_config is not None and os.path.isfile(args.tts_config): + taco2_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 + , taco2_config=taco2_config + , vocoder_config=vocoder_config + , args=args.__dict__ + ) @app.route('/api/tts', methods=['GET']) def tts(): diff --git a/TTS/server/templates/details.html b/TTS/server/templates/details.html new file mode 100644 index 00000000..2db60657 --- /dev/null +++ b/TTS/server/templates/details.html @@ -0,0 +1,131 @@ + + + + + + + + + + + TTS engine + + + + + + + + + + Fork me on GitHub + + {% if show_details == true %} + +
+ Model details +
+ +
+
+ CLI arguments: + + + + + + + {% for key, value in args.items() %} + + + + + + + {% endfor %} +
CLI key Value
{{ key }}{{ value }}
+
+

+ +
+ + {% if taco2_config != None %} + +
+ Tacotron2 model config: + + + + + + + + + {% for key, value in taco2_config.items() %} + + + + + + + {% endfor %} + +
Key Value
{{ key }}{{ value }}
+
+ + {% endif %} + +

+ + + +
+ {% if vocoder_config != None %} +
+ Vocoder model config: + + + + + + + + + {% for key, value in vocoder_config.items() %} + + + + + + + {% endfor %} + + +
Key Value
{{ key }}{{ value }}
+
+ {% endif %} +

+ + {% else %} +
+ Please start server with --show_details=true to see details. +
+ + {% endif %} + + + + \ No newline at end of file From afb7db2a1dec6d7a135e94a808e73b04f6135e7e Mon Sep 17 00:00:00 2001 From: Thorsten Mueller Date: Fri, 22 Jan 2021 16:22:50 +0100 Subject: [PATCH 2/2] Removed unneeded check and removed specific taco2 model name. --- TTS/server/server.py | 6 ++---- TTS/server/templates/details.html | 6 +++--- 2 files changed, 5 insertions(+), 7 deletions(-) diff --git a/TTS/server/server.py b/TTS/server/server.py index f2412bb8..99d3a92e 100644 --- a/TTS/server/server.py +++ b/TTS/server/server.py @@ -73,9 +73,7 @@ def index(): @app.route('/details') def details(): - if args.tts_config is not None and os.path.isfile(args.tts_config): - taco2_config = load_config(args.tts_config) - + 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: @@ -83,7 +81,7 @@ def details(): return render_template('details.html', show_details=args.show_details - , taco2_config=taco2_config + , model_config=model_config , vocoder_config=vocoder_config , args=args.__dict__ ) diff --git a/TTS/server/templates/details.html b/TTS/server/templates/details.html index 2db60657..51c9ed85 100644 --- a/TTS/server/templates/details.html +++ b/TTS/server/templates/details.html @@ -62,10 +62,10 @@
- {% if taco2_config != None %} + {% if model_config != None %}
- Tacotron2 model config: + Model config: @@ -74,7 +74,7 @@ - {% for key, value in taco2_config.items() %} + {% for key, value in model_config.items() %}
{{ key }}