import gradio as gr import os, shutil import subprocess from datetime import datetime os.environ["rmvpe_root"] = "assets/rmvpe" os.environ['index_root']="logs" os.environ['weight_root']="assets/weights" def convert(audio_picker,model_picker): now = datetime.now().strftime("%d%m%Y%H%M%S") command = [ "python", "tools/infer_cli.py", "--f0up_key", "0", "--input_path", f"audios/{audio_picker}", "--index_path", f"logs/{model_picker}/*.index", "--f0method", "rmvpe", "--opt_path", f"audios/cli_output_{now}.wav", "--model_name", f"{model_picker}", "--index_rate", "0.8", "--device", "cpu", "--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"} def update_audio_player(choice): return os.path.join("audios",choice) 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=[url]) 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]) with gr.Row(): audio_player = gr.Audio() audio_picker.change(fn=update_audio_player, inputs=[audio_picker],outputs=[audio_player]) app.queue().launch()