Update src/main.py

src/main.py

@@ -62,20 +62,24 @@ def get_youtube_video_id(url, ignore_playlist=True): # this code unused here!
 
 def yt_download(link):
     ydl_opts = {
-        'format': 'bestaudio',
-        'outtmpl': '%(title)s.%(ext)s',
-
-        'extractaudio': True,
-        'postprocessors': [{'key': 'FFmpegExtractAudio', 'preferredcodec': 'mp3'}],
+        'format': 'bestaudio/best',
+        'outtmpl': 'ytdl/%(title)s.%(ext)s',
+        'postprocessors': [{
+            'key': 'FFmpegExtractAudio',
+            'preferredcodec': 'wav',
+            'preferredquality': '192',
+        }],
     }
-    try:
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            result = ydl.extract_info(link, download=True)
-            download_path = ydl.prepare_filename(result, outtmpl='%(title)s.mp3')
-    except Exception as e:
-        raise Exception(f"Error downloading video: {e}")
-    
-    return download_path
+
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        info_dict = ydl.extract_info(url, download=True)
+        download_path = ydl.prepare_filename(info_dict).rsplit('.', 1)[0] + '.wav'
+        sample_rate, audio_data = read(file_path)
+        audio_array = np.asarray(audio_data, dtype=np.int16)
+
+        return download_path
+
+