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

kabigon/ytdlp.py

@@ -1,6 +1,5 @@
-import hashlib
 import os
-import tempfile
+import uuid
 from pathlib import Path
 
 import yt_dlp
@@ -26,9 +25,10 @@ def download_audio(url: str, outtmpl: str | None = None) -> None:
         ydl_opts["outtmpl"] = outtmpl
 
     ffmpeg_path = os.getenv("FFMPEG_PATH")
-    if ffmpeg_path is None:
+    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])
 
@@ -46,16 +46,16 @@ class YtdlpLoader(Loader):
         self.load_audio = whisper.load_audio
 
     def load(self, url: str) -> str:
-        temp_dir = Path(tempfile.gettempdir())
-        outtmpl = hashlib.sha512(url.encode("utf-8")).hexdigest()
-        path = (temp_dir / outtmpl).with_suffix(".mp3")
-
+        outtmpl = uuid.uuid4().hex[:20]
+        path = str(Path(outtmpl).with_suffix(".mp3"))
         download_audio(url, outtmpl=outtmpl)
-        audio = self.load_audio(path)
 
-        # Clean up the audio file
-        os.remove(path)
+        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: {}", path)
-        result = self.model.transcribe(audio)
         return result.get("text", "")

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

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

{kabigon-0.8.9.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=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,,
+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,,