1. Self contained typers

2. Asr force-aligner visualization
3. streamlit state management abstraction
4. new utils / reorganize
5. added verbose flags
6. add tts by name

plume/cli/__init__.py

@@ -7,12 +7,12 @@ from .eval import app as eval_app
 from .serve import app as serve_app
 
 app = typer.Typer()
-app.add_typer(data_app, name="data")
-app.add_typer(ui_app, name="ui")
-app.add_typer(train_app, name="train")
-app.add_typer(eval_app, name="eval")
-app.add_typer(serve_app, name="serve")
-app.add_typer(utils_app, name='utils')
+app.add_typer(data_app)
+app.add_typer(ui_app)
+app.add_typer(train_app)
+app.add_typer(eval_app)
+app.add_typer(serve_app)
+app.add_typer(utils_app)
 
 
 def main():

plume/cli/data/__init__.py

@@ -27,6 +27,13 @@ app.add_typer(generate_app, name="generate")
 app.add_typer(wav2vec2_app, name="wav2vec2")
 
 
+@app.callback()
+def data():
+    """
+    data sub commands
+    """
+
+
 @app.command()
 def fix_path(dataset_path: Path, force: bool = False):
     manifest_path = dataset_path / Path("manifest.json")

plume/cli/eval.py

@@ -3,3 +3,10 @@ from ..models.wav2vec2.eval import app as wav2vec2_app
 
 app = typer.Typer()
 app.add_typer(wav2vec2_app, name="wav2vec2")
+
+
+@app.callback()
+def eval():
+    """
+    eval sub commands
+    """

plume/cli/serve.py

@@ -5,3 +5,10 @@ from ..models.jasper.serve import app as jasper_app
 app = typer.Typer()
 app.add_typer(wav2vec2_app, name="wav2vec2")
 app.add_typer(jasper_app, name="jasper")
+
+
+@app.callback()
+def serve():
+    """
+    serve sub commands
+    """

plume/cli/train.py

@@ -1,5 +1,12 @@
 import typer
-from ..models.wav2vec2.train import app as train_app
+from ..models.wav2vec2.train import app as wav2vec2_app
 
 app = typer.Typer()
-app.add_typer(train_app, name="wav2vec2")
+app.add_typer(wav2vec2_app, name="wav2vec2")
+
+
+@app.callback()
+def train():
+    """
+    train sub commands
+    """