app.py

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

cachedir = ".cache"


def clip(yt_url, start_timestamp, end_timestamp):
    start_timestamp = int(start_timestamp)
    duration = int(end_timestamp) - int(start_timestamp)
    
    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([yt_url])
            except Exception as e:
                print(e)
                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([yt_url])
        clip_path = f"{uuid4()}.mp3"
        print(f'ffmpeg -ss {start_timestamp} -t {duration} -i "{video_output_path}" "{clip_path}"')
        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)")
end_timestamp = gr.Textbox(label="Clip end timestamp (in seconds)")

interface = gr.Interface(clip, [url_link, start_timestamp, end_timestamp], "audio", title="YouTube Clipper", description="After downloading the clips, please go to https://demo.deepsync.co/11-voice-cloning to clone the voice.")

if __name__=="__main__":
    interface.queue().launch(auth=(os.environ.get("GRADIO_USERNAME"), os.environ.get("GRADIO_PASSWORD")))