pipeline.py

from utils_audio import convert_to_wav
from to_cha import to_cha_from_wav
from cha_json import cha_to_json_path
from model_infer import AphasiaClassifier
from output import format_result

def run_pipeline(in_media_path: str, out_style: str = "json") -> str:
    wav = convert_to_wav(in_media_path, sr=16000, mono=True)
    cha = to_cha_from_wav(wav, lang="eng")
    jsn = cha_to_json_path(cha)

    clf = AphasiaClassifier()
    pred = clf.predict_from_json(jsn)

    return format_result(pred, style=out_style)

if __name__ == "__main__":
    import sys
    media = sys.argv[1]
    print(run_pipeline(media, out_style="text"))  # or "json"