kabigon 0.8.8__py3-none-any.whl → 0.8.9__py3-none-any.whl

kabigon/ytdlp.py

@@ -1,6 +1,7 @@
 import hashlib
 import os
 import tempfile
+from pathlib import Path
 
 import yt_dlp
 from loguru import logger
@@ -8,9 +9,7 @@ from loguru import logger
 from .loader import Loader
 
 
-def download_audio(url: str) -> str:
-    filename = os.path.join(tempfile.gettempdir(), hashlib.sha512(url.encode("utf-8")).hexdigest())
-
+def download_audio(url: str, outtmpl: str | None = None) -> None:
     ydl_opts = {
         "format": "bestaudio/best",
         "postprocessors": [
@@ -20,16 +19,19 @@ def download_audio(url: str) -> str:
                 "preferredquality": "192",
             }
         ],
-        "outtmpl": filename,
-        "ffmpeg_location": os.getenv("FFMPEG_PATH", "ffmpeg"),
         "match_filter": yt_dlp.match_filter_func(["!is_live"]),
     }
 
+    if outtmpl is not None:
+        ydl_opts["outtmpl"] = outtmpl
+
+    ffmpeg_path = os.getenv("FFMPEG_PATH")
+    if ffmpeg_path is None:
+        ydl_opts["ffmpeg_location"] = ffmpeg_path
+
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         ydl.download([url])
 
-    return filename + ".mp3"
-
 
 class YtdlpLoader(Loader):
     def __init__(self, model: str = "tiny") -> None:
@@ -44,12 +46,16 @@ class YtdlpLoader(Loader):
         self.load_audio = whisper.load_audio
 
     def load(self, url: str) -> str:
-        audio_file = download_audio(url)
-        audio = self.load_audio(audio_file)
+        temp_dir = Path(tempfile.gettempdir())
+        outtmpl = hashlib.sha512(url.encode("utf-8")).hexdigest()
+        path = (temp_dir / outtmpl).with_suffix(".mp3")
+
+        download_audio(url, outtmpl=outtmpl)
+        audio = self.load_audio(path)
 
         # Clean up the audio file
-        os.remove(audio_file)
+        os.remove(path)
 
-        logger.info("Transcribing audio file: {}", audio_file)
+        logger.info("Transcribing audio file: {}", path)
         result = self.model.transcribe(audio)
         return result.get("text", "")

{kabigon-0.8.8.dist-info → kabigon-0.8.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kabigon
-Version: 0.8.8
+Version: 0.8.9
 Author-email: narumi <toucans-cutouts0f@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.10

{kabigon-0.8.8.dist-info → kabigon-0.8.9.dist-info}/RECORD

@@ -13,9 +13,9 @@ kabigon/twitter.py,sha256=aRqAiFxIwln6lteWdoF6SmvbzO62yBTQRzcB7UcVJwk,1046
 kabigon/utils.py,sha256=eNTLtHLSB2erDac2HH3jWemgfr8Ou_ozwVb8h9BD-4g,922
 kabigon/youtube.py,sha256=F9GpLa0iUy03wYU94RrrnrXa6ExqbG6CZpqx5bPENWE,1106
 kabigon/youtube_ytdlp.py,sha256=Y6h55wYF-5PdxLRxsznFtJDypCedD8g-hY9fgCnfvbg,722
-kabigon/ytdlp.py,sha256=OGVrKRGXY_eHePgDmN52Kkui12NN44LsIPonD_61QBM,1501
-kabigon-0.8.8.dist-info/METADATA,sha256=HtqjSawC0X_3f0dBHzoTvx_aJ_JGuz_mdlZDhjigOBY,1264
-kabigon-0.8.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-kabigon-0.8.8.dist-info/entry_points.txt,sha256=O3FYAO9w-NQvlGMJrBvtrnGHSK2QkUnQBTa30YXRbVE,45
-kabigon-0.8.8.dist-info/licenses/LICENSE,sha256=H2T3_RTgmcngMeC7p_SXT3GwBLkd2DaNgAZuxulcfiA,1066
-kabigon-0.8.8.dist-info/RECORD,,
+kabigon/ytdlp.py,sha256=ex_at_1irwHgcl20Mm0mRZp-rTM92OHnVC5dWQCkI1c,1677
+kabigon-0.8.9.dist-info/METADATA,sha256=T8EHDM6dcghksnn6mOQ1548AvmYBfYlCefo4eIF9DBo,1264
+kabigon-0.8.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+kabigon-0.8.9.dist-info/entry_points.txt,sha256=O3FYAO9w-NQvlGMJrBvtrnGHSK2QkUnQBTa30YXRbVE,45
+kabigon-0.8.9.dist-info/licenses/LICENSE,sha256=H2T3_RTgmcngMeC7p_SXT3GwBLkd2DaNgAZuxulcfiA,1066
+kabigon-0.8.9.dist-info/RECORD,,