File size: 6,797 Bytes
643df07
ed6a17b
3d8e1f4
a03fefd
22ac523
 
 
 
3d8e1f4
a03fefd
3d8e1f4
 
 
 
 
22ac523
3d8e1f4
a03fefd
3d8e1f4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
20765d0
c5e2020
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3d8e1f4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
c5e2020
ed6a17b
d414b06
ed6a17b
 
 
3d8e1f4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ed6a17b
3d8e1f4
 
643df07
3d8e1f4
 
22ac523
 
 
 
ed6a17b
6a6e875
3d8e1f4
 
 
 
 
 
 
 
 
 
 
 
266a456
3d8e1f4
 
 
 
 
 
 
 
 
 
 
 
22ac523
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
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()