insert debug info

TalkBank · May 21, 2024 · 80b57c3 · 80b57c3
1 parent 0a29c0c
commit 80b57c3
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 9 deletions.
diff --git a/batchalign/cli/cli.py b/batchalign/cli/cli.py
@@ -149,19 +149,22 @@ def transcribe(ctx, in_dir, out_dir, lang, num_speakers, **kwargs):
     def loader(file):
         return file
 
+    asr = "rev"
+    if kwargs["whisper"]:
+        asr = "whisper"
+    if kwargs["whisperx"]:
+        asr = "whisperx"
+
+
     def writer(doc, output):
+        doc.content.insert(0, CustomLine(id="Comment", type=CustomLineType.INDEPENDENT,
+                                         content=f"Batchalign {VERSION_NUMBER.strip()}, ASR Engine {asr}"))
         CHATFile(doc=doc, special_mor_=True).write(output
                                                    .replace(".wav", ".cha")
                                                    .replace(".mp4", ".cha")
                                                    .replace(".mp3", ".cha"),
                                                    write_wor=kwargs.get("wor", False))
 
-    asr = "rev"
-    if kwargs["whisper"]:
-        asr = "whisper"
-    if kwargs["whisperx"]:
-        asr = "whisperx"
-
     if kwargs.get("diarize"):
         _dispatch("transcribe_s",
                   lang, num_speakers, ["mp3", "mp4", "wav"], ctx,

diff --git a/batchalign/version b/batchalign/version
@@ -1,3 +1,3 @@
-0.7.1-beta.6
-May 17th, 2024
-remove Hebrew for UD
+0.7.1-beta.7
+May 21st, 2024
+insert debug info to transcribe file