Upload infer_new.py

infer_new.py

@@ -318,126 +318,10 @@ def get_vc(sid, to_return_protect0):
         )
     )
 
-def find_audio_files(folder_path, extensions):
-    audio_files = []
-    for root, dirs, files in os.walk(folder_path):
-        for file in files:
-            if any(file.endswith(ext) for ext in extensions):
-                audio_files.append(file)
-    return audio_files
-
-def vc_multi(
-    spk_item,
-    vc_input,
-    vc_output,
-    vc_transform0,
-    f0method0,
-    file_index,
-    index_rate,
-    filter_radius,
-    resample_sr,
-    rms_mix_rate,
-    protect,
-):
-    global tgt_sr, net_g, vc, hubert_model, version, cpt
-    logs = []
-    logs.append("Converting...")
-    yield "\n".join(logs)
-    print()
-    try:
-        if os.path.exists(vc_input):
-            folder_path = vc_input
-            extensions = [".mp3", ".wav", ".flac", ".ogg"]
-            audio_files = find_audio_files(folder_path, extensions)
-            for index, file in enumerate(audio_files, start=1):
-                audio, sr = librosa.load(os.path.join(folder_path, file), sr=16000, mono=True)
-                input_audio_path = folder_path, file
-                f0_up_key = int(vc_transform0)
-                times = [0, 0, 0]
-                if hubert_model == None:
-                    load_hubert()
-                if_f0 = cpt.get("f0", 1)
-                audio_opt = vc.pipeline(
-                    hubert_model,
-                    net_g,
-                    spk_item,
-                    audio,
-                    input_audio_path,
-                    times,
-                    f0_up_key,
-                    f0method0,
-                    file_index,
-                    index_rate,
-                    if_f0,
-                    filter_radius,
-                    tgt_sr,
-                    resample_sr,
-                    rms_mix_rate,
-                    version,
-                    protect,
-                    f0_file=None
-                )
-                if resample_sr >= 16000 and tgt_sr != resample_sr:
-                    tgt_sr = resample_sr
-                output_path = f"{os.path.join(vc_output, file)}"
-                os.makedirs(os.path.join(vc_output), exist_ok=True)
-                sf.write(
-                    output_path,
-                    audio_opt,
-                    tgt_sr,
-                )
-                info = f"{index} / {len(audio_files)} | {file}"
-                print(info)
-                logs.append(info)
-                yield "\n".join(logs)
-        else:
-            logs.append("Folder not found or path doesn't exist.")
-            yield "\n".join(logs)
-    except:
-        info = traceback.format_exc()
-        print(info)
-        logs.append(info)
-        yield "\n".join(logs)
-
-def download_audio(url, audio_provider):
-    logs = []
-    os.makedirs("dl_audio", exist_ok=True)
-    if url == "":
-        logs.append("URL required!")
-        yield None, "\n".join(logs)
-        return None, "\n".join(logs)
-    if audio_provider == "Youtube":
-        logs.append("Downloading the audio...")
-        yield None, "\n".join(logs)
-        ydl_opts = {
-            'noplaylist': True,
-            'format': 'bestaudio/best',
-            'postprocessors': [{
-                'key': 'FFmpegExtractAudio',
-                'preferredcodec': 'wav',
-            }],
-            "outtmpl": 'result/dl_audio/audio',
-        }
-        audio_path = "result/dl_audio/audio.wav"
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            ydl.download([url])
-        logs.append("Download Complete.")
-        yield audio_path, "\n".join(logs)
-
-def cut_vocal_and_inst_yt(split_model,spk_id):
-    logs = []
-    logs.append("Starting the audio splitting process...")
-    yield "\n".join(logs), None, None, None
-    command = f"demucs --two-stems=vocals -n {split_model} result/dl_audio/audio.wav -o output"
-    result = subprocess.Popen(command.split(), stdout=subprocess.PIPE, text=True)
-    for line in result.stdout:
-        logs.append(line)
-        yield "\n".join(logs), None, None, None
-    print(result.stdout)
-    vocal = f"output/{split_model}/{spk_id}_input_audio/vocals.wav"
-    inst = f"output/{split_model}/{spk_id}_input_audio/no_vocals.wav"
-    logs.append("Audio splitting complete.")
-    yield "\n".join(logs), vocal, inst, vocal
+
+
+
+
 
 def cut_vocal_and_inst(audio_path,spk_id):
     
