Update youtube_video.py

youtube_video.py

@@ -1,9 +1,10 @@
 import pytube
 import os
 from pytube import YouTube
+from pydub import AudioSegment
 
-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."""
+def download_and_compress_youtube_audio(youtube_url):
+    """Downloads the audio from a YouTube video, compresses it, renames it to the first twelve characters, and returns the downloaded file path."""
 
     try:
         # Create a YouTube object
@@ -20,13 +21,19 @@ def download_youtube_audio(youtube_url):
         original_filename = audio.default_filename
 
         # Extract the first twelve characters and change the file extension to .mp3
-        new_filename = original_filename[:12] + '.mp3'
+        compressed_filename = original_filename[:12] + '_compressed.mp3'
 
-        # Rename the downloaded file
-        os.rename(original_filename, new_filename)
+        # Load the downloaded audio file
+        audio_segment = AudioSegment.from_file(original_filename)
 
-        print("Download complete! Audio saved to:", new_filename)
-        return new_filename
+        # Compress the audio file (e.g., by reducing the bitrate)
+        audio_segment.export(compressed_filename, format="mp3", bitrate="64k")
+
+        # Remove the original downloaded file
+        os.remove(original_filename)
+
+        print("Download and compression complete! Audio saved to:", compressed_filename)
+        return compressed_filename
 
     except Exception as e:
         print("An error occurred:", e)
@@ -34,4 +41,4 @@ def download_youtube_audio(youtube_url):
 
 # Example usage
 youtube_url = "https://www.youtube.com/watch?v=your_video_id"
-download_youtube_audio(youtube_url)
+download_and_compress_youtube_audio(youtube_url)