testing

App/Transcription/TranscriptionRoutes.py

@@ -2,7 +2,7 @@ from fastapi import APIRouter, status, Form, UploadFile, File, Query, Background
 from typing_extensions import Annotated
 from .Schemas import UserDetails
 from App import bot
-import aiofiles, os
+import aiofiles, os, re
 import tempfile
 from celery.result import AsyncResult
 from App.Worker import transcription_task, downloadfile
@@ -30,10 +30,33 @@ async def download_audio(
     user = await User.objects.filter(id=userId).first()
     if user == None:
         return {"code": 400, "message": "doesn't exist", "payload": None}
+
+    ydl_opts_info = {
+        "quiet": True,
+    }
+
+    with youtube_dl.YoutubeDL(ydl_opts_info) as ydl:
+        info_dict = ydl.extract_info(url, download=False)
+        video_title = info_dict.get("title", None)
+
+    sanitized_title = re.sub(
+        r"(?u)[^-\w.]", "", video_title
+    )  # Ensure the title is file-friendly
+    filename = f"{sanitized_title}.mp3"
+    file_path = os.path.join(os.path.expanduser("~"), "Downloads", filename)
+
     ydl_opts = {
         "format": "bestaudio/best",
-        "outtmpl": os.path.join("./", "%(title)s.%(ext)s"),
+        "postprocessors": [
+            {
+                "key": "FFmpegExtractAudio",
+                "preferredcodec": "mp3",
+                "preferredquality": "192",
+            }
+        ],
+        "outtmpl": file_path,
     }
+
     task = downloadfile.delay(url, ydl_opts, model)
     transcription_enrty = await Transcriptions.objects.create(
         task_id=task.id, user=user

App/Worker.py

@@ -21,11 +21,8 @@ def downloadfile(self, url, ydl_opts, model_size="tiny"):
     self.update_state(state="Downloading File..", meta={})
 
     ####
-    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        info = ydl.extract_info(url, download=True)
-        audio_file = ydl.prepare_filename(info)
-        # final_audio_file = f"{audio_file}.mp3"
-        # os.rename(final_audio_file, audio_file)
+    with youtube_dl.YoutubeDL(ydl_opts) as ydl:
+        ydl.download([url])
 
     # updated
     self.update_state(state="Downloading complete", meta={})