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

kabigon/ytdlp.py

@@ -1,6 +1,6 @@
-import hashlib
 import os
-import tempfile
+import uuid
+from pathlib import Path
 
 import yt_dlp
 from loguru import logger
@@ -8,9 +8,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 +18,20 @@ 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 not None:
+        ydl_opts["ffmpeg_location"] = ffmpeg_path
+
+    logger.info("Downloading audio from URL: {} with options: {}", url, ydl_opts)
     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)
+        outtmpl = uuid.uuid4().hex[:20]
+        path = str(Path(outtmpl).with_suffix(".mp3"))
+        download_audio(url, outtmpl=outtmpl)
 
-        # Clean up the audio file
-        os.remove(audio_file)
+        try:
+            audio = self.load_audio(path)
+            logger.info("Transcribing audio file: {}", path)
+            result = self.model.transcribe(audio)
+        finally:
+            # Clean up the audio file
+            os.remove(path)
 
-        logger.info("Transcribing audio file: {}", audio_file)
-        result = self.model.transcribe(audio)
         return result.get("text", "")

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

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

{kabigon-0.8.8.dist-info → kabigon-0.8.10.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=4AjPY0W8IxbEhCNVXpi_KbBQLGrWn5ThrchhffcJIIU,1718
+kabigon-0.8.10.dist-info/METADATA,sha256=XGIIEjC0MWovjzhfXrnXsBE3XxGN2V82Y9Y3eiQTjpQ,1265
+kabigon-0.8.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+kabigon-0.8.10.dist-info/entry_points.txt,sha256=O3FYAO9w-NQvlGMJrBvtrnGHSK2QkUnQBTa30YXRbVE,45
+kabigon-0.8.10.dist-info/licenses/LICENSE,sha256=H2T3_RTgmcngMeC7p_SXT3GwBLkd2DaNgAZuxulcfiA,1066
+kabigon-0.8.10.dist-info/RECORD,,