dasemu/Transcriptarr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update subgen.py

Browse Source
This commit is contained in:
McCloudS
2023-10-19 16:12:04 -06:00
committed by GitHub
parent 44eb52b62b
commit 82fb0126f8
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
subgen/subgen.py
View File

@@ -91,7 +91,7 @@ def gen_subtitles(inputvideo):
try:
print(f"Transcribing file: {inputvideo}")
result = model.transcribe_stable(inputvideo)
result.to_srt_vtt(inputvideo + subextension, word_level=word_level_highlight)
result.to_srt_vtt(inputvideo.rsplit('.', 1)[0] + ' ' + subextension, word_level=word_level_highlight)
print(f"Transcription of {file_path} is completed.")
files_to_transcribe.remove(inputvideo)
except Exception as e:
@@ -104,7 +104,7 @@ def add_file_for_transcription(file_path):
if has_subtitle_language(file_path, skipifinternalsublang):
print("File already has an internal sub we want, skipping generation")
return "File already has an internal sub we want, skipping generation"
elif os.path.exists(file_path + subextension):
elif os.path.exists(file_path.rsplit('.', 1)[0] + ' ' + subextension):
print("We already have a subgen created for this file, skipping it")
return "We already have a subgen created for this file, skipping it"