import gradio as gr 
import yt_dlp
import os 

def dl(inp):
    out = None
    file_out=[]
    try:
        inp_out=inp.replace("https://","")
        inp_out=inp_out.replace("/","_").replace(".","_")
        os.system(f'yt-dlp "{inp}" --trim-filenames 100 --write-description -o "{inp_out}.description"')  

        os.system(f'yt-dlp "{inp}" --trim-filenames 100 -o "{inp_out}.mp4"')  
        out = f"{inp_out}.mp4"
        out_file.append(out)
        out_file.append("{inp_out}.description")
        
        '''
        with open(f"{inp_out}.description", "rb") as f:
            f_out = f.readlines()
            print (str(f_out, 'UTF-8'))
        '''
    except Exception as e:
        print (e)
        out = None
    return out,file_out


with gr.Blocks() as app:
    inp_url = gr.Textbox()
    go_btn = gr.Button()
    with gr.Row():
        outp_vid=gr.Video()
        outp_files=gr.Files()
    go_btn.click(dl,inp_url,[outp_vid,outp_files])

app.launch()