more robust filename-extension splitting (#114)

MahmoudAshraf97 · Oct 22, 2023 · 08b8b2e · 08b8b2e
1 parent f740cd1
commit 08b8b2e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/diarize.py b/diarize.py
@@ -65,7 +65,10 @@
         vocal_target = args.audio
     else:
         vocal_target = os.path.join(
-            "temp_outputs", "htdemucs", os.path.basename(args.audio[:-4]), "vocals.wav"
+            "temp_outputs",
+            "htdemucs",
+            os.path.splitext(os.path.basename(args.audio))[0],
+            "vocals.wav",
         )
 else:
     vocal_target = args.audio
@@ -180,10 +183,10 @@
 
 ssm = get_sentences_speaker_mapping(wsm, speaker_ts)
 
-with open(f"{args.audio[:-4]}.txt", "w", encoding="utf-8-sig") as f:
+with open(f"{os.path.splitext(args.audio)[0]}.txt", "w", encoding="utf-8-sig") as f:
     get_speaker_aware_transcript(ssm, f)
 
-with open(f"{args.audio[:-4]}.srt", "w", encoding="utf-8-sig") as srt:
+with open(f"{os.path.splitext(args.audio)[0]}.srt", "w", encoding="utf-8-sig") as srt:
     write_srt(ssm, srt)
 
 cleanup(temp_path)
diff --git a/diarize_parallel.py b/diarize_parallel.py
@@ -64,7 +64,10 @@
         vocal_target = args.audio
     else:
         vocal_target = os.path.join(
-            "temp_outputs", "htdemucs", os.path.basename(args.audio[:-4]), "vocals.wav"
+            "temp_outputs",
+            "htdemucs",
+            os.path.splitext(os.path.basename(args.audio))[0],
+            "vocals.wav",
         )
 else:
     vocal_target = args.audio
@@ -170,10 +173,10 @@
 
 ssm = get_sentences_speaker_mapping(wsm, speaker_ts)
 
-with open(f"{args.audio[:-4]}.txt", "w", encoding="utf-8-sig") as f:
+with open(f"{os.path.splitext(args.audio)[0]}.txt", "w", encoding="utf-8-sig") as f:
     get_speaker_aware_transcript(ssm, f)
 
-with open(f"{args.audio[:-4]}.srt", "w", encoding="utf-8-sig") as srt:
+with open(f"{os.path.splitext(args.audio)[0]}.srt", "w", encoding="utf-8-sig") as srt:
     write_srt(ssm, srt)
 
 cleanup(temp_path)