Spaces:
Runtime error
Runtime error
File size: 1,080 Bytes
9c676f9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
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
)
)
|