@@ -497,73 +381,5 @@ def combine_vocal_and_inst(vocal_path, inst_path):
     #print(result.stdout.decode())
     #return output_path
 
-def download_and_extract_models(urls):
-    logs = []
-    os.makedirs("zips", exist_ok=True)
-    os.makedirs(os.path.join("zips", "extract"), exist_ok=True)
-    os.makedirs(os.path.join(weight_root), exist_ok=True)
-    os.makedirs(os.path.join(index_root), exist_ok=True)
-    for link in urls.splitlines():
-        url = link.strip()
-        if not url:
-            raise gr.Error("URL Required!")
-            return "No URLs provided."
-        model_zip = urlparse(url).path.split('/')[-2] + '.zip'
-        model_zip_path = os.path.join('zips', model_zip)
-        logs.append(f"Downloading...")
-        yield "\n".join(logs)
-        if "drive.google.com" in url:
-            gdown.download(url, os.path.join("zips", "extract"), quiet=False)
-        elif "mega.nz" in url:
-            m = Mega()
-            m.download_url(url, 'zips')
-        else:
-            os.system(f"wget {url} -O {model_zip_path}")
-        logs.append(f"Extracting...")
-        yield "\n".join(logs)
-        for filename in os.listdir("zips"):
-            archived_file = os.path.join("zips", filename)
-            if filename.endswith(".zip"):
-                shutil.unpack_archive(archived_file, os.path.join("zips", "extract"), 'zip')
-            elif filename.endswith(".rar"):
-                with rarfile.RarFile(archived_file, 'r') as rar:
-                    rar.extractall(os.path.join("zips", "extract"))
-        for _, dirs, files in os.walk(os.path.join("zips", "extract")):
-            logs.append(f"Searching Model and Index...")
-            yield "\n".join(logs)
-            model = False
-            index = False
-            if files:
-                for file in files:
-                    if file.endswith(".pth"):
-                        basename = file[:-4]
-                        shutil.move(os.path.join("zips", "extract", file), os.path.join(weight_root, file))
-                        model = True
-                    if file.endswith('.index') and "trained" not in file:
-                        shutil.move(os.path.join("zips", "extract", file), os.path.join(index_root, file))
-                        index = True
-            else:
-                logs.append("No model in main folder.")
-                yield "\n".join(logs)
-                logs.append("Searching in subfolders...")
-                yield "\n".join(logs)
-                for sub_dir in dirs:
-                    for _, _, sub_files in os.walk(os.path.join("zips", "extract", sub_dir)):
-                        for file in sub_files:
-                            if file.endswith(".pth"):
-                                basename = file[:-4]
-                                shutil.move(os.path.join("zips", "extract", sub_dir, file), os.path.join(weight_root, file))
-                                model = True
-                            if file.endswith('.index') and "trained" not in file:
-                                shutil.move(os.path.join("zips", "extract", sub_dir, file), os.path.join(index_root, file))
-                                index = True  
-                        shutil.rmtree(os.path.join("zips", "extract", sub_dir))
-            if index is False:
-                logs.append("Model only file, no Index file detected.")
-                yield "\n".join(logs)
-        logs.append("Download Completed!")
-        yield "\n".join(logs)
-    logs.append("Successfully download all models! Refresh your model list to load the model")
-    yield "\n".join(logs)
 if __name__ == '__main__':
     app.run(debug=False, port=5000,host='0.0.0.0')