hh / app.py
aashush's picture
Update app.py
1707ea7
raw
history blame
1.01 kB
pip install pytube
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()