Upload 6 files

app.py

@@ -0,0 +1,32 @@
+import streamlit as st
+import tube as tb
+
+tb.clear_cache()
+
+
+md = '''
+# YouTube Downloader
+'''
+
+st.markdown(md)
+
+
+url = st.text_input(
+    placeholder="https://www.youtube.com/",
+    label='**Enter the url of the youtube:**',
+    key='title'
+)
+
+
+
+if url is not None and ('https' in url or 'http' in url):
+    tb.download_yt(url)
+
+
+
+
+
+
+
+
+

requirements.txt

@@ -0,0 +1,3 @@
+streamlit
+pytube==12.1.2
+

tube/__init__.py

@@ -0,0 +1,3 @@
+from .download import download_yt
+from .utils import clear_cache
+from .var import OUTPUT_DIR

tube/download.py

@@ -0,0 +1,48 @@
+from pathlib import Path
+from pytube import YouTube
+
+import streamlit as st
+from .utils import  clear_cache
+
+
+
+
+def download_yt(yt_url:str, output_dir:str = './downloads'):
+    yt = YouTube(yt_url)
+
+    prompt = st.markdown(f'''`downloading...`''')
+
+    while True:
+        try:
+            yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first().download(
+                output_path=output_dir,
+                filename='download.mp4'
+            )
+            prompt.empty()
+            break
+        except Exception as e:
+            print(e)
+
+    download_file(folder_name= output_dir)
+
+
+
+
+
+def download_file(folder_name):
+    def tmp(*,folder_name:str):
+        st.session_state["title"] = ""
+        clear_cache(folder_name)
+
+
+    with open(Path('downloads').joinpath('download.mp4'), "rb") as file:
+        btn = st.download_button(
+            label="Download",
+            data=file,
+            file_name='download.mp4',
+            on_click= tmp,kwargs=dict(
+                folder_name = folder_name
+            )
+        )
+
+

tube/utils.py

@@ -0,0 +1,36 @@
+import shutil
+import streamlit as st
+from pathlib import Path
+from .var import OUTPUT_DIR
+
+
+
+
+def compress_folder_2_zip(output_filename: str, dir_name:str):
+    path = Path(output_filename+'.zip')
+    if path.exists():
+        return
+
+    prompt = st.info('Start compressing...')
+    with st.spinner("Compressing"):
+        shutil.make_archive(output_filename.replace('.zip', ''), 'zip', dir_name)
+    prompt.empty()
+
+
+def remove_dir_rec(pth):
+    pth = Path(pth)
+    if pth.exists():
+        for child in pth.glob('*'):
+            if child.is_file():
+                child.unlink()
+            else:
+                remove_dir_rec(child)
+        pth.rmdir()
+def clear_cache(dir_name:str = OUTPUT_DIR):
+    remove_dir_rec(dir_name)
+
+
+
+
+if __name__ == '__main__':
+    compress_folder_2_zip('test',dir_name='../downloads')

tube/var.py

@@ -0,0 +1,4 @@
+from pathlib import Path
+
+PROJECT_ROOT = Path(__package__).absolute().parent
+OUTPUT_DIR = Path(PROJECT_ROOT).joinpath('downloads')