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: try:
print(f"Transcribing file: {inputvideo}") print(f"Transcribing file: {inputvideo}")
result = model.transcribe_stable(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.") print(f"Transcription of {file_path} is completed.")
files_to_transcribe.remove(inputvideo) files_to_transcribe.remove(inputvideo)
except Exception as e: except Exception as e:
@@ -104,7 +104,7 @@ def add_file_for_transcription(file_path):
if has_subtitle_language(file_path, skipifinternalsublang): if has_subtitle_language(file_path, skipifinternalsublang):
print("File already has an internal sub we want, skipping generation") print("File already has an internal sub we want, skipping generation")
return "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") print("We already have a subgen created for this file, skipping it")
return "We already have a subgen created for this file, skipping it" return "We already have a subgen created for this file, skipping it"