Update thaTube_bck.py

thaTube_bck.py

@@ -1,201 +1,3 @@
-
-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
-playlist_url='https://www.youtube.com/playlist?list=PLUwVJKJgdARRdQ3Y3NFB9HMTjXTa5INcz'
-class thaTube2:
-    def __init__(self, playlist_url):
-        self.playlist_url = playlist_url
-        self.current_audio_file = None
-        self.current_audio=None
-        self.video_url = None
-        self.tubelist = playlist_url
-        # 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}")
-
-
-    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:
-    def __init__(self, tubelist):
-        # Autoplay as soon as the object is created
-        self.streamlit_ui()
-        self.playlist_url = tubelist
-        self.current_audio_file = None
-        self.current_audio = None
-        self.is_playing = False
-        self.video_url = 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."""
-        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 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}") 
-
-
-
-'''
 import random
 import tempfile
 import streamlit as st
@@ -203,30 +5,26 @@ from pytube import Playlist, YouTube
 from pydub import AudioSegment
 import pygame
 
-class thaTube:
-    def __init__(self, tubelist='PLUwVJKJgdARRdQ3Y3NFB9HMTjXTa5INcz'):
-        self.playlist_url = tubelist
+class YouTubeAudioPlayer:
+    def __init__(self, playlist_url='PLUwVJKJgdARRdQ3Y3NFB9HMTjXTa5INcz'):
+        self.playlist_url = playlist_url
         self.current_audio_file = None
         self.current_audio = None
         self.is_playing = False
         self.video_url = None
-        self.tubelist = tubelist
-        
+
         pygame.init()
         pygame.mixer.init()
 
-        # Call the method to set up the Streamlit UI
-        self.streamlit_ui()
+        self.create_ui()
 
     def select_random_video(self):
-        """Select a random video from the playlist."""
-        playlist = Playlist(self.tubelist)
+        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.")
 
@@ -244,34 +42,28 @@ class thaTube:
                 return self.current_audio_file
 
     def play_audio(self):
-        """Play the extracted audio."""
         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."""
+    def create_ui(self):
         st.title('YouTube Audio Player')
 
         if st.button('Play Random Video'):
@@ -290,6 +82,4 @@ class thaTube:
 
         if st.button('Next Video'):
             self.next_video()
-            st.write(f"Playing next video: {self.video_url}") 
-
-''' 
+            st.write(f"Playing next video: {self.video_url}")