refactor without type hint

modules/whisper/whisper_base.py

@@ -130,7 +130,8 @@ class WhisperBase(ABC):
                         files,
                         file_format,
                         add_timestamp,
-                        #*whisper_params,
+                        *whisper_params,
+                        progress=gr.Progress(),
                         ):
         """
         Write subtitle file from Files
@@ -155,49 +156,48 @@ class WhisperBase(ABC):
         result_file_path:
             Output file path to return to gr.Files()
         """
-        print("transcrption started: ")
-        # try:
-        #     files_info = {}
-        #     for file in files:
-        #         print("run started: ")
-        #         transcribed_segments, time_for_task = self.run(
-        #             file.name,
-        #             progress,
-        #             *whisper_params,
-        #         )
-        #         print("run finished: ")
-        #
-        #         file_name, file_ext = os.path.splitext(os.path.basename(file.name))
-        #         file_name = safe_filename(file_name)
-        #         subtitle, file_path = self.generate_and_write_file(
-        #             file_name=file_name,
-        #             transcribed_segments=transcribed_segments,
-        #             add_timestamp=add_timestamp,
-        #             file_format=file_format,
-        #             output_dir=self.output_dir
-        #         )
-        #         print("generated sub finished: ")
-        #         files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path}
-        #
-        #     total_result = ''
-        #     total_time = 0
-        #     for file_name, info in files_info.items():
-        #         total_result += '------------------------------------\n'
-        #         total_result += f'{file_name}\n\n'
-        #         total_result += f'{info["subtitle"]}'
-        #         total_time += info["time_for_task"]
-        #
-        #     result_str = f"Done in {self.format_time(total_time)}! Subtitle is in the outputs folder.\n\n{total_result}"
-        #     result_file_path = [info['path'] for info in files_info.values()]
-        #
-        #     return [result_str, result_file_path]
-        #
-        # except Exception as e:
-        #     print(f"Error transcribing file: {e}")
-        # finally:
-        #     self.release_cuda_memory()
-        #     if not files:
-        #         self.remove_input_files([file.name for file in files])
+        try:
+            files_info = {}
+            for file in files:
+                print("run started: ")
+                transcribed_segments, time_for_task = self.run(
+                    file.name,
+                    progress,
+                    *whisper_params,
+                )
+                print("run finished: ")
+
+                file_name, file_ext = os.path.splitext(os.path.basename(file.name))
+                file_name = safe_filename(file_name)
+                subtitle, file_path = self.generate_and_write_file(
+                    file_name=file_name,
+                    transcribed_segments=transcribed_segments,
+                    add_timestamp=add_timestamp,
+                    file_format=file_format,
+                    output_dir=self.output_dir
+                )
+                print("generated sub finished: ")
+                files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path}
+
+            total_result = ''
+            total_time = 0
+            for file_name, info in files_info.items():
+                total_result += '------------------------------------\n'
+                total_result += f'{file_name}\n\n'
+                total_result += f'{info["subtitle"]}'
+                total_time += info["time_for_task"]
+
+            result_str = f"Done in {self.format_time(total_time)}! Subtitle is in the outputs folder.\n\n{total_result}"
+            result_file_path = [info['path'] for info in files_info.values()]
+
+            return [result_str, result_file_path]
+
+        except Exception as e:
+            print(f"Error transcribing file: {e}")
+        finally:
+            self.release_cuda_memory()
+            if not files:
+                self.remove_input_files([file.name for file in files])
 
     @spaces.GPU(duration=120)
     def transcribe_mic(self,