Add code to download youtube video

.devcontainer/devcontainer.json

@@ -0,0 +1,22 @@
+// For format details, see https://aka.ms/devcontainer.json. For config options, see the
+// README at: https://github.com/devcontainers/templates/tree/main/src/python
+{
+	"name": "Python 3",
+	// Or use a Dockerfile or Docker Compose file. More info: https://containers.dev/guide/dockerfile
+	"image": "mcr.microsoft.com/devcontainers/python:1-3.9-bookworm"
+
+	// Features to add to the dev container. More info: https://containers.dev/features.
+	// "features": {},
+
+	// Use 'forwardPorts' to make a list of ports inside the container available locally.
+	// "forwardPorts": [],
+
+	// Use 'postCreateCommand' to run commands after the container is created.
+	// "postCreateCommand": "pip3 install --user -r requirements.txt",
+
+	// Configure tool-specific properties.
+	// "customizations": {},
+
+	// Uncomment to connect as root instead. More info: https://aka.ms/dev-containers-non-root.
+	// "remoteUser": "root"
+}

README.md

@@ -1,8 +1,8 @@
 ---
 title: Youtube Video Downloader
-emoji: 👁
-colorFrom: indigo
-colorTo: indigo
+emoji: 📊
+colorFrom: pink
+colorTo: red
 sdk: streamlit
 sdk_version: 1.26.0
 app_file: app.py

app.py

@@ -0,0 +1,85 @@
+import streamlit as st
+from pytube import YouTube
+import os
+import base64
+import time
+
+#@deprecated: not used anymore due to streamlit download_button feature
+def get_binary_file_downloader_html(bin_file, file_label='File'):
+    with open(bin_file, 'rb') as f:
+        data = f.read()
+    bin_str = base64.b64encode(data).decode()
+    href = f'<a href="data:application/octet-stream;base64,{bin_str}" download="{os.path.basename(bin_file)}">Download {file_label}</a>'
+    return href
+
+def on_progress(stream, chunk, bytes_remaining):
+    print(' progress:', bytes_remaining, "           ", end='\r', flush=True)
+
+def on_complete(stream, filename):
+    print('--- Completed ---')
+    print('stream:', stream)
+    print('filename:', filename)
+
+def delete_file(filepath):
+    if os.path.exists(filepath):
+        os.remove(filepath)
+
+def on_download_file_click(filepath):
+    # sleep Xsec can delete file
+    time.sleep(2)
+    delete_file(filepath)
+
+def main():
+    # Set the title and background color
+    st.title("YouTube video downloader 🎥")
+    st.markdown('<style>h1{color: orange; text-align: center;}</style>', unsafe_allow_html=True)
+    
+    # st.subheader('Built with the Llama 2 🦙, Haystack, Streamlit and ❤️')
+    # st.markdown('<style>h3{color: pink;  text-align: center;}</style>', unsafe_allow_html=True)
+
+    # Expander for app details
+    with st.expander("About the App (Works well on chrome browser)"):
+        st.write("This app allows you to download video from  YouTube  url.")
+        st.write("Enter a YouTube URL in the input box below and click on 'Submit'")
+
+    # Input box for YouTube URL
+    youtube_url = st.text_input("Enter YouTube URL")
+
+    # Submit button
+    if st.button("Submit") and youtube_url:
+        progress_bar = st.progress(0, text='Processing...')
+        yt = YouTube(youtube_url)
+        yt.register_on_progress_callback(on_progress)
+        yt.register_on_complete_callback(on_complete)
+        # extract only audio
+        video = yt.streams.first()
+        # video = yt.streams.get_highest_resolution()
+
+        initial_filename = yt.title
+        # download the file
+        destination = '.' # current folder
+        out_file = video.download(output_path=destination, filename=str(initial_filename))
+
+        # save the file
+        new_file = initial_filename
+        # base, ext = os.path.splitext(out_file)
+        # new_file = '-'.join(initial_filename.split(' '))[-15:] + '.mp3' # base[0:10] + '.mp3'
+        # os.rename(out_file, new_file)
+
+        # Display layout with 2 rows
+        st.video(youtube_url)
+        # st.header("Audio file in MP3 format")
+        # st.write(new_file)
+        # st.markdown(get_binary_file_downloader_html('photo.jpg', 'Picture'), unsafe_allow_html=True)
+        # st.markdown(get_binary_file_downloader_html('data.csv', 'My Data'), unsafe_allow_html=True)
+        # st.markdown(get_binary_file_downloader_html(new_file, 'Audio'), unsafe_allow_html=True)
+
+        # download video file
+        with open(new_file, 'rb') as f:
+            st.download_button('Download Video', f, file_name=new_file, on_click=on_download_file_click(new_file))
+                
+        progress_bar.progress(100, text='DONE')
+        st.success('Successfull!!')
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -0,0 +1,2 @@
+streamlit
+pytube