Spaces:
Running
Running
| import gradio as gr | |
| import os, shutil | |
| import subprocess | |
| def convert(audio_picker,model_picker): | |
| command = [ | |
| "python", | |
| "tools/infer_cli.py", | |
| "--f0up_key", "0", | |
| "--input_path", f"audios/{audio_picker}", | |
| "--index_path", "", | |
| "--f0method", "rmvpe", | |
| "--opt_path", "cli_output.wav", | |
| "--model_name", f"{model_picker}", | |
| "--index_rate", "0.8", | |
| "--device", "cpu", | |
| "--is_half", "False", | |
| "--filter_radius", "3", | |
| "--resample_sr", "0", | |
| "--rms_mix_rate", "0.21", | |
| "--protect", "0" | |
| ] | |
| try: | |
| process = subprocess.Popen(command) | |
| process.wait() # Wait for the subprocess to finish | |
| print("Script executed successfully.") | |
| except subprocess.CalledProcessError as e: | |
| print(f"Error: {e}") | |
| assets_folder = "assets" | |
| if not os.path.exists(assets_folder): | |
| os.makedirs(assets_folder) | |
| files = { | |
| "rmvpe/rmvpe.pt":"https://huggingface.co/Rejekts/project/resolve/main/rmvpe.pt", | |
| "hubert/hubert_base.pt":"https://huggingface.co/Rejekts/project/resolve/main/hubert_base.pt", | |
| "pretrained_v2/D40k.pth":"https://huggingface.co/Rejekts/project/resolve/main/D40k.pth", | |
| "pretrained_v2/G40k.pth":"https://huggingface.co/Rejekts/project/resolve/main/G40k.pth", | |
| "pretrained_v2/f0D40k.pth":"https://huggingface.co/Rejekts/project/resolve/main/f0D40k.pth", | |
| "pretrained_v2/f0G40k.pth":"https://huggingface.co/Rejekts/project/resolve/main/f0G40k.pth" | |
| } | |
| for file, link in files.items(): | |
| file_path = os.path.join(assets_folder, file) | |
| if not os.path.exists(file_path): | |
| try: | |
| subprocess.run(['wget', link, '-O', file_path], check=True) | |
| except subprocess.CalledProcessError as e: | |
| print(f"Error downloading {file}: {e}") | |
| def download_from_url(url, model): | |
| if url == '': | |
| return "URL cannot be left empty." | |
| if model =='': | |
| return "You need to name your model. For example: My-Model" | |
| url = url.strip() | |
| zip_dirs = ["zips", "unzips"] | |
| for directory in zip_dirs: | |
| if os.path.exists(directory): | |
| shutil.rmtree(directory) | |
| os.makedirs("zips", exist_ok=True) | |
| os.makedirs("unzips", exist_ok=True) | |
| zipfile = model + '.zip' | |
| zipfile_path = './zips/' + zipfile | |
| try: | |
| if "drive.google.com" in url: | |
| subprocess.run(["gdown", url, "--fuzzy", "-O", zipfile_path]) | |
| elif "mega.nz" in url: | |
| m = Mega() | |
| m.download_url(url, './zips') | |
| else: | |
| subprocess.run(["wget", url, "-O", zipfile_path]) | |
| for filename in os.listdir("./zips"): | |
| if filename.endswith(".zip"): | |
| zipfile_path = os.path.join("./zips/",filename) | |
| shutil.unpack_archive(zipfile_path, "./unzips", 'zip') | |
| else: | |
| return "No zipfile found." | |
| for root, dirs, files in os.walk('./unzips'): | |
| for file in files: | |
| file_path = os.path.join(root, file) | |
| if file.endswith(".index"): | |
| os.mkdir(f'./logs/{model}') | |
| shutil.copy2(file_path,f'./logs/{model}') | |
| elif "G_" not in file and "D_" not in file and file.endswith(".pth"): | |
| shutil.copy(file_path,f'./assets/weights/{model}.pth') | |
| shutil.rmtree("zips") | |
| shutil.rmtree("unzips") | |
| return "Success." | |
| except: | |
| return "There's been an error." | |
| def show_available(filepath): | |
| return os.listdir(filepath) | |
| def upload_file(file): | |
| audio_formats = ['.wav', '.mp3', '.ogg', '.flac', '.aac'] | |
| _, ext = os.path.splitext(file.name) | |
| filename = os.path.basename(file.name) | |
| file_path = file.name | |
| if ext.lower() in audio_formats: | |
| if os.path.exists(f'audios/{filename}'): | |
| os.remove(f'audios/{filename}') | |
| shutil.move(file_path,'audios') | |
| elif ext.lower().endswith('.pth'): | |
| if os.path.exists(f'assets/weights/{filename}'): | |
| os.remove(f'assets/weights/{filename}') | |
| shutil.move(file_path,'assets/weights') | |
| elif ext.lower().endswith('.index'): | |
| if os.path.exists(f'logs/{filename}'): | |
| os.remove(f'logs/{filename}') | |
| shutil.move(file_path,'logs') | |
| else: | |
| gr.Warning('File incompatible') | |
| return {"choices":show_available('audios'),"__type__": "update"},{"choices":show_available('assets/weights'),"__type__": "update"},None | |
| def refresh(): | |
| return {"choices":show_available('audios'),"__type__": "update"},{"choices":show_available('assets/weights'),"__type__": "update"} | |
| with gr.Blocks() as app: | |
| with gr.Row(): | |
| with gr.Column(): | |
| with gr.Tabs(): | |
| with gr.TabItem("1.Choose a voice model:"): | |
| model_picker = gr.Dropdown(label="",choices=show_available('assets/weights'),value='',interactive=True) | |
| with gr.TabItem("(Or download a model here)"): | |
| with gr.Row(): | |
| url = gr.Textbox(label="Paste the URL here:",value="",placeholder="(i.e. https://huggingface.co/repo/model/resolve/main/model.zip)") | |
| with gr.Row(): | |
| with gr.Column(): | |
| model_rename = gr.Textbox(placeholder="My-Model", label="Name your model:",value="") | |
| with gr.Column(): | |
| download_button = gr.Button("Download") | |
| download_button.click(fn=download_from_url,inputs=[url,model_rename],outputs=[]) | |
| with gr.Row(): | |
| with gr.Tabs(): | |
| with gr.TabItem("2.Choose an audio file:"): | |
| audio_picker = gr.Dropdown(label="",choices=show_available('audios'),value='',interactive=True) | |
| with gr.TabItem("(Or upload a new file here)"): | |
| dropbox = gr.File(label="Drop an audio here. (You can also drop a .pth or .index file here)") | |
| dropbox.upload(fn=upload_file, inputs=[dropbox],outputs=[audio_picker,model_picker,dropbox]) | |
| audio_refresher = gr.Button("Refresh") | |
| audio_refresher.click(fn=refresh,inputs=[],outputs=[audio_picker,model_picker]) | |
| convert_button = gr.Button("Convert") | |
| convert_button.click(convert, inputs=[audio_picker,model_picker]) | |
| app.queue() | |
| app.launch() |