import gradio as gr from main.tools import huggingface from main.configs.config import Config from main.app.based.utils import * def utils_tabs(): with gr.TabItem("utils"): with gr.Tabs(): with gr.TabItem(translations["audio_editing"], visible=False): gr.Markdown(translations["audio_editing_info"]) with gr.Row(): gr.Markdown(translations["audio_editing_markdown"]) with gr.Row(): with gr.Column(): with gr.Group(): with gr.Row(): save_compute = gr.Checkbox(label=translations["save_compute"], value=True, interactive=True) tar_prompt = gr.Textbox(label=translations["target_prompt"], info=translations["target_prompt_info"], placeholder="Piano and violin cover", lines=5, interactive=True) with gr.Column(): cfg_scale_src = gr.Slider(value=3, minimum=0.5, maximum=25, label=translations["cfg_scale_src"], info=translations["cfg_scale_src_info"], interactive=True) cfg_scale_tar = gr.Slider(value=12, minimum=0.5, maximum=25, label=translations["cfg_scale_tar"], info=translations["cfg_scale_tar_info"], interactive=True) with gr.Row(): edit_button = gr.Button(translations["editing"], variant="primary") with gr.Row(): with gr.Column(): drop_audio_file = gr.File(label=translations["drop_audio"], file_types=[".wav", ".mp3", ".flac", ".ogg", ".opus", ".m4a", ".mp4", ".aac", ".alac", ".wma", ".aiff", ".webm", ".ac3"]) display_audio = gr.Audio(show_download_button=True, interactive=False, label=translations["input_audio"]) with gr.Column(): with gr.Accordion(translations["input_output"], open=False): with gr.Column(): export_audio_format = gr.Radio(label=translations["export_format"], info=translations["export_info"], choices=["wav", "mp3", "flac", "ogg", "opus", "m4a", "mp4", "aac", "alac", "wma", "aiff", "webm", "ac3"], value="wav", interactive=True) input_audiopath = gr.Dropdown(label=translations["audio_path"], value="", choices=paths_for_files, info=translations["provide_audio"], allow_custom_value=True, interactive=True) output_audiopath = gr.Textbox(label=translations["output_path"], value="audios/output.wav", placeholder="audios/output.wav", info=translations["output_path_info"], interactive=True) with gr.Column(): refesh_audio = gr.Button(translations["refesh"]) with gr.Accordion(translations["setting"], open=False): audioldm2_model = gr.Radio(label=translations["audioldm2_model"], info=translations["audioldm2_model_info"], choices=["audioldm2", "audioldm2-large", "audioldm2-music"], value="audioldm2-music", interactive=True) with gr.Row(): src_prompt = gr.Textbox(label=translations["source_prompt"], lines=2, interactive=True, info=translations["source_prompt_info"], placeholder="A recording of a happy upbeat classical music piece") with gr.Row(): with gr.Column(): audioldm2_sample_rate = gr.Slider(minimum=8000, maximum=96000, label=translations["sr"], info=translations["sr_info"], value=44100, step=1, interactive=True) t_start = gr.Slider(minimum=15, maximum=85, value=45, step=1, label=translations["t_start"], interactive=True, info=translations["t_start_info"]) steps = gr.Slider(value=50, step=1, minimum=10, maximum=300, label=translations["steps_label"], info=translations["steps_info"], interactive=True) with gr.Row(): gr.Markdown(translations["output_audio"]) with gr.Row(): output_audioldm2 = gr.Audio(show_download_button=True, interactive=False, label=translations["output_audio"]) with gr.Row(): refesh_audio.click(fn=change_audios_choices, inputs=[input_audiopath], outputs=[input_audiopath]) drop_audio_file.upload(fn=lambda audio_in: shutil.move(audio_in.name, os.path.join("audios")), inputs=[drop_audio_file], outputs=[input_audiopath]) input_audiopath.change(fn=lambda audio: audio if os.path.isfile(audio) else None, inputs=[input_audiopath], outputs=[display_audio]) with gr.Row(): edit_button.click( fn=run_audioldm2, inputs=[ input_audiopath, output_audiopath, export_audio_format, audioldm2_sample_rate, audioldm2_model, src_prompt, tar_prompt, steps, cfg_scale_src, cfg_scale_tar, t_start, save_compute ], outputs=[output_audioldm2], api_name="audioldm2" ) with gr.TabItem(translations["audio_effects"], visible=configs.get("effects_tab", True)): gr.Markdown(translations["apply_audio_effects"]) with gr.Row(): gr.Markdown(translations["audio_effects_edit"]) with gr.Row(): with gr.Column(): with gr.Row(): reverb_check_box = gr.Checkbox(label=translations["reverb"], value=False, interactive=True) chorus_check_box = gr.Checkbox(label=translations["chorus"], value=False, interactive=True) delay_check_box = gr.Checkbox(label=translations["delay"], value=False, interactive=True) phaser_check_box = gr.Checkbox(label=translations["phaser"], value=False, interactive=True) compressor_check_box = gr.Checkbox(label=translations["compressor"], value=False, interactive=True) more_options = gr.Checkbox(label=translations["more_option"], value=False, interactive=True) with gr.Row(): with gr.Accordion(translations["input_output"], open=False): with gr.Row(): upload_audio = gr.File(label=translations["drop_audio"], file_types=[".wav", ".mp3", ".flac", ".ogg", ".opus", ".m4a", ".mp4", ".aac", ".alac", ".wma", ".aiff", ".webm", ".ac3"]) with gr.Row(): audio_in_path = gr.Dropdown(label=translations["input_audio"], value="", choices=paths_for_files, info=translations["provide_audio"], interactive=True, allow_custom_value=True) audio_out_path = gr.Textbox(label=translations["output_audio"], value="audios/audio_effects.wav", placeholder="audios/audio_effects.wav", info=translations["provide_output"], interactive=True) with gr.Row(): with gr.Column(): audio_combination = gr.Checkbox(label=translations["merge_instruments"], value=False, interactive=True) audio_combination_input = gr.Dropdown(label=translations["input_audio"], value="", choices=paths_for_files, info=translations["provide_audio"], interactive=True, allow_custom_value=True, visible=audio_combination.value) with gr.Row(): audio_effects_refesh = gr.Button(translations["refesh"]) with gr.Row(): audio_output_format = gr.Radio(label=translations["export_format"], info=translations["export_info"], choices=["wav", "mp3", "flac", "ogg", "opus", "m4a", "mp4", "aac", "alac", "wma", "aiff", "webm", "ac3"], value="wav", interactive=True) with gr.Row(): apply_effects_button = gr.Button(translations["apply"], variant="primary", scale=2) with gr.Row(): with gr.Column(): with gr.Row(): with gr.Accordion(translations["reverb"], open=False, visible=reverb_check_box.value) as reverb_accordion: reverb_freeze_mode = gr.Checkbox(label=translations["reverb_freeze"], info=translations["reverb_freeze_info"], value=False, interactive=True) reverb_room_size = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.15, label=translations["room_size"], info=translations["room_size_info"], interactive=True) reverb_damping = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.7, label=translations["damping"], info=translations["damping_info"], interactive=True) reverb_wet_level = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.2, label=translations["wet_level"], info=translations["wet_level_info"], interactive=True) reverb_dry_level = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.8, label=translations["dry_level"], info=translations["dry_level_info"], interactive=True) reverb_width = gr.Slider(minimum=0, maximum=1, step=0.01, value=1, label=translations["width"], info=translations["width_info"], interactive=True) with gr.Row(): with gr.Accordion(translations["chorus"], open=False, visible=chorus_check_box.value) as chorus_accordion: chorus_depth = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.5, label=translations["chorus_depth"], info=translations["chorus_depth_info"], interactive=True) chorus_rate_hz = gr.Slider(minimum=0.1, maximum=10, step=0.1, value=1.5, label=translations["chorus_rate_hz"], info=translations["chorus_rate_hz_info"], interactive=True) chorus_mix = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.5, label=translations["chorus_mix"], info=translations["chorus_mix_info"], interactive=True) chorus_centre_delay_ms = gr.Slider(minimum=0, maximum=50, step=1, value=10, label=translations["chorus_centre_delay_ms"], info=translations["chorus_centre_delay_ms_info"], interactive=True) chorus_feedback = gr.Slider(minimum=-1, maximum=1, step=0.01, value=0, label=translations["chorus_feedback"], info=translations["chorus_feedback_info"], interactive=True) with gr.Row(): with gr.Accordion(translations["delay"], open=False, visible=delay_check_box.value) as delay_accordion: delay_second = gr.Slider(minimum=0, maximum=5, step=0.01, value=0.5, label=translations["delay_seconds"], info=translations["delay_seconds_info"], interactive=True) delay_feedback = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.5, label=translations["delay_feedback"], info=translations["delay_feedback_info"], interactive=True) delay_mix = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.5, label=translations["delay_mix"], info=translations["delay_mix_info"], interactive=True) with gr.Column(): with gr.Row(): with gr.Accordion(translations["more_option"], open=False, visible=more_options.value) as more_accordion: with gr.Row(): fade = gr.Checkbox(label=translations["fade"], value=False, interactive=True) bass_or_treble = gr.Checkbox(label=translations["bass_or_treble"], value=False, interactive=True) limiter = gr.Checkbox(label=translations["limiter"], value=False, interactive=True) resample_checkbox = gr.Checkbox(label=translations["resample"], value=False, interactive=True) with gr.Row(): distortion_checkbox = gr.Checkbox(label=translations["distortion"], value=False, interactive=True) gain_checkbox = gr.Checkbox(label=translations["gain"], value=False, interactive=True) bitcrush_checkbox = gr.Checkbox(label=translations["bitcrush"], value=False, interactive=True) clipping_checkbox = gr.Checkbox(label=translations["clipping"], value=False, interactive=True) with gr.Accordion(translations["fade"], open=True, visible=fade.value) as fade_accordion: with gr.Row(): fade_in = gr.Slider(minimum=0, maximum=10000, step=100, value=0, label=translations["fade_in"], info=translations["fade_in_info"], interactive=True) fade_out = gr.Slider(minimum=0, maximum=10000, step=100, value=0, label=translations["fade_out"], info=translations["fade_out_info"], interactive=True) with gr.Accordion(translations["bass_or_treble"], open=True, visible=bass_or_treble.value) as bass_treble_accordion: with gr.Row(): bass_boost = gr.Slider(minimum=0, maximum=20, step=1, value=0, label=translations["bass_boost"], info=translations["bass_boost_info"], interactive=True) bass_frequency = gr.Slider(minimum=20, maximum=200, step=10, value=100, label=translations["bass_frequency"], info=translations["bass_frequency_info"], interactive=True) with gr.Row(): treble_boost = gr.Slider(minimum=0, maximum=20, step=1, value=0, label=translations["treble_boost"], info=translations["treble_boost_info"], interactive=True) treble_frequency = gr.Slider(minimum=1000, maximum=10000, step=500, value=3000, label=translations["treble_frequency"], info=translations["treble_frequency_info"], interactive=True) with gr.Accordion(translations["limiter"], open=True, visible=limiter.value) as limiter_accordion: with gr.Row(): limiter_threashold_db = gr.Slider(minimum=-60, maximum=0, step=1, value=-1, label=translations["limiter_threashold_db"], info=translations["limiter_threashold_db_info"], interactive=True) limiter_release_ms = gr.Slider(minimum=10, maximum=1000, step=1, value=100, label=translations["limiter_release_ms"], info=translations["limiter_release_ms_info"], interactive=True) with gr.Column(): pitch_shift_semitones = gr.Slider(minimum=-20, maximum=20, step=1, value=0, label=translations["pitch"], info=translations["pitch_info"], interactive=True) audio_effect_resample_sr = gr.Slider(minimum=0, maximum=96000, step=1, value=0, label=translations["resample"], info=translations["resample_info"], interactive=True, visible=resample_checkbox.value) distortion_drive_db = gr.Slider(minimum=0, maximum=50, step=1, value=20, label=translations["distortion"], info=translations["distortion_info"], interactive=True, visible=distortion_checkbox.value) gain_db = gr.Slider(minimum=-60, maximum=60, step=1, value=0, label=translations["gain"], info=translations["gain_info"], interactive=True, visible=gain_checkbox.value) clipping_threashold_db = gr.Slider(minimum=-60, maximum=0, step=1, value=-1, label=translations["clipping_threashold_db"], info=translations["clipping_threashold_db_info"], interactive=True, visible=clipping_checkbox.value) bitcrush_bit_depth = gr.Slider(minimum=1, maximum=24, step=1, value=16, label=translations["bitcrush_bit_depth"], info=translations["bitcrush_bit_depth_info"], interactive=True, visible=bitcrush_checkbox.value) with gr.Row(): with gr.Accordion(translations["phaser"], open=False, visible=phaser_check_box.value) as phaser_accordion: phaser_depth = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.5, label=translations["phaser_depth"], info=translations["phaser_depth_info"], interactive=True) phaser_rate_hz = gr.Slider(minimum=0.1, maximum=10, step=0.1, value=1, label=translations["phaser_rate_hz"], info=translations["phaser_rate_hz_info"], interactive=True) phaser_mix = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.5, label=translations["phaser_mix"], info=translations["phaser_mix_info"], interactive=True) phaser_centre_frequency_hz = gr.Slider(minimum=50, maximum=5000, step=10, value=1000, label=translations["phaser_centre_frequency_hz"], info=translations["phaser_centre_frequency_hz_info"], interactive=True) phaser_feedback = gr.Slider(minimum=-1, maximum=1, step=0.01, value=0, label=translations["phaser_feedback"], info=translations["phaser_feedback_info"], interactive=True) with gr.Row(): with gr.Accordion(translations["compressor"], open=False, visible=compressor_check_box.value) as compressor_accordion: compressor_threashold_db = gr.Slider(minimum=-60, maximum=0, step=1, value=-20, label=translations["compressor_threashold_db"], info=translations["compressor_threashold_db_info"], interactive=True) compressor_ratio = gr.Slider(minimum=1, maximum=20, step=0.1, value=1, label=translations["compressor_ratio"], info=translations["compressor_ratio_info"], interactive=True) compressor_attack_ms = gr.Slider(minimum=0.1, maximum=100, step=0.1, value=10, label=translations["compressor_attack_ms"], info=translations["compressor_attack_ms_info"], interactive=True) compressor_release_ms = gr.Slider(minimum=10, maximum=1000, step=1, value=100, label=translations["compressor_release_ms"], info=translations["compressor_release_ms_info"], interactive=True) with gr.Row(): gr.Markdown(translations["output_audio"]) with gr.Row(): audio_play_input = gr.Audio(show_download_button=True, interactive=False, label=translations["input_audio"]) audio_play_output = gr.Audio(show_download_button=True, interactive=False, label=translations["output_audio"]) with gr.Row(): reverb_check_box.change(fn=visible, inputs=[reverb_check_box], outputs=[reverb_accordion]) chorus_check_box.change(fn=visible, inputs=[chorus_check_box], outputs=[chorus_accordion]) delay_check_box.change(fn=visible, inputs=[delay_check_box], outputs=[delay_accordion]) with gr.Row(): compressor_check_box.change(fn=visible, inputs=[compressor_check_box], outputs=[compressor_accordion]) phaser_check_box.change(fn=visible, inputs=[phaser_check_box], outputs=[phaser_accordion]) more_options.change(fn=visible, inputs=[more_options], outputs=[more_accordion]) with gr.Row(): fade.change(fn=visible, inputs=[fade], outputs=[fade_accordion]) bass_or_treble.change(fn=visible, inputs=[bass_or_treble], outputs=[bass_treble_accordion]) limiter.change(fn=visible, inputs=[limiter], outputs=[limiter_accordion]) resample_checkbox.change(fn=visible, inputs=[resample_checkbox], outputs=[audio_effect_resample_sr]) with gr.Row(): distortion_checkbox.change(fn=visible, inputs=[distortion_checkbox], outputs=[distortion_drive_db]) gain_checkbox.change(fn=visible, inputs=[gain_checkbox], outputs=[gain_db]) clipping_checkbox.change(fn=visible, inputs=[clipping_checkbox], outputs=[clipping_threashold_db]) bitcrush_checkbox.change(fn=visible, inputs=[bitcrush_checkbox], outputs=[bitcrush_bit_depth]) with gr.Row(): upload_audio.upload(fn=lambda audio_in: shutil.move(audio_in.name, os.path.join("audios")), inputs=[upload_audio], outputs=[audio_in_path]) audio_in_path.change(fn=lambda audio: audio if audio else None, inputs=[audio_in_path], outputs=[audio_play_input]) audio_effects_refesh.click(fn=lambda a, b: [change_audios_choices(a), change_audios_choices(b)], inputs=[audio_in_path, audio_combination_input], outputs=[audio_in_path, audio_combination_input]) with gr.Row(): more_options.change(fn=lambda: [False]*8, inputs=[], outputs=[fade, bass_or_treble, limiter, resample_checkbox, distortion_checkbox, gain_checkbox, clipping_checkbox, bitcrush_checkbox]) audio_combination.change(fn=visible, inputs=[audio_combination], outputs=[audio_combination_input]) with gr.Row(): apply_effects_button.click( fn=audio_effects, inputs=[ audio_in_path, audio_out_path, resample_checkbox, audio_effect_resample_sr, chorus_depth, chorus_rate_hz, chorus_mix, chorus_centre_delay_ms, chorus_feedback, distortion_drive_db, reverb_room_size, reverb_damping, reverb_wet_level, reverb_dry_level, reverb_width, reverb_freeze_mode, pitch_shift_semitones, delay_second, delay_feedback, delay_mix, compressor_threashold_db, compressor_ratio, compressor_attack_ms, compressor_release_ms, limiter_threashold_db, limiter_release_ms, gain_db, bitcrush_bit_depth, clipping_threashold_db, phaser_rate_hz, phaser_depth, phaser_centre_frequency_hz, phaser_feedback, phaser_mix, bass_boost, bass_frequency, treble_boost, treble_frequency, fade_in, fade_out, audio_output_format, chorus_check_box, distortion_checkbox, reverb_check_box, delay_check_box, compressor_check_box, limiter, gain_checkbox, bitcrush_checkbox, clipping_checkbox, phaser_check_box, bass_or_treble, fade, audio_combination, audio_combination_input ], outputs=[audio_play_output], api_name="audio_effects" ) with gr.TabItem(translations["f0_extractor_tab"], visible=configs.get("f0_extractor_tab", True)): gr.Markdown(translations["f0_extractor_markdown"]) with gr.Row(): gr.Markdown(translations["f0_extractor_markdown_2"]) with gr.Row(): extractor_button = gr.Button(translations["extract_button"].replace("2. ", ""), variant="primary") with gr.Row(): with gr.Column(): upload_audio_file = gr.File(label=translations["drop_audio"], file_types=[".wav", ".mp3", ".flac", ".ogg", ".opus", ".m4a", ".mp4", ".aac", ".alac", ".wma", ".aiff", ".webm", ".ac3"]) audioplay = gr.Audio(show_download_button=True, interactive=False, label=translations["input_audio"]) with gr.Column(): with gr.Accordion(translations["f0_method"], open=False): with gr.Group(): onnx_f0_mode3 = gr.Checkbox(label=translations["f0_onnx_mode"], info=translations["f0_onnx_mode_info"], value=False, interactive=True) f0_method_extract = gr.Radio(label=translations["f0_method"], info=translations["f0_method_info"], choices=method_f0, value="rmvpe", interactive=True) with gr.Accordion(translations["audio_path"], open=True): input_audio_path = gr.Dropdown(label=translations["audio_path"], value="", choices=paths_for_files, allow_custom_value=True, interactive=True) refesh_audio_button = gr.Button(translations["refesh"]) with gr.Row(): gr.Markdown("___") with gr.Row(): file_output = gr.File(label="", file_types=[".txt"], interactive=False) image_output = gr.Image(label="", interactive=False, show_download_button=True) with gr.Row(): upload_audio_file.upload(fn=lambda audio_in: shutil.move(audio_in.name, os.path.join("audios")), inputs=[upload_audio_file], outputs=[input_audio_path]) input_audio_path.change(fn=lambda audio: audio if os.path.isfile(audio) else None, inputs=[input_audio_path], outputs=[audioplay]) refesh_audio_button.click(fn=change_audios_choices, inputs=[input_audio_path], outputs=[input_audio_path]) with gr.Row(): extractor_button.click( fn=f0_extract, inputs=[ input_audio_path, f0_method_extract, onnx_f0_mode3 ], outputs=[file_output, image_output], api_name="f0_extract" )