|
import yt_dlp |
|
import gradio as gr |
|
import os |
|
from glob import glob |
|
|
|
def convert_time_to_seconds(time_str): |
|
hours, minutes, seconds = map(int, time_str.split(':')) |
|
return hours * 3600 + minutes * 60 + seconds |
|
|
|
def download_video(video_url, start_time, end_time): |
|
start_seconds = convert_time_to_seconds(start_time) |
|
end_seconds = convert_time_to_seconds(end_time) |
|
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(): |
|
interface = gr.Interface( |
|
fn=download_video, |
|
inputs=[ |
|
gr.Textbox(lines=2, placeholder="Enter YouTube video URL here...", label="YouTube Video URL"), |
|
gr.Textbox(placeholder="Start Time (e.g., 00:00:00)", label="Start Time (HH:MM:SS)"), |
|
gr.Textbox(placeholder="End Time (e.g., 00:05:30)", label="End Time (HH:MM:SS)") |
|
], |
|
outputs=[ |
|
gr.Text(label="Status Message"), |
|
gr.File(label="Download 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() |
