Upload 2 files

main.py

@@ -0,0 +1,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()