kabigon 0.8.2__py3-none-any.whl → 0.8.3__py3-none-any.whl

kabigon/ytdlp.py

@@ -1,45 +1,15 @@
-# import functools
 import hashlib
 import os
-
-# import subprocess
 import tempfile
-from typing import Final
 
-# import numpy as np
 import yt_dlp
 from loguru import logger
 
 from .loader import Loader
 
-try:
-    import whisper
-except ImportError:
-    logger.info("OpenAI Whisper not installed. Please install it with `pip install openai-whisper`.")
-
-DEFAULT_FFMPEG_PATH: Final[str] = "ffmpeg"
-
-
-def hash_url(url: str) -> str:
-    return hashlib.sha512(url.encode("utf-8")).hexdigest()
-
-
-def get_ffmpeg_path() -> str:
-    path = os.getenv("FFMPEG_PATH")
-    if not path:
-        path = DEFAULT_FFMPEG_PATH
-        logger.warning("FFMPEG_PATH not set, using default: {}", DEFAULT_FFMPEG_PATH)
-
-    return path
-
 
 def download_audio(url: str) -> str:
-    ffmpeg_path = get_ffmpeg_path()
-
-    filename = os.path.join(
-        tempfile.gettempdir(),
-        hash_url(url),
-    )
+    filename = os.path.join(tempfile.gettempdir(), hashlib.sha512(url.encode("utf-8")).hexdigest())
 
     ydl_opts = {
         "format": "bestaudio/best",
@@ -51,7 +21,7 @@ def download_audio(url: str) -> str:
             }
         ],
         "outtmpl": filename,
-        "ffmpeg_location": ffmpeg_path,
+        "ffmpeg_location": os.getenv("FFMPEG_PATH", "ffmpeg"),
         "match_filter": yt_dlp.match_filter_func(["!is_live"]),
     }
 
@@ -61,58 +31,25 @@ def download_audio(url: str) -> str:
     return filename + ".mp3"
 
 
-# def load_audio(file: str, sr: int = 16000):
-#     """
-#     Open an audio file and read as mono waveform, resampling as necessary
-
-#     Parameters
-#     ----------
-#     file: str
-#         The audio file to open
-
-#     sr: int
-#         The sample rate to resample the audio if necessary
-
-#     Returns
-#     -------
-#     A NumPy array containing the audio waveform, in float32 dtype.
-#     """
-#     ffmpeg_path = get_ffmpeg_path()
-
-#     # This launches a subprocess to decode audio while down-mixing
-#     # and resampling as necessary.  Requires the ffmpeg CLI in PATH.
-#     # fmt: off
-#     cmd = [
-#         ffmpeg_path,
-#         "-nostdin",
-#         "-threads", "0",
-#         "-i", file,
-#         "-f", "s16le",
-#         "-ac", "1",
-#         "-acodec", "pcm_s16le",
-#         "-ar", str(sr),
-#         "-"
-#     ]
-#     # fmt: on
-#     try:
-#         out = subprocess.run(cmd, capture_output=True, check=True).stdout
-#     except subprocess.CalledProcessError as e:
-#         raise RuntimeError(f"Failed to load audio: {e.stderr.decode()}") from e
-
-#     return np.frombuffer(out, np.int16).flatten().astype(np.float32) / 32768.0
-
-
 class YtdlpLoader(Loader):
     def __init__(self, model: str = "tiny") -> None:
+        try:
+            import whisper
+        except ImportError as e:
+            raise ImportError(
+                "OpenAI Whisper not installed. Please install it with `pip install openai-whisper`."
+            ) from e
+
         self.model = whisper.load_model(model)
+        self.load_audio = whisper.load_audio
 
     def load(self, url: str) -> str:
         audio_file = download_audio(url)
-        # audio = load_audio(audio_file)
-        audio = whisper.load_audio(audio_file)
+        audio = self.load_audio(audio_file)
 
         # Clean up the audio file
         os.remove(audio_file)
 
+        logger.info("Transcribing audio file: {}", audio_file)
         result = self.model.transcribe(audio)
         return result.get("text", "")

{kabigon-0.8.2.dist-info → kabigon-0.8.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kabigon
-Version: 0.8.2
+Version: 0.8.3
 Author-email: narumi <toucans-cutouts0f@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.10
@@ -29,7 +29,7 @@ Description-Content-Type: text/markdown
 ## Installation
 
 ```shell
-pip install kabigon
+pip install kabigon[all]
 playwright install chromium
 ```
 

{kabigon-0.8.2.dist-info → kabigon-0.8.3.dist-info}/RECORD

@@ -12,9 +12,9 @@ kabigon/reel.py,sha256=qOwWCvcp7xNKg0JDunq_Bsl8yqqMzrnAOI9k5mSqrOU,874
 kabigon/twitter.py,sha256=aRqAiFxIwln6lteWdoF6SmvbzO62yBTQRzcB7UcVJwk,1046
 kabigon/utils.py,sha256=eNTLtHLSB2erDac2HH3jWemgfr8Ou_ozwVb8h9BD-4g,922
 kabigon/youtube.py,sha256=F9GpLa0iUy03wYU94RrrnrXa6ExqbG6CZpqx5bPENWE,1106
-kabigon/ytdlp.py,sha256=oeFoE7nWZWaT0dR9nwt_SZh_FE0gJ6Gulh6QzGCB6xo,2956
-kabigon-0.8.2.dist-info/METADATA,sha256=UEq0yUqP0dgzyqMhxiHgLvP0wT_nf0PlQLy8VkufrEg,1287
-kabigon-0.8.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-kabigon-0.8.2.dist-info/entry_points.txt,sha256=O3FYAO9w-NQvlGMJrBvtrnGHSK2QkUnQBTa30YXRbVE,45
-kabigon-0.8.2.dist-info/licenses/LICENSE,sha256=H2T3_RTgmcngMeC7p_SXT3GwBLkd2DaNgAZuxulcfiA,1066
-kabigon-0.8.2.dist-info/RECORD,,
+kabigon/ytdlp.py,sha256=OGVrKRGXY_eHePgDmN52Kkui12NN44LsIPonD_61QBM,1501
+kabigon-0.8.3.dist-info/METADATA,sha256=uidye_pVnhpAivxY26MlZMSascT1MmuMjBuCAw84Ics,1292
+kabigon-0.8.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+kabigon-0.8.3.dist-info/entry_points.txt,sha256=O3FYAO9w-NQvlGMJrBvtrnGHSK2QkUnQBTa30YXRbVE,45
+kabigon-0.8.3.dist-info/licenses/LICENSE,sha256=H2T3_RTgmcngMeC7p_SXT3GwBLkd2DaNgAZuxulcfiA,1066
+kabigon-0.8.3.dist-info/RECORD,,