Update thaTube.py

thaTube.py

@@ -1,13 +1,24 @@
 import random
 import tempfile
+import streamlit as st
 from pytube import Playlist, YouTube
 from pydub import AudioSegment
+import pygame
 
-class YouTubeAudioExtractor:
-    def __init__(self, playlist_url):
+class thaTube:
+    def __init__(self, playlist_url, tubelist='PLUwVJKJgdARRdQ3Y3NFB9HMTjXTa5INcz'):
         self.playlist_url = playlist_url
+        self.current_audio_file = None
+        self.current_audio = None
+        self.is_playing = False
         self.video_url = None
-        self.audio_file = None
+        self.tubelist = tubelist
+        
+        pygame.init()
+        pygame.mixer.init()
+
+        # Call the method to set up the Streamlit UI
+        self.streamlit_ui()
 
     def select_random_video(self):
         """Select a random video from the playlist."""
@@ -31,15 +42,55 @@ class YouTubeAudioExtractor:
             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.audio_file = mp3_file.name
-                return self.audio_file
-
-class YouTubeAudioPlayer:
-    def __init__(self, audio_file):
-        self.audio_file = audio_file
+                self.current_audio_file = mp3_file.name
+                return self.current_audio_file
 
     def play_audio(self):
         """Play the extracted audio."""
-        # This method can be expanded to include playback logic.
-        # Currently, it only provides the path to the audio file.
-        return self.audio_file
+        if not self.current_audio_file:
+            raise ValueError("No audio file to play. Please extract audio first.")
+
+        self.current_audio = pygame.mixer.Sound(self.current_audio_file)
+        self.current_audio.play()
+        self.is_playing = True
+
+    def stop_audio(self):
+        """Stop the currently playing audio."""
+        if self.is_playing:
+            pygame.mixer.stop()
+            self.is_playing = False
+
+    def set_volume(self, volume):
+        """Set the volume for playback (volume is a float between 0.0 and 1.0)."""
+        if self.current_audio:
+            self.current_audio.set_volume(volume)
+
+    def next_video(self):
+        """Select a new random video and start playing its audio."""
+        self.stop_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('Stop Audio'):
+            self.stop_audio()
+            st.write("Audio stopped.")
+
+        volume = st.slider('Volume', 0.0, 1.0, 0.5)
+        self.set_volume(volume)
+        st.write(f"Volume set to {volume}")
+
+        if st.button('Next Video'):
+            self.next_video()
+            st.write(f"Playing next video: {self.video_url}")
+