import gradio as gr import os, shutil import subprocess, glob from datetime import datetime from tools.useftools import * os.environ["rmvpe_root"] = "assets/rmvpe" os.environ['index_root']="logs" os.environ['weight_root']="assets/weights" from infer.modules.vc.modules import VC from configs.config import Config import torch from next import * os.makedirs(os.path.join(".", "audios"), exist_ok=True) config = Config() vc = VC(config) def warn(text): try: gr.Warning(text) except: pass def load_model(model_picker,index_picker): logs = show_available("logs") if model_picker.replace(".pth","") in logs: log = model_picker.replace(".pth","") else: log = index_picker warn("Could not find a matching index file.") vc.get_vc(model_picker,0,0) return {"choices":logs,"value":log,"__type__": "update"} def convert(audio_picker,model_picker,index_picker,index_rate,pitch,method): warn("Your audio is being converted. Please wait.") now = datetime.now().strftime("%d%m%Y%H%M%S") index_files = glob.glob(f"logs/{index_picker}/*.index") if index_files: print(f"Found index: {index_files[0]}") else: warn("Sorry, I couldn't find your .index file.") print("Did not find a matching .index file") index_files = [f'logs/{model_picker}/fake_index.index'] device = "cuda" if torch.cuda.is_available() else "cpu" command = [ "python", "tools/infer_cli.py", "--f0up_key", str(pitch), "--input_path", f"audios/{audio_picker}", "--index_path", index_files[0], "--f0method", method, "--opt_path", f"audios/cli_output_{now}.wav", "--model_name", f"{model_picker}", "--index_rate", str(float(index_rate)), "--device", device, "--filter_radius", "3", "--resample_sr", "0", "--rms_mix_rate", "0.0", "--protect", "0" ] try: process = subprocess.run(command, check=True) print("Script executed successfully.") return {"choices":show_available("audios"),"__type__":"update","value":f"cli_output_{now}.wav"},f"audios/cli_output_{now}.wav" except subprocess.CalledProcessError as e: print(f"Error: {e}") return {"choices":show_available("audios"),"__type__":"update"}, None assets_folder = "assets" if not os.path.exists(assets_folder): os.makedirs(assets_folder) files = { "rmvpe/rmvpe.pt":"https://huggingface.co/Hev832/baser-vc/blob/main/rmvpe.pt", "hubert/hubert_base.pt":"https://huggingface.co/Hev832/baser-vc/blob/main/hubert_base.pt" } 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 model =='': try: model = url.split('/')[-1].split('?')[0] except: return "You need to name your model. For example: My-Model", {"choices":show_available("assets/weights"),"__type__":"update"} url=url.replace('/blob/main/','/resolve/main/') model=model.replace('.pth','').replace('.index','').replace('.zip','') print(f"Model name: {model}") if url == '': return "URL cannot be left empty.", {"choices":show_available("assets/weights"),"__type__":"update"} 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 url.endswith('.pth'): subprocess.run(["wget", url, "-O", f'./assets/weights/{model}.pth']) return f"Sucessfully downloaded as {model}.pth", {"choices":show_available("assets/weights"),"__type__":"update"} elif url.endswith('.index'): if not os.path.exists(f'./logs/{model}'): os.makedirs(f'./logs/{model}') subprocess.run(["wget", url, "-O", f'./logs/{model}/added_{model}.index']) return f"Successfully downloaded as added_{model}.index", {"choices":show_available("assets/weights"),"__type__":"update"} 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') 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') elif filename.endswith(".pth"): shutil.copy2(os.path.join("./zips/",filename),f'./assets/weights/{model}.pth') elif filename.endswith(".index"): os.mkdir(f'./logs/{model}') shutil.copy2(os.path.join("./zips/",filename),f'./logs/{model}/') else: return "No zipfile found.", {"choices":show_available("assets/weights"),"__type__":"update"} shutil.rmtree("zips") shutil.rmtree("unzips") return "Success.", {"choices":show_available("assets/weights"),"__type__":"update"} except: return "There's been an error.", {"choices":show_available("assets/weights"),"__type__":"update"} def import_from_name(model): try: url = models[f'{model}'] except: return "", {"__type__":"update"} url=url.replace('/blob/main/','/resolve/main/') print(f"Model name: {model}") if url == '': return "", {"__type__":"update"} 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 url.endswith('.pth'): subprocess.run(["wget", url, "-O", f'./assets/weights/{model}.pth']) return f"", {"choices":show_available("assets/weights"),"__type__":"update","value":f"{model}.pth"} 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') 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') elif filename.endswith(".pth"): shutil.copy2(os.path.join("./zips/",filename),f'./assets/weights/{model}.pth') elif filename.endswith(".index"): os.mkdir(f'./logs/{model}') shutil.copy2(os.path.join("./zips/",filename),f'./logs/{model}/') else: return "", {"__type__":"update"} shutil.rmtree("zips") shutil.rmtree("unzips") return "", {"choices":show_available("assets/weights"),"__type__":"update","value":f"{model}.pth"} except: return "", {"__type__":"update"} def show_available(filepath,format=None): if format: print(f"Format: {format}") files = [] for file in os.listdir(filepath): if file.endswith(format): print(f"Matches format: {file}") files.append(file) else: print(f"Does not match format: {file}") print(f"Matches: {files}") if len(files) < 1: return [''] return files if len(os.listdir(filepath)) < 1: return [''] return os.listdir(filepath) def upload_file(file): audio_formats = ['.wav', '.mp3', '.ogg', '.flac', '.aac'] print(file) try: _, ext = os.path.splitext(file.name) filename = os.path.basename(file.name) file_path = file.name except AttributeError: _, ext = os.path.splitext(file) filename = os.path.basename(file) file_path = file if ext.lower() in audio_formats: if os.path.exists(f'audios/{filename}'): os.remove(f'audios/{filename}') shutil.move(file_path,'audios') else: warn('File incompatible') return {"choices":show_available('audios'),"__type__": "update","value":filename} def refresh_audio(): return {"choices":show_available("audios"),"__type__": "update"} def refresh_model(): return {"choices":show_available("assets/weights",".pth"),"__type__": "update"},{"choices":show_available("logs"),"__type__": "update"} def update_audio_player(choice): return os.path.join("audios",choice) with gr.Blocks(theme=blue_theme) as app: gr.Markdown("# Nex RVC 💻") gr.Markdown("## Easy and fast RVC 🗣️") with gr.TabItem("inference"): model_picker = gr.Dropdown(label="Model: ", choices=show_available('assets/weights', '.pth'), value=show_available('assets/weights', '.pth')[0], interactive=True, allow_custom_value=True) index_picker = gr.Dropdown(label="Index:", choices=show_available('logs'), value=show_available('logs')[0], interactive=True, allow_custom_value=True) model_picker.change(fn=load_model, inputs=[model_picker, index_picker], outputs=[index_picker]) with gr.Row(): model_refresher = gr.Button("Refresh Model") model_refresher.click(fn=refresh_model, inputs=[], outputs=[model_picker, index_picker]) with gr.Accordion('settings', open=False): index_rate = gr.Slider(label='Index Rate:', minimum=0, maximum=1, value=0.66, step=0.01) pitch = gr.Slider(label='Pitch (-12 to 12):', minimum=-12, maximum=12, step=1, value=0, interactive=True) method = gr.Dropdown(label="Method:", choices=["rmvpe", "pm"], value="rmvpe") recorder = gr.Microphone(label="Record audio here...", type='filepath') audio_picker = gr.Dropdown(label="", choices=show_available('audios'), value='', interactive=True) recorder.upload(upload_file, inputs=[recorder], outputs=[audio_picker]) dropbox = gr.File(label="Drop an audio here.", file_types=['.wav', '.mp3', '.ogg', '.flac', '.aac']) dropbox.upload(fn=upload_file, inputs=[dropbox], outputs=[audio_picker]) audio_refresher = gr.Button("Refresh Audio") audio_refresher.click(fn=refresh_audio, inputs=[], outputs=[audio_picker]) convert_button = gr.Button("Convert") audio_player = gr.Audio() inputs = [audio_picker, model_picker, index_picker, index_rate, pitch, method] audio_picker.change(fn=update_audio_player, inputs=[audio_picker], outputs=[audio_player]) convert_button.click(convert, inputs=inputs, outputs=[audio_picker, audio_player]) with gr.TabItem("Download models form url"): url = gr.Textbox(label="Paste the URL here:", value="", placeholder="(i.e. https://huggingface.co/repo/model/resolve/main/model.zip)") model_rename = gr.Textbox(placeholder="My-Model", label="Name your model:", value="") download_button = gr.Button("Download") download_button.click(fn=download_from_url, inputs=[url, model_rename], outputs=[url, model_picker]) with gr.Row(): selected_import = gr.Dropdown(choices=list(models.keys()), label="Search Models (Quality UNKNOWN)", scale=5) import_model = gr.Button("Download") import_model.click(fn=import_from_name, inputs=[selected_import], outputs=[selected_import, model_picker]) app.queue(max_size=20).launch(debug=True, share=True)