MultiTabImageGenerator / thaTube_bck.py
K00B404's picture
Rename thaTube.py to thaTube_bck.py
b737674 verified
raw
history blame
10.4 kB
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
from pytube import Playlist, YouTube
from pydub import AudioSegment
import pygame
class thaTube:
def __init__(self, tubelist='PLUwVJKJgdARRdQ3Y3NFB9HMTjXTa5INcz'):
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.tubelist)
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."""
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}")
'''