Spaces:
Runtime error
Runtime error
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 | |
) | |
) | |