put all TTS endpoints in same module

tts_service/src/endpoints/mod.rs

@@ -1,12 +1,9 @@
 pub mod helpers;
 pub mod index;
-pub mod legacy_speak;
 pub mod liveness;
 pub mod models;
 pub mod readiness;
 pub mod sentences;
 pub mod speak;
-pub mod speak_with_spectrogram;
 pub mod speakers;
-pub mod tts;
-pub mod words;
+pub mod words;

tts_service/src/endpoints/speak/mod.rs

@@ -0,0 +1,4 @@
+pub mod legacy_speak;
+pub mod legacy_tts;
+pub mod speak;
+pub mod speak_with_spectrogram;

tts_service/src/main.rs

@@ -30,27 +30,27 @@ use actix_web::middleware::{Logger, DefaultHeaders};
 use actix_web::{App, HttpResponse, HttpServer, web, http};
 use anyhow::Result as AnyhowResult;
 
+use arpabet::Arpabet;
 use crate::database::connector::DatabaseConnector;
 use crate::endpoints::index::get_root;
-use crate::endpoints::legacy_speak::legacy_get_speak;
 use crate::endpoints::liveness::get_liveness;
 use crate::endpoints::models::get_models;
 use crate::endpoints::readiness::get_readiness;
 use crate::endpoints::sentences::get_sentences;
-use crate::endpoints::speak::post_speak;
-use crate::endpoints::speak_with_spectrogram::post_speak_with_spectrogram;
+use crate::endpoints::speak::legacy_speak::legacy_get_speak;
+use crate::endpoints::speak::legacy_tts::post_tts;
+use crate::endpoints::speak::speak::post_speak;
+use crate::endpoints::speak::speak_with_spectrogram::post_speak_with_spectrogram;
 use crate::endpoints::speakers::get_speakers;
-use crate::endpoints::tts::post_tts;
 use crate::endpoints::words::get_words;
 use crate::model::model_cache::ModelCache;
 use crate::model::model_config::ModelConfigs;
-use crate::rate_limiter::redis_rate_limiter::RedisRateLimiter;
-use crate::text::checker::TextChecker;
-use arpabet::Arpabet;
-use limitation::Limiter;
-use std::time::Duration;
 use crate::rate_limiter::RateLimiter;
 use crate::rate_limiter::noop_rate_limiter::NoOpRateLimiter;
+use crate::rate_limiter::redis_rate_limiter::RedisRateLimiter;
+use crate::text::checker::TextChecker;
+use limitation::Limiter;
+use std::time::Duration;
 
 const ENV_ARPABET_EXTRAS_FILE : &'static str = "ARPABET_EXTRAS_FILE";
 const ENV_ASSET_DIRECTORY: &'static str = "ASSET_DIRECTORY";

tts_service/src/model/pipelines.rs

@@ -1,4 +1,7 @@
 use arpabet::Arpabet;
+use crate::endpoints::speak::speak_with_spectrogram::Spectrogram;
+use crate::model::arpabet_glow_tts_model::ArpabetGlowTtsModel;
+use crate::model::arpabet_glow_tts_multi_speaker_model::ArpabetGlowTtsMultiSpeakerModel;
 use crate::model::arpabet_tacotron_model::ArpabetTacotronModel;
 use crate::model::melgan_model::MelganModel;
 use crate::model::old_model::TacoMelModel;
@@ -8,9 +11,6 @@ use hound::WavSpec;
 use hound::WavWriter;
 use std::io::{Cursor, BufWriter};
 use tch::Tensor;
-use crate::model::arpabet_glow_tts_model::ArpabetGlowTtsModel;
-use crate::model::arpabet_glow_tts_multi_speaker_model::ArpabetGlowTtsMultiSpeakerModel;
-use crate::endpoints::speak_with_spectrogram::Spectrogram;
 
 // TODO: This might be useful to implement as a multi-stage
 //  state machine struct with functions, that way you can pull out