File size: 992 Bytes
eb6d386 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
import streamlit as st
import pandas as pd
from pytube import YouTube
import base64
from io import BytesIO
def main():
path = st.text_input('Enter URL of any youtube video')
option = st.selectbox(
'Select type of download',
('audio', 'highest_resolution', 'lowest_resolution'))
matches = ['audio', 'highest_resolution', 'lowest_resolution']
if st.button("download"):
video_object = YouTube(path)
st.write("Title of Video: " + str(video_object.title))
st.write("Number of Views: " + str(video_object.views))
if option == 'audio':
video_object.streams.get_audio_only().download() # base64.b64encode("if file is too large").decode()
elif option == 'highest_resolution':
video_object.streams.get_highest_resolution().download()
elif option == 'lowest_resolution':
video_object.streams.get_lowest_resolution().download()
if st.button("view"):
st.video(path)
main() |