Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -433,7 +433,7 @@ class WhisperBase(ABC):
             result_str = total_result.rstrip("\n")
             result_file_path = [info['path'] for info in files_to_download.values()]
             
-            return [result_str,result_file_path,total_info]
+            return [self.transform_text_to_list(result_str),result_file_path,total_info]
         
         except Exception as e:
             print(f"Error transcribing file: {e}")
@@ -624,6 +624,19 @@ class WhisperBase(ABC):
         if self.device == "cuda":
             self.release_cuda_memory()
         gc.collect()
+
+    def transform_text_to_list(inputdata):
+    
+    outputdata = []
+    temp_inputdata = (inputdata.strip("\n")).splitlines()
+        for temp_line in temp_inputdata:
+        temp_line_list = []	
+        temp_line_items = temp_line.split("\t")
+        for temp_line_item in temp_line_items:
+            temp_line_list.append(temp_line_item)	
+        outputdata.append(temp_line_list)
+    
+    return outputdata 
     
     @staticmethod
     def format_time(elapsed_time: float) -> str: