import os
import yt_dlp
import gradio as gr
import subprocess
from uuid import uuid4
from tempfile import TemporaryDirectory

cachedir = ".cache"


def clip(yt_url, start_timestamp, duration):
    start_timestamp = round(float(start_timestamp), 3)
    duration = round(float(duration), 3)
    
    with TemporaryDirectory() as tmpdir:
        video_output_path = os.path.join(tmpdir, "video.mp4")
        ydl_opts = {
            "quiet": True,
            "outtmpl": video_output_path,
            "postprocessors": [{"key": "FFmpegVideoConvertor", "preferedformat": "mp4"}],
            "cachedir": cachedir
        }
        if not os.path.exists(video_output_path):
            try:
                ydl_opts["format"] = "22"
                with yt_dlp.YoutubeDL(ydl_opts) as ydl:
                    ydl.download([video_url])
            except:
                ydl_opts["format"] = "bestvideo[height<=480][ext=mp4]+bestaudio[ext=m4a]/best[height<=480[ext=mp4]/best"
                sleep(5)
                with yt_dlp.YoutubeDL(ydl_opts) as ydl:
                    ydl.download([video_url])
        clip_path = os.path.join(tmpdir, f"{uuid4()}.mp3")
        subprocess.run(f'ffmpeg -ss {start_timestamp} -t {duration} -i "{video_output_path}" "{clip_path}"', shell=True)
    return clip_path


url_link = gr.Textbox(label="YouTube URL")
start_timestamp = gr.Textbox(label="Clip start timestamp (in seconds)")
clip_length = gr.Textbox(label="Clip length (in seconds)")

interface = gr.Interface(clip, [url_link, start_timestamp, clip_length], "audio", title="YouTube Clipper")

if __name__=="__main__":
    interface.queue().launch()