from pytube import YouTube import ffmpeg from gradio_client import Client import subprocess import gradio as gr client = Client("abidlabs/Acapellify") placeholder_url = "https://www.youtube.com/watch?v=bfATETsvqt4&ab_channel=MiniMuslims" placeholder_file = "example_acapella_video.mp4" def acapellify(url): if url == placeholder_url: return placeholder_file yt = YouTube(url) try: video = yt.streams.get_highest_resolution() video.download(max_retries=5) except: raise gr.Error("YouTube video could not be downloaded. Try clicking Submit again and make sure that the URL is correct.") video_path = video.get_file_path() ( ffmpeg .input(video_path) .audio .output('audio.m4a', format='mp4') .overwrite_output() .run() ) result = client.predict( 'audio.m4a', api_name="/predict" ) subprocess.call(['ffmpeg', '-y', '-i', video_path, '-i', result[0], '-map', '0:v', '-map', '1:a', '-c:v', 'copy', '-c:a', 'aac', '-strict', 'experimental', 'acapella_video.mp4']) return "acapella_video.mp4" io = gr.Interface(acapellify, gr.Textbox(label="URL to YouTube video", value=placeholder_url), gr.Video(label="Acapella output")) with gr.Blocks() as demo: gr.Markdown("# Acapellify: Get a Vocals-only from any YouTube Video in 1 Click") with gr.Row(): with gr.Column(): gr.Markdown("**Example Original Video**") gr.Video("old_video.mp4") with gr.Column(): gr.Markdown("**Example Acapella Video**") gr.Video("acapella_video.mp4") gr.Markdown("**Try it yourself:** (for now, I recommend short videos <3 minutes)") io.render() demo.launch()