Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -251,7 +251,7 @@ class WhisperBase(ABC):
 #                
 #                files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language}
 
-                ## Add 1st file as txt
+                ## Add output file as txt
                 file_name, file_ext = os.path.splitext(os.path.basename(file))
                 subtitle, file_path = self.generate_and_write_file(
                     file_name=file_name,
@@ -260,10 +260,9 @@ class WhisperBase(ABC):
                     file_format="txt",
                     output_dir=self.output_dir
                 )				
+                files_info[file_name+"_txt"] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language}
                 
-                files_info["txt_file"] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language}
-                
-                ## Add 2nd file as srt
+                ## Add output file as srt
                 file_name, file_ext = os.path.splitext(os.path.basename(file))
                 subtitle, file_path = self.generate_and_write_file(
                     file_name=file_name,
@@ -272,8 +271,9 @@ class WhisperBase(ABC):
                     file_format="srt",
                     output_dir=self.output_dir
                 )				
-                
-                files_info["srt_file"] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language}
+                files_info[file_name+"_srt"] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language}
+
+                break ## Processing of 1 line
             
             total_result = ''
             total_info = ''
@@ -282,14 +282,14 @@ class WhisperBase(ABC):
                 total_result += f'{info["subtitle"]}'
                 total_time += info["time_for_task"]
                 #total_info += f'{info["lang"]}'
-                total_info += f"Language {info['lang']} detected for file '{file_name}{file_ext}'" 
+                total_info += f"Language {info['lang']} detected for file '{file_name}{file_ext}'"
+                break ## Processing of 1 line
             
             #result_str = f"Processing of file '{file_name}{file_ext}' done in {self.format_time(total_time)}:\n\n{total_result}"
             total_info += f"\nTranscription process done in {self.format_time(total_time)}"
             result_str = total_result
             result_file_path = [info['path'] for info in files_info.values()]
 
-            total_info +=str(len(files_info))
             return [result_str, result_file_path, total_info]
 
         except Exception as e: