Create pipeline.py

pipeline.py

@@ -0,0 +1,20 @@
+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"