Update modules/whisper/whisper_base.py
Browse files
modules/whisper/whisper_base.py
CHANGED
@@ -286,12 +286,15 @@ class WhisperBase(ABC):
|
|
286 |
Output file path to return to gr.Files()
|
287 |
"""
|
288 |
|
289 |
-
file_count_total = 0
|
290 |
-
if input_multi != "Single":
|
291 |
-
files = files_multi
|
292 |
-
file_count_total = len(files)
|
293 |
-
|
294 |
try:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
295 |
if input_folder_path:
|
296 |
files = get_media_files(input_folder_path)
|
297 |
if isinstance(files, str):
|
@@ -362,9 +365,10 @@ class WhisperBase(ABC):
|
|
362 |
else:
|
363 |
translation_note = "Already in " + target_lang
|
364 |
|
365 |
-
## Get
|
366 |
file_name, file_ext = os.path.splitext(os.path.basename(file))
|
367 |
-
|
|
|
368 |
if add_timestamp_preview:
|
369 |
subtitle = get_txt(transcribed_segments)
|
370 |
else:
|
@@ -373,7 +377,6 @@ class WhisperBase(ABC):
|
|
373 |
|
374 |
## Add output file as txt, srt and/or csv
|
375 |
for output_format in file_format:
|
376 |
-
file_name, file_ext = os.path.splitext(os.path.basename(file))
|
377 |
subtitle, file_path = self.generate_and_write_file(
|
378 |
file_name=file_name,
|
379 |
transcribed_segments=transcribed_segments,
|
|
|
286 |
Output file path to return to gr.Files()
|
287 |
"""
|
288 |
|
|
|
|
|
|
|
|
|
|
|
289 |
try:
|
290 |
+
for file in files:
|
291 |
+
print("File to process: "+str(file))
|
292 |
+
|
293 |
+
file_count_total = 0
|
294 |
+
if input_multi != "Single":
|
295 |
+
files = files_multi
|
296 |
+
file_count_total = len(files)
|
297 |
+
|
298 |
if input_folder_path:
|
299 |
files = get_media_files(input_folder_path)
|
300 |
if isinstance(files, str):
|
|
|
365 |
else:
|
366 |
translation_note = "Already in " + target_lang
|
367 |
|
368 |
+
## Get input filename & extension
|
369 |
file_name, file_ext = os.path.splitext(os.path.basename(file))
|
370 |
+
|
371 |
+
## Get output as preview with or without timestamps
|
372 |
if add_timestamp_preview:
|
373 |
subtitle = get_txt(transcribed_segments)
|
374 |
else:
|
|
|
377 |
|
378 |
## Add output file as txt, srt and/or csv
|
379 |
for output_format in file_format:
|
|
|
380 |
subtitle, file_path = self.generate_and_write_file(
|
381 |
file_name=file_name,
|
382 |
transcribed_segments=transcribed_segments,
|