move cli
This commit is contained in:
@@ -1,33 +0,0 @@
|
||||
import re
|
||||
from typing import Optional, Pattern
|
||||
|
||||
import click
|
||||
|
||||
from rvc.cli.handler.infer import infer
|
||||
from rvc.cli.handler.train import train
|
||||
from rvc.cli.handler.uvr5 import uvr
|
||||
from rvc.cli.utils.dlmodel import dlmodel
|
||||
from rvc.cli.utils.env import env
|
||||
from rvc.cli.utils.initialize import init
|
||||
|
||||
|
||||
@click.group(
|
||||
context_settings={"help_option_names": ["-h", "--help"]},
|
||||
help="rvc cli feature list",
|
||||
)
|
||||
def cli():
|
||||
pass
|
||||
|
||||
|
||||
def main():
|
||||
cli.add_command(infer)
|
||||
cli.add_command(train)
|
||||
cli.add_command(uvr)
|
||||
cli.add_command(dlmodel)
|
||||
cli.add_command(env)
|
||||
cli.add_command(init)
|
||||
cli()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@@ -1,130 +0,0 @@
|
||||
import logging
|
||||
from pathlib import Path
|
||||
|
||||
import click
|
||||
from dotenv import load_dotenv
|
||||
from scipy.io import wavfile
|
||||
|
||||
from rvc.modules.vc.modules import VC
|
||||
|
||||
logging.getLogger("numba").setLevel(logging.WARNING)
|
||||
|
||||
|
||||
@click.command(
|
||||
context_settings={"help_option_names": ["-h", "--help"]},
|
||||
help="inference audio",
|
||||
)
|
||||
@click.option(
|
||||
"-m",
|
||||
"--modelPath",
|
||||
is_flag=False,
|
||||
type=str,
|
||||
help="Model path or filename (reads in the directory set in env)",
|
||||
required=True,
|
||||
)
|
||||
@click.option(
|
||||
"-i",
|
||||
"--inputPath",
|
||||
is_flag=False,
|
||||
type=Path,
|
||||
help="input audio path or folder",
|
||||
required=True,
|
||||
)
|
||||
@click.option(
|
||||
"-o",
|
||||
"--outputPath",
|
||||
is_flag=False,
|
||||
type=Path,
|
||||
help="output audio path or folder",
|
||||
required=True,
|
||||
)
|
||||
@click.option(
|
||||
"-s", "--sid", is_flag=False, type=int, help="Speaker/Singer id", default=0
|
||||
)
|
||||
@click.option("-fu", "--f0upkey", is_flag=False, type=int, help="Transpose", default=0)
|
||||
@click.option(
|
||||
"-fm",
|
||||
"--f0method",
|
||||
is_flag=False,
|
||||
type=str,
|
||||
help="Pitch extraction algorith",
|
||||
default="rmvpe",
|
||||
)
|
||||
@click.option(
|
||||
"-ff", "--f0file", is_flag=False, type=Path, help="F0 curve file (optional)"
|
||||
)
|
||||
@click.option("-if", "--indexFile", is_flag=False, type=Path, help="Feature index file")
|
||||
@click.option(
|
||||
"-ir",
|
||||
"--indexRate",
|
||||
is_flag=False,
|
||||
type=float,
|
||||
help="Search feature ratio",
|
||||
default=0.75,
|
||||
)
|
||||
@click.option(
|
||||
"-fr",
|
||||
"--filterRadius",
|
||||
is_flag=False,
|
||||
type=int,
|
||||
help="Apply median filtering",
|
||||
default=3,
|
||||
)
|
||||
@click.option(
|
||||
"-rsr",
|
||||
"--resamplesr",
|
||||
is_flag=False,
|
||||
type=int,
|
||||
help="Resample the output audio",
|
||||
default=0,
|
||||
)
|
||||
@click.option(
|
||||
"-rmr",
|
||||
"--rmsmixrate",
|
||||
is_flag=False,
|
||||
type=float,
|
||||
help="Adjust the volume envelope scaling",
|
||||
default=0.25,
|
||||
)
|
||||
@click.option(
|
||||
"-p",
|
||||
"--protect",
|
||||
is_flag=False,
|
||||
type=float,
|
||||
help="Protect voiceless consonants and breath sounds",
|
||||
default=0.33,
|
||||
)
|
||||
def infer(
|
||||
modelpath,
|
||||
inputpath,
|
||||
outputpath,
|
||||
sid,
|
||||
f0upkey,
|
||||
f0method,
|
||||
f0file,
|
||||
indexfile,
|
||||
indexrate,
|
||||
filterradius,
|
||||
resamplesr,
|
||||
rmsmixrate,
|
||||
protect,
|
||||
):
|
||||
load_dotenv()
|
||||
vc = VC()
|
||||
vc.get_vc(modelpath)
|
||||
tgt_sr, audio_opt, times, _ = vc.vc_single(
|
||||
sid,
|
||||
inputpath,
|
||||
f0upkey,
|
||||
f0method,
|
||||
f0file,
|
||||
indexfile,
|
||||
indexrate,
|
||||
filterradius,
|
||||
resamplesr,
|
||||
rmsmixrate,
|
||||
protect,
|
||||
)
|
||||
wavfile.write(outputpath, tgt_sr, audio_opt)
|
||||
click.echo(times)
|
||||
click.echo(f"Finish inference. Check {outputpath}")
|
||||
@@ -1,6 +0,0 @@
|
||||
import click
|
||||
|
||||
|
||||
@click.command()
|
||||
def train():
|
||||
pass
|
||||
@@ -1,6 +0,0 @@
|
||||
import click
|
||||
|
||||
|
||||
@click.command()
|
||||
def uvr():
|
||||
pass
|
||||
@@ -1,9 +0,0 @@
|
||||
import urllib
|
||||
|
||||
import click
|
||||
|
||||
|
||||
@click.command()
|
||||
def dlmodel() -> None:
|
||||
# Download models [harvest, uvr5, and more ]
|
||||
pass
|
||||
@@ -1,40 +0,0 @@
|
||||
"""
|
||||
setup or cleanup enviroment file
|
||||
usage: rvc env [create / cleanup]
|
||||
Default: [nowDir/.env]
|
||||
|
||||
"""
|
||||
|
||||
import os
|
||||
|
||||
import click
|
||||
|
||||
|
||||
@click.group()
|
||||
def env():
|
||||
pass
|
||||
|
||||
|
||||
@env.command()
|
||||
def create():
|
||||
env_file_path = os.path.join(os.getcwd(), ".env")
|
||||
|
||||
if not os.path.exists(env_file_path):
|
||||
default_values = {
|
||||
"weight_root": "",
|
||||
"weight_uvr5_root": "",
|
||||
"index_root": "",
|
||||
"rmvpe_root": "",
|
||||
"hubert_path": "",
|
||||
"save_uvr_path": "",
|
||||
"TEMP": "",
|
||||
"pretrained": "",
|
||||
}
|
||||
|
||||
with open(env_file_path, "w") as env_file:
|
||||
for key, value in default_values.items():
|
||||
env_file.write(f"{key}={value}\n")
|
||||
|
||||
click.echo(f"{env_file_path} created successfully.")
|
||||
else:
|
||||
click.echo(f"{env_file_path} already exists, no change")
|
||||
@@ -1,11 +0,0 @@
|
||||
"""
|
||||
Uage: rvc init
|
||||
download model and setup environmmnt file
|
||||
|
||||
"""
|
||||
import click
|
||||
|
||||
|
||||
@click.command()
|
||||
def init():
|
||||
pass
|
||||
Reference in New Issue
Block a user