|
import gradio as gr |
|
import yt_dlp |
|
import os |
|
from glob import glob |
|
|
|
def convert_time_to_seconds(hours, minutes, seconds): |
|
return int(hours) * 3600 + int(minutes) * 60 + int(seconds) |
|
|
|
def download_video(video_url, start_hour, start_minute, start_second, end_hour, end_minute, end_second): |
|
start_seconds = convert_time_to_seconds(start_hour, start_minute, start_second) |
|
end_seconds = convert_time_to_seconds(end_hour, end_minute, end_second) |
|
options = { |
|
'format': 'bestaudio/best', |
|
'postprocessors': [{ |
|
'key': 'FFmpegExtractAudio', |
|
'preferredcodec': 'mp3', |
|
'preferredquality': '128', |
|
}], |
|
'outtmpl': '%(title)s.%(ext)s', |
|
'postprocessor_args': [ |
|
'-ss', str(start_seconds), |
|
'-to', str(end_seconds) |
|
] |
|
} |
|
|
|
with yt_dlp.YoutubeDL(options) as ydl: |
|
ydl.download([video_url]) |
|
|
|
download_files = glob('*.mp3') |
|
if download_files: |
|
return f"Download successful: {download_files[0]}", download_files[0] |
|
return "No MP3 file found.", None |
|
|
|
def setup_interface(): |
|
|
|
start_time_hours = gr.Dropdown(list(range(24)), label="Start Hour") |
|
start_time_minutes = gr.Dropdown(list(range(60)), label="Start Minute") |
|
start_time_seconds = gr.Number(value=0, label="Start Seconds", maximum=59) |
|
end_time_hours = gr.Dropdown(list(range(24)), label="End Hour") |
|
end_time_minutes = gr.Dropdown(list(range(60)), label="End Minute") |
|
end_time_seconds = gr.Number(value=0, label="End Seconds", maximum=59) |
|
|
|
interface = gr.Interface( |
|
fn=download_video, |
|
inputs=[ |
|
gr.Textbox(lines=2, placeholder="Enter YouTube video URL here...", label="YouTube Video URL"), |
|
start_time_hours, |
|
start_time_minutes, |
|
start_time_seconds, |
|
end_time_hours, |
|
end_time_minutes, |
|
end_time_seconds |
|
], |
|
outputs=[ |
|
gr.Text(label="Status Message"), |
|
gr.File(label="Downloaded MP3") |
|
], |
|
title="YouTube Video Downloader", |
|
description="Enter YouTube video URL and specify start and end times to download audio as MP3." |
|
) |
|
return interface |
|
|
|
if __name__ == "__main__": |
|
iface = setup_interface() |
|
iface.launch() |
|
|
