add dir parameters

app.py

@@ -18,9 +18,10 @@ class App:
         self.app = gr.Blocks(css=CSS, theme=self.args.theme)
         self.whisper_inf = WhisperFactory.create_whisper_inference(
             whisper_type=self.args.whisper_type,
-            model_dir=self.args.faster_whisper_model_dir,
+            whisper_model_dir=self.args.whisper_model_dir,
+            faster_whisper_model_dir=self.args.faster_whisper_model_dir,
+            insanely_fast_whisper_model_dir=self.args.insanely_fast_whisper_model_dir,
             output_dir=self.args.output_dir,
-            args=self.args
         )
         print(f"Use \"{self.args.whisper_type}\" implementation")
         print(f"Device \"{self.whisper_inf.device}\" is detected")

modules/whisper/whisper_factory.py

@@ -1,4 +1,4 @@
-from argparse import Namespace
+from typing import Optional
 import os
 
 from modules.whisper.faster_whisper_inference import FasterWhisperInference
@@ -11,27 +11,32 @@ class WhisperFactory:
     @staticmethod
     def create_whisper_inference(
         whisper_type: str,
-        model_dir: str,
-        output_dir: str,
-        args: Namespace
+        whisper_model_dir: Optional[str] = None,
+        faster_whisper_model_dir: Optional[str] = None,
+        insanely_fast_whisper_model_dir: Optional[str] = None,
+        diarization_model_dir: Optional[str] = None,
+        output_dir: Optional[str] = None,
     ) -> "WhisperBase":
         """
         Create a whisper inference class based on the provided whisper_type.
 
         Parameters
         ----------
-        whisper_type: str
-            The repository name of whisper inference to use. Supported values are:
-            - "faster-whisper" from
-            - "whisper"
-            - insanely-fast-whisper", "insanely_fast_whisper", "insanelyfastwhisper",
-              "insanely-faster-whisper", "insanely_faster_whisper", "insanelyfasterwhisper"
-        model_dir: str
-            The directory path where the whisper model is located.
-        output_dir: str
-            The directory path where the output files will be saved.
-        args: Any
-            Additional arguments to be passed to the whisper inference object.
+        whisper_type : str
+            The type of Whisper implementation to use. Supported values (case-insensitive):
+            - "faster-whisper": https://github.com/openai/whisper
+            - "whisper": https://github.com/openai/whisper
+            - "insanely-fast-whisper": https://github.com/Vaibhavs10/insanely-fast-whisper
+        whisper_model_dir : str
+            Directory path for the Whisper model.
+        faster_whisper_model_dir : str
+            Directory path for the Faster Whisper model.
+        insanely_fast_whisper_model_dir : str
+            Directory path for the Insanely Fast Whisper model.
+        diarization_model_dir : str
+            Directory path for the diarization model.
+        output_dir : str
+            Directory path where output files will be saved.
 
         Returns
         -------
@@ -51,10 +56,26 @@ class WhisperFactory:
         ]
 
         if whisper_type in faster_whisper_typos:
-            return FasterWhisperInference(model_dir, output_dir, args)
+            return FasterWhisperInference(
+                model_dir=faster_whisper_model_dir,
+                output_dir=output_dir,
+                diarization_model_dir=diarization_model_dir
+            )
         elif whisper_type in whisper_typos:
-            return WhisperInference(model_dir, output_dir, args)
+            return WhisperInference(
+                model_dir=whisper_model_dir,
+                output_dir=output_dir,
+                diarization_model_dir=diarization_model_dir
+            )
         elif whisper_type in insanely_fast_whisper_typos:
-            return InsanelyFastWhisperInference(model_dir, output_dir, args)
+            return InsanelyFastWhisperInference(
+                model_dir=insanely_fast_whisper_model_dir,
+                output_dir=output_dir,
+                diarization_model_dir=diarization_model_dir
+            )
         else:
-            return FasterWhisperInference(model_dir, output_dir, args)
+            return FasterWhisperInference(
+                model_dir=faster_whisper_model_dir,
+                output_dir=output_dir,
+                diarization_model_dir=diarization_model_dir
+            )