Refactor Whisper logic into individual files

CrimeIsDown · Sep 2, 2024 · 00a99e8 · 00a99e8
1 parent fbe4828
commit 00a99e8
Show file tree

Hide file tree

Showing 17 changed files with 668 additions and 593 deletions.
diff --git a/app/analog.py b/app/analog.py
@@ -2,7 +2,8 @@
 from threading import Lock
 
 from .transcript import Transcript
-from .whisper import WhisperResult, transcribe
+from .whisper.base import WhisperResult
+from .whisper.transcribe import transcribe
 
 
 def build_transcribe_kwargs(audio_file: str, initial_prompt: str = "") -> dict:

diff --git a/app/api.py b/app/api.py
@@ -31,7 +31,7 @@
     transcribe_from_db_task,
     transcribe_from_db_batch_task,
 )
-from .whisper import WhisperTask
+from .whisper.task import WhisperTask
 
 sentry_dsn = os.getenv("SENTRY_DSN")
 if sentry_dsn:

diff --git a/app/digital.py b/app/digital.py
@@ -3,7 +3,8 @@
 
 from .metadata import Metadata, SrcListItem
 from .transcript import Transcript
-from .whisper import WhisperResult, WhisperSegment, transcribe
+from .whisper.base import WhisperSegment, WhisperResult
+from .whisper.transcribe import transcribe
 
 
 def get_closest_src(srcList: list[SrcListItem], segment: WhisperSegment):