Create download_music.py

tools/download_music.py

@@ -0,0 +1,50 @@
+from typing import Any, Optional
+from smolagents.tools import Tool
+
+import yt_dlp
+import json
+import time
+
+avar = ""
+
+def my_hook(d):
+    if d['status'] == 'finished':
+        print("")
+        print(f"Download complete. File saved to: {d['filename']}")
+        print(f"A next step can be to play this file: {d['filename'].replace('.webm', '.mp3')}")
+        avar = d['filename']
+        
+class DownloadSomeMusicMeastro(Tool):
+    name = "download_music"
+    description = "Downloads a mp3 file which can be played later to the listener. This tool does not play the music, that will be done in the final answer"
+    inputs = {'youtube_id': {'type': 'string', 'description': 'The id of the song to play.'}}
+    output_type = "string" # because it is actualy a side-effect what this tool does do it download a file
+
+    def __init__(self, *args, **kwargs):
+        super().__init__()
+
+    def forward(self, youtube_id: str) -> str:
+        #json_file = open('json_config.txt')
+        #json_data = json.load(json_file)
+        json_data = {
+            "youtube_url": f"https://www.youtube.com/watch?v={youtube_id}",
+            "youtube_dl": {
+                "format": "bestaudio/best",
+                #"outtmpl": f"./downloads/{youtube_id} - %(title)s.%(ext)s",
+                "outtmpl": f"./downloads/{youtube_id}.%(ext)s",
+                "postprocessors": [{
+                    "key": "FFmpegExtractAudio",
+                    "preferredcodec": "mp3",
+                    "preferredquality": "192"
+                }],
+                "progress_hooks": [my_hook]
+            }
+        }
+
+        with yt_dlp.YoutubeDL(json_data['youtube_dl']) as ydl:
+
+            print([json_data['youtube_url']])
+            ydl.download([json_data['youtube_url']])
+
+        time.sleep(2)
+        return f"File {youtube_id}.mp3 is downloaded"