Spaces:
Runtime error
Runtime error
| import os | |
| from scipy.io.wavfile import write | |
| import gradio as gr | |
| import yt_dlp | |
| os.makedirs("ytdl", exist_ok=True) | |
| # applio theme imports | |
| from gradio.themes.Soft import Base | |
| from gradio.themes.utils import colors, fonts, sizes | |
| from typing import Iterable | |
| # Applio Theme | |
| class Applio(Base): | |
| def __init__( | |
| self, | |
| *, | |
| primary_hue: colors.Color | str = colors.neutral, | |
| secondary_hue: colors.Color | str = colors.neutral, | |
| neutral_hue: colors.Color | str = colors.neutral, | |
| spacing_size: sizes.Size | str = sizes.spacing_md, | |
| radius_size: sizes.Size | str = sizes.radius_md, | |
| text_size: sizes.Size | str = sizes.text_lg, | |
| font: fonts.Font | str | Iterable[fonts.Font | str] = ( | |
| "Syne V", | |
| fonts.GoogleFont("Syne"), | |
| "ui-sans-serif", | |
| "system-ui", | |
| ), | |
| font_mono: fonts.Font | str | Iterable[fonts.Font | str] = ( | |
| "ui-monospace", | |
| fonts.GoogleFont("Nunito Sans"), | |
| ), | |
| ): | |
| super().__init__( | |
| primary_hue=primary_hue, | |
| secondary_hue=secondary_hue, | |
| neutral_hue=neutral_hue, | |
| spacing_size=spacing_size, | |
| radius_size=radius_size, | |
| text_size=text_size, | |
| font=font, | |
| font_mono=font_mono, | |
| ) | |
| self.name = ("Applio",) | |
| self.secondary_100 = ("#dbeafe",) | |
| self.secondary_200 = ("#bfdbfe",) | |
| self.secondary_300 = ("#93c5fd",) | |
| self.secondary_400 = ("#60a5fa",) | |
| self.secondary_50 = ("#eff6ff",) | |
| self.secondary_500 = ("#3b82f6",) | |
| self.secondary_600 = ("#2563eb",) | |
| self.secondary_700 = ("#1d4ed8",) | |
| self.secondary_800 = ("#1e40af",) | |
| self.secondary_900 = ("#1e3a8a",) | |
| self.secondary_950 = ("#1d3660",) | |
| super().set( | |
| # Blaise | |
| background_fill_primary="#110F0F", | |
| background_fill_primary_dark="#110F0F", | |
| background_fill_secondary="#110F0F", | |
| background_fill_secondary_dark="#110F0F", | |
| block_background_fill="*neutral_800", | |
| block_background_fill_dark="*neutral_800", | |
| block_border_color="*border_color_primary", | |
| block_border_color_dark="*border_color_primary", | |
| block_border_width="1px", | |
| block_border_width_dark="1px", | |
| block_info_text_color="*body_text_color_subdued", | |
| block_info_text_color_dark="*body_text_color_subdued", | |
| block_info_text_size="*text_sm", | |
| block_info_text_weight="400", | |
| block_label_background_fill="*background_fill_primary", | |
| block_label_background_fill_dark="*background_fill_secondary", | |
| block_label_border_color="*border_color_primary", | |
| block_label_border_color_dark="*border_color_primary", | |
| block_label_border_width="1px", | |
| block_label_border_width_dark="1px", | |
| block_label_margin="0", | |
| block_label_padding="*spacing_sm *spacing_lg", | |
| block_label_radius="calc(*radius_lg - 1px) 0 calc(*radius_lg - 1px) 0", | |
| block_label_right_radius="0 calc(*radius_lg - 1px) 0 calc(*radius_lg - 1px)", | |
| block_label_shadow="*block_shadow", | |
| block_label_text_color="*#110F0F", | |
| block_label_text_color_dark="*#110F0F", | |
| block_label_text_weight="400", | |
| block_padding="*spacing_xl", | |
| block_radius="*radius_md", | |
| block_shadow="none", | |
| block_shadow_dark="none", | |
| block_title_background_fill="rgb(255,255,255)", | |
| block_title_background_fill_dark="rgb(255,255,255)", | |
| block_title_border_color="none", | |
| block_title_border_color_dark="none", | |
| block_title_border_width="0px", | |
| block_title_padding="*block_label_padding", | |
| block_title_radius="*block_label_radius", | |
| block_title_text_color="#110F0F", | |
| block_title_text_color_dark="#110F0F", | |
| block_title_text_size="*text_md", | |
| block_title_text_weight="600", | |
| body_background_fill="#110F0F", | |
| body_background_fill_dark="#110F0F", | |
| body_text_color="white", | |
| body_text_color_dark="white", | |
| body_text_color_subdued="*neutral_400", | |
| body_text_color_subdued_dark="*neutral_400", | |
| body_text_size="*text_md", | |
| body_text_weight="400", | |
| border_color_accent="*neutral_600", | |
| border_color_accent_dark="*neutral_600", | |
| border_color_primary="*neutral_800", | |
| border_color_primary_dark="*neutral_800", | |
| button_border_width="*input_border_width", | |
| button_border_width_dark="*input_border_width", | |
| button_cancel_background_fill="*button_secondary_background_fill", | |
| button_cancel_background_fill_dark="*button_secondary_background_fill", | |
| button_cancel_background_fill_hover="*button_cancel_background_fill", | |
| button_cancel_background_fill_hover_dark="*button_cancel_background_fill", | |
| button_cancel_border_color="*button_secondary_border_color", | |
| button_cancel_border_color_dark="*button_secondary_border_color", | |
| button_cancel_border_color_hover="*button_cancel_border_color", | |
| button_cancel_border_color_hover_dark="*button_cancel_border_color", | |
| button_cancel_text_color="#110F0F", | |
| button_cancel_text_color_dark="#110F0F", | |
| button_cancel_text_color_hover="#110F0F", | |
| button_cancel_text_color_hover_dark="#110F0F", | |
| button_large_padding="*spacing_lg calc(2 * *spacing_lg)", | |
| button_large_radius="*radius_lg", | |
| button_large_text_size="*text_lg", | |
| button_large_text_weight="600", | |
| button_primary_background_fill="*primary_600", | |
| button_primary_background_fill_dark="*primary_600", | |
| button_primary_background_fill_hover="*primary_500", | |
| button_primary_background_fill_hover_dark="*primary_500", | |
| button_primary_border_color="*primary_500", | |
| button_primary_border_color_dark="*primary_500", | |
| button_primary_border_color_hover="*primary_400", | |
| button_primary_border_color_hover_dark="*primary_400", | |
| button_primary_text_color="white", | |
| button_primary_text_color_dark="white", | |
| button_primary_text_color_hover="#110F0F", | |
| button_primary_text_color_hover_dark="#110F0F", | |
| button_secondary_background_fill="transparent", | |
| button_secondary_background_fill_dark="transparent", | |
| button_secondary_background_fill_hover="*neutral_800", | |
| button_secondary_background_fill_hover_dark="*neutral_800", | |
| button_secondary_border_color="*neutral_700", | |
| button_secondary_border_color_dark="*neutral_700", | |
| button_secondary_border_color_hover="*neutral_600", | |
| button_secondary_border_color_hover_dark="*neutral_600", | |
| button_secondary_text_color="white", | |
| button_secondary_text_color_dark="white", | |
| button_secondary_text_color_hover="*button_secondary_text_color", | |
| button_secondary_text_color_hover_dark="*button_secondary_text_color", | |
| button_shadow="none", | |
| button_shadow_active="*shadow_inset", | |
| button_shadow_hover="none", | |
| button_small_padding="*spacing_sm calc(2 * *spacing_sm)", | |
| button_small_radius="*radius_lg", | |
| button_small_text_size="*text_md", | |
| button_small_text_weight="400", | |
| button_transition="0.3s ease all", | |
| checkbox_background_color="*neutral_700", | |
| checkbox_background_color_dark="*neutral_700", | |
| checkbox_background_color_focus="*checkbox_background_color", | |
| checkbox_background_color_focus_dark="*checkbox_background_color", | |
| checkbox_background_color_hover="*checkbox_background_color", | |
| checkbox_background_color_hover_dark="*checkbox_background_color", | |
| checkbox_background_color_selected="*secondary_600", | |
| checkbox_background_color_selected_dark="*secondary_600", | |
| checkbox_border_color="*neutral_700", | |
| checkbox_border_color_dark="*neutral_700", | |
| checkbox_border_color_focus="*secondary_500", | |
| checkbox_border_color_focus_dark="*secondary_500", | |
| checkbox_border_color_hover="*neutral_600", | |
| checkbox_border_color_hover_dark="*neutral_600", | |
| checkbox_border_color_selected="*secondary_600", | |
| checkbox_border_color_selected_dark="*secondary_600", | |
| checkbox_border_radius="*radius_sm", | |
| checkbox_border_width="*input_border_width", | |
| checkbox_border_width_dark="*input_border_width", | |
| checkbox_check="url(\"data:image/svg+xml,%3csvg viewBox='0 0 16 16' fill='white' xmlns='http://www.w3.org/2000/svg'%3e%3cpath d='M12.207 4.793a1 1 0 010 1.414l-5 5a1 1 0 01-1.414 0l-2-2a1 1 0 011.414-1.414L6.5 9.086l4.293-4.293a1 1 0 011.414 0z'/%3e%3c/svg%3e\")", | |
| checkbox_label_background_fill="transparent", | |
| checkbox_label_background_fill_dark="transparent", | |
| checkbox_label_background_fill_hover="transparent", | |
| checkbox_label_background_fill_hover_dark="transparent", | |
| checkbox_label_background_fill_selected="transparent", | |
| checkbox_label_background_fill_selected_dark="transparent", | |
| checkbox_label_border_color="transparent", | |
| checkbox_label_border_color_dark="transparent", | |
| checkbox_label_border_color_hover="transparent", | |
| checkbox_label_border_color_hover_dark="transparent", | |
| checkbox_label_border_width="transparent", | |
| checkbox_label_border_width_dark="transparent", | |
| checkbox_label_gap="*spacing_lg", | |
| checkbox_label_padding="*spacing_md calc(2 * *spacing_md)", | |
| checkbox_label_shadow="none", | |
| checkbox_label_text_color="*body_text_color", | |
| checkbox_label_text_color_dark="*body_text_color", | |
| checkbox_label_text_color_selected="*checkbox_label_text_color", | |
| checkbox_label_text_color_selected_dark="*checkbox_label_text_color", | |
| checkbox_label_text_size="*text_md", | |
| checkbox_label_text_weight="400", | |
| checkbox_shadow="*input_shadow", | |
| color_accent="*primary_500", | |
| color_accent_soft="*primary_50", | |
| color_accent_soft_dark="*neutral_700", | |
| container_radius="*radius_xl", | |
| embed_radius="*radius_lg", | |
| error_background_fill="*background_fill_primary", | |
| error_background_fill_dark="*background_fill_primary", | |
| error_border_color="*border_color_primary", | |
| error_border_color_dark="*border_color_primary", | |
| error_border_width="1px", | |
| error_border_width_dark="1px", | |
| error_text_color="#ef4444", | |
| error_text_color_dark="#ef4444", | |
| form_gap_width="0px", | |
| input_background_fill="*neutral_900", | |
| input_background_fill_dark="*neutral_900", | |
| input_background_fill_focus="*secondary_600", | |
| input_background_fill_focus_dark="*secondary_600", | |
| input_background_fill_hover="*input_background_fill", | |
| input_background_fill_hover_dark="*input_background_fill", | |
| input_border_color="*neutral_700", | |
| input_border_color_dark="*neutral_700", | |
| input_border_color_focus="*secondary_600", | |
| input_border_color_focus_dark="*primary_600", | |
| input_border_color_hover="*input_border_color", | |
| input_border_color_hover_dark="*input_border_color", | |
| input_border_width="1px", | |
| input_border_width_dark="1px", | |
| input_padding="*spacing_xl", | |
| input_placeholder_color="*neutral_500", | |
| input_placeholder_color_dark="*neutral_500", | |
| input_radius="*radius_lg", | |
| input_shadow="none", | |
| input_shadow_dark="none", | |
| input_shadow_focus="*input_shadow", | |
| input_shadow_focus_dark="*input_shadow", | |
| input_text_size="*text_md", | |
| input_text_weight="400", | |
| layout_gap="*spacing_xxl", | |
| link_text_color="*secondary_500", | |
| link_text_color_active="*secondary_500", | |
| link_text_color_active_dark="*secondary_500", | |
| link_text_color_dark="*secondary_500", | |
| link_text_color_hover="*secondary_400", | |
| link_text_color_hover_dark="*secondary_400", | |
| link_text_color_visited="*secondary_600", | |
| link_text_color_visited_dark="*secondary_600", | |
| loader_color="*color_accent", | |
| loader_color_dark="*color_accent", | |
| panel_background_fill="*background_fill_secondary", | |
| panel_background_fill_dark="*background_fill_secondary", | |
| panel_border_color="*border_color_primary", | |
| panel_border_color_dark="*border_color_primary", | |
| panel_border_width="1px", | |
| panel_border_width_dark="1px", | |
| prose_header_text_weight="600", | |
| prose_text_size="*text_md", | |
| prose_text_weight="400", | |
| radio_circle="url(\"data:image/svg+xml,%3csvg viewBox='0 0 16 16' fill='white' xmlns='http://www.w3.org/2000/svg'%3e%3ccircle cx='8' cy='8' r='3'/%3e%3c/svg%3e\")", | |
| section_header_text_size="*text_md", | |
| section_header_text_weight="400", | |
| shadow_drop="rgba(0,0,0,0.05) 0px 1px 2px 0px", | |
| shadow_drop_lg="0 1px 3px 0 rgb(0 0 0 / 0.1), 0 1px 2px -1px rgb(0 0 0 / 0.1)", | |
| shadow_inset="rgba(0,0,0,0.05) 0px 2px 4px 0px inset", | |
| shadow_spread="3px", | |
| shadow_spread_dark="1px", | |
| slider_color="#9E9E9E", | |
| slider_color_dark="#9E9E9E", | |
| stat_background_fill="*primary_500", | |
| stat_background_fill_dark="*primary_500", | |
| table_border_color="*neutral_700", | |
| table_border_color_dark="*neutral_700", | |
| table_even_background_fill="*neutral_950", | |
| table_even_background_fill_dark="*neutral_950", | |
| table_odd_background_fill="*neutral_900", | |
| table_odd_background_fill_dark="*neutral_900", | |
| table_radius="*radius_lg", | |
| table_row_focus="*color_accent_soft", | |
| table_row_focus_dark="*color_accent_soft", | |
| ) | |
| applio = Applio() | |
| def separator(audio): | |
| os.makedirs("output", exist_ok=True) | |
| write('test.wav', audio[0], audio[1]) | |
| result = os.system("audio-separator test.wav --model_filename model_bs_roformer_ep_317_sdr_12.9755.ckpt --output_dir=./output --output_format=wav") | |
| files = [ | |
| "./output/test_(Instrumental)_model_bs_roformer_ep_317_sdr_12.wav", | |
| "./output/test_(Vocals)_model_bs_roformer_ep_317_sdr_12.wav" | |
| ] | |
| for file in files: | |
| if not os.path.isfile(file): | |
| print(f"File not found: {file}") | |
| else: | |
| print(f"File exists: {file}") | |
| return files | |
| def download_audio(url): | |
| ydl_opts = { | |
| 'format': 'bestaudio/best', | |
| 'outtmpl': 'ytdl/%(title)s.%(ext)s', | |
| 'postprocessors': [{ | |
| 'key': 'FFmpegExtractAudio', | |
| 'preferredcodec': 'mp3', | |
| 'preferredquality': '192', | |
| }], | |
| } | |
| with yt_dlp.YoutubeDL(ydl_opts) as ydl: | |
| info_dict = ydl.extract_info(url, download=True) | |
| file_path = ydl.prepare_filename(info_dict).rsplit('.', 1)[0] + '.mp3' | |
| return file_path | |
| def download_and_show(url): | |
| file_path = download_audio(url) | |
| return file_path | |
| with gr.Blocks(theme=applio) as demo: | |
| gr.Markdown(" BS FORMER ESPARATION!") | |
| with gr.Tab("main"): | |
| audio_input = gr.Audio(type="numpy", label="Input") | |
| output1 = gr.Audio(type="filepath", label="Stem 1", interactive=False) | |
| output2 = gr.Audio(type="filepath", label="Stem 2", interactive=False) | |
| submit_button = gr.Button("Separate") | |
| submit_button.click( | |
| separator, | |
| inputs=[audio_input], | |
| outputs=[output1, output2] | |
| ) | |
| gr.Markdown("Download YouTube audio for separation") | |
| url = gr.Textbox(label="URL to YouTube") | |
| download = gr.Button("Download") | |
| download.click( | |
| download_and_show, | |
| inputs=[url], | |
| outputs=[audio_input] | |
| ) | |
| demo.launch() | |