File size: 1,848 Bytes
69c11ea 98b6fef 45b4ac0 98b6fef b37dc17 45b4ac0 98b6fef b37dc17 98b6fef 1a92ad7 98b6fef 1a92ad7 251705c 98b6fef |
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 50 |
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, 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([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 = os.path.join(tmpdir, 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)
print(os.listdir(tmpdir))
sleep(3)
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() |