giers10/Heimgeist
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add runtime error handling for whisper package in corpus_builder.py

Browse Source
This commit is contained in:
Victor Giers
2026-03-19 23:01:42 +01:00
parent 47f1e165da
commit 927f2033bf
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
backend/rag/corpus_builder.py
View File

@@ -1416,6 +1416,14 @@ def slice_audio(audio_path: Path, out_dir: Path, num_slices: int, overlap_sec: f
_WHISPER_MODEL = None _WHISPER_MODEL = None
def whisper_runtime_error() -> Optional[str]:
if whisper is None:
return (
"Audio/video transcription requires the optional 'openai-whisper' package. "
"Install it in backend/.venv, for example: pip install -U openai-whisper"
)
return None
def _resolve_whisper_device(flag: str) -> Optional[str]: def _resolve_whisper_device(flag: str) -> Optional[str]:
if flag and flag != "auto": if flag and flag != "auto":
return flag return flag
@@ -1467,6 +1475,10 @@ def merge_transcripts(files_idx_text: List[Tuple[int, str]], max_overlap_words:
return " ".join(merged_words).strip() return " ".join(merged_words).strip()
def process_media(path: Path, args) -> List[Record]: def process_media(path: Path, args) -> List[Record]:
whisper_error = whisper_runtime_error()
if whisper_error:
raise RuntimeError(whisper_error)
probe = ffprobe_json(args.ffprobe, path) probe = ffprobe_json(args.ffprobe, path)
duration_s = None duration_s = None
if probe: if probe: