Add progress bar

app.py

@@ -10,10 +10,20 @@ def get_binary_file_downloader_html(bin_file, file_label='File'):
     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 main():
     # Set the title and background color
     st.title("YouTube To MP3 Converter 🎥")
     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)
 
@@ -27,7 +37,10 @@ def main():
 
     # Submit button
     if st.button("Submit") and youtube_url:
-        yt = YouTube(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.filter(only_audio=True).first()
 
@@ -47,6 +60,7 @@ def main():
         # 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)
+        progress_bar.progress(100, text='DONE')
         st.success('Successfull!!')
 
 if __name__ == "__main__":