PR: Catching pydub's warning of ffmpeg or avconv missing

microsoft · Dec 15, 2024 · b7f5662 · b7f5662
1 parent 81e3f24
commit b7f5662
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/markitdown/_markitdown.py b/src/markitdown/_markitdown.py
@@ -14,6 +14,7 @@
 import traceback
 from typing import Any, Dict, List, Optional, Union
 from urllib.parse import parse_qs, quote, unquote, urlparse, urlunparse
+from warnings import catch_warnings
 
 import mammoth
 import markdownify
@@ -29,7 +30,12 @@
 
 # Optional Transcription support
 try:
-    import pydub
+    # Using warnings' catch_warnings to catch
+    # pydub's warning of ffmpeg or avconv missing
+    with catch_warnings(record=True) as w:
+        import pydub
+        if w:
+            raise ModuleNotFoundError
     import speech_recognition as sr
 
     IS_AUDIO_TRANSCRIPTION_CAPABLE = True