Update thaTube.py

thaTube.py

@@ -1,12 +1,16 @@
 
+import random
+import tempfile
+import streamlit as st
+from pytube import Playlist, YouTube
+from pydub import AudioSegment
 import random
 import tempfile
 import streamlit as st
 from pytube import Playlist, YouTube
 from pydub import AudioSegment
 
-import pygame
-class thaTube2:
+class thaTube:
     def __init__(self, playlist_url):
         self.playlist_url = playlist_url
         self.current_audio_file = None
@@ -68,9 +72,38 @@ class thaTube2:
             self.next_video()
             st.write(f"Playing next video: {self.video_url}")
 
+
+    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 thaTube:
+class thaTube2:
     def __init__(self, tubelist='https://www.youtube.com/playlist?list=PLUwVJKJgdARRdQ3Y3NFB9HMTjXTa5INcz'):
                 # Autoplay as soon as the object is created
         self.streamlit_ui()