src/cli.py

import argparse
import os
import warnings
import subprocess


warnings.filterwarnings("ignore")


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--gradio", action="store_true", help="Launch gradio UI")
    parser.add_argument("--server", action="store_true", help="Launch flask server")
    parser.add_argument("--test", action="store_true", help="Launch test")
    parser.add_argument("--infer", help="Path to audio file for infernace")
    parser.add_argument("--audio_location", help="Path to audio files")
    args = parser.parse_args()

    if args.gradio:
        subprocess.run(["python", "src/gradio_app.py"])

    if args.infer:
        if not os.path.exists(args.infer):
            print(f"File not found: {args.infer}")
            return
        from inference_file import inference

        result = inference(args.infer)
        print(result)
    if args.test:
        if args.audio_location:
            if not os.path.exists(args.audio_location):
                print(f"Folder not found: {args.audio_location}")
                return
            os.environ["AUDIO_LOCATION"] = args.audio_location
            subprocess.run(["python", "src/model_test.py"])
        else:
            print("Missings option --audio_location")
    if args.server:
        subprocess.run(["python", "src/flask_server.py"])


if __name__ == "__main__":
    main()