Update youtube_video.py

youtube_video.py

@@ -1,34 +1,37 @@
-import pytube
-import os
-from pytube import YouTube
-
-
-def download_youtube_video(youtube_url):
-    """Downloads a YouTube video, renames it to the first three characters, and returns the downloaded file path."""
-
-    try:
-        # Create a YouTube object
-        yt = pytube.YouTube(youtube_url)
-
-        # Get the highest resolution progressive stream
-        video = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first()
-
-        # Download the video
-        print("Downloading...")
-        video.download()
-
-        # Get the original filename
-        original_filename = video.default_filename
-
-        # Extract the first three characters and keep the file extension
-        new_filename = original_filename[:12] + os.path.splitext(original_filename)[1]
-
-        # Rename the downloaded file
-        os.rename(original_filename, new_filename)
-
-        print("Download complete! Video saved to:", new_filename)
-        return new_filename
-
-    except Exception as e:
-        print("An error occurred:", e)
-        return None
+import pytube
+import os
+from pytube import YouTube
+
+def download_youtube_audio(youtube_url):
+    """Downloads the audio from a YouTube video, renames it to the first twelve characters, and returns the downloaded file path."""
+
+    try:
+        # Create a YouTube object
+        yt = YouTube(youtube_url)
+
+        # Get the audio stream
+        audio = yt.streams.filter(only_audio=True).first()
+
+        # Download the audio
+        print("Downloading...")
+        audio.download()
+
+        # Get the original filename
+        original_filename = audio.default_filename
+
+        # Extract the first twelve characters and change the file extension to .mp3
+        new_filename = original_filename[:12] + '.mp3'
+
+        # Rename the downloaded file
+        os.rename(original_filename, new_filename)
+
+        print("Download complete! Audio saved to:", new_filename)
+        return new_filename
+
+    except Exception as e:
+        print("An error occurred:", e)
+        return None
+
+# Example usage
+youtube_url = "https://www.youtube.com/watch?v=your_video_id"
+download_youtube_audio(youtube_url)