Update thaTube.py

thaTube.py

@@ -4,9 +4,73 @@ import tempfile
 import streamlit as st
 from pytube import Playlist, YouTube
 from pydub import AudioSegment
-import pygame
 
+import pygame
 class thaTube:
+    def __init__(self, playlist_url):
+        self.playlist_url = playlist_url
+        self.current_audio_file = None
+        self.video_url = None
+
+        # Autoplay as soon as the object is created
+        self.streamlit_ui()
+
+    def select_random_video(self):
+        """Select a random video from the playlist."""
+        playlist = Playlist(self.playlist_url)
+        videos = list(playlist.video_urls)
+        self.video_url = random.choice(videos)
+        return self.video_url
+
+    def extract_audio(self):
+        """Extract audio from the selected video and save it as an MP3 file."""
+        if not self.video_url:
+            raise ValueError("No video selected. Please select a video first.")
+
+        video = YouTube(self.video_url)
+        audio_stream = video.streams.filter(only_audio=True).first()
+
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as temp_audio_file:
+            audio_stream.download(output_path=temp_audio_file.name)
+            temp_audio_file.close()
+
+            audio = AudioSegment.from_file(temp_audio_file.name)
+            with tempfile.NamedTemporaryFile(delete=False, suffix='.mp3') as mp3_file:
+                audio.export(mp3_file.name, format="mp3")
+                self.current_audio_file = mp3_file.name
+                return self.current_audio_file
+
+    def play_audio(self):
+        """Play the extracted audio using Streamlit's audio player."""
+        if not self.current_audio_file:
+            raise ValueError("No audio file to play. Please extract audio first.")
+        
+        audio_bytes = open(self.current_audio_file, 'rb').read()
+        st.audio(audio_bytes, format='audio/mpeg', start_time=0)
+
+    def next_video(self):
+        """Select a new random video and start playing its audio."""
+        self.select_random_video()
+        self.extract_audio()
+        self.play_audio()
+
+    def streamlit_ui(self):
+        """Create a Streamlit interface for controlling playback."""
+        st.title('YouTube Audio Player')
+
+        if st.button('Play Random Video'):
+            self.select_random_video()
+            self.extract_audio()
+            self.play_audio()
+            st.write(f"Playing audio from: {self.video_url}")
+
+        if st.button('Next Video'):
+            self.next_video()
+            st.write(f"Playing next video: {self.video_url}")
+
+# Usage
+#tuber = thaTube('https://www.youtube.com/playlist?list=PLUwVJKJgdARRdQ3Y3NFB9HMTjXTa5INcz')
+class thaTube2:
     def __init__(self, tubelist='PLUwVJKJgdARRdQ3Y3NFB9HMTjXTa5INcz'):
         self.playlist_url = tubelist
         self.current_audio_file = None