Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -260,7 +260,7 @@ class WhisperBase(ABC):
                     file_format="txt",
                     output_dir=self.output_dir
                 )
-                #files_to_download[file_name+"_txt"] = {"path": file_path}
+                files_to_download[file_name+"_txt"] = {"path": file_path}
                 
                 ## Add output file as srt
                 file_name, file_ext = os.path.splitext(os.path.basename(file))
@@ -271,21 +271,21 @@ class WhisperBase(ABC):
                     file_format="srt",
                     output_dir=self.output_dir
                 )
-                #files_to_download[file_name+"_srt"] = {"path": file_path}
+                files_to_download[file_name+"_srt"] = {"path": file_path}
             
             total_result = ''
             total_info = ''
             total_time = 0
-            for file_name, info in files_info.items():
-                total_result += f'{info["subtitle"]}'
-                total_time += info["time_for_task"]
-                total_info += f'Input file: {info["input_source_file"]}\nLanguage prediction: {info["lang"]}\n'
-            
-            total_info += f"\nTranscription duration: {self.format_time(total_time)}"
+#            for file_name, info in files_info.items():
+#                total_result += f'{info["subtitle"]}'
+#                total_time += info["time_for_task"]
+#                total_info += f'Input file: {info["input_source_file"]}\nLanguage prediction: {info["lang"]}\n'
+#            
+#            total_info += f"\nTranscription duration: {self.format_time(total_time)}"
             result_str = total_result
             result_file_path = [info['path'] for info in files_to_download.values()]
 
-            return result_str, result_file_path, total_info
+            return [result_str,result_file_path,total_info]
         
         except Exception as e:
             print(f"Error transcribing file: {e}")