Update app.py

app.py

@@ -13,6 +13,20 @@ from lib.infer import infer_audio
 import edge_tts
 import tempfile
 import anyio
+from pathlib import Path
+import os
+import zipfile
+import shutil
+import urllib.request
+import gdown
+import subprocess
+
+main_dir = Path().resolve()
+print(main_dir)
+
+os.chdir(main_dir)
+models_dir = "models"
+
 
 
 language_dict = {
@@ -363,6 +377,60 @@ async def text_to_speech_edge(text, language_code):
 
 
 
+
+def extract_zip(extraction_folder, zip_name):
+    os.makedirs(extraction_folder)
+    with zipfile.ZipFile(zip_name, 'r') as zip_ref:
+        zip_ref.extractall(extraction_folder)
+    os.remove(zip_name)
+
+    index_filepath, model_filepath = None, None
+    for root, dirs, files in os.walk(extraction_folder):
+        for name in files:
+            if name.endswith('.index') and os.stat(os.path.join(root, name)).st_size > 1024 * 100:
+                index_filepath = os.path.join(root, name)
+
+            if name.endswith('.pth') and os.stat(os.path.join(root, name)).st_size > 1024 * 1024 * 40:
+                model_filepath = os.path.join(root, name)
+
+    if not model_filepath:
+        raise Exception(f'No .pth model file was found in the extracted zip. Please check {extraction_folder}.')
+
+    # move model and index file to extraction folder
+    os.rename(model_filepath, os.path.join(extraction_folder, os.path.basename(model_filepath)))
+    if index_filepath:
+        os.rename(index_filepath, os.path.join(extraction_folder, os.path.basename(index_filepath)))
+
+    # remove any unnecessary nested folders
+    for filepath in os.listdir(extraction_folder):
+        if os.path.isdir(os.path.join(extraction_folder, filepath)):
+            shutil.rmtree(os.path.join(extraction_folder, filepath))
+
+def download_online_model(url, dir_name):
+    try:
+        print(f'[~] Downloading voice model with name {dir_name}...')
+        zip_name = url.split('/')[-1]
+        extraction_folder = os.path.join(models_dir, dir_name)
+        if os.path.exists(extraction_folder):
+            raise Exception(f'Voice model directory {dir_name} already exists! Choose a different name for your voice model.')
+
+        if 'pixeldrain.com' in url:
+            url = f'https://pixeldrain.com/api/file/{zip_name}'
+        if 'drive.google.com' in url:
+          zip_name = dir_name + ".zip"
+          gdown.download(url, output=zip_name, use_cookies=True, quiet=True, fuzzy=True)
+        else:
+            urllib.request.urlretrieve(url, zip_name)
+
+        print(f'[~] Extracting zip file...')
+        extract_zip(extraction_folder, zip_name)
+        print(f'[+] {dir_name} Model successfully downloaded!')
+
+    except Exception as e:
+        raise Exception(str(e))
+
+
+
 # Gradio Blocks Interface with Tabs
 with gr.Blocks(title="Hex RVC") as app:
     gr.Markdown("# Hex RVC")
@@ -419,6 +487,11 @@ with gr.Blocks(title="Hex RVC") as app:
             outputs=output_audio
         )
 
+    with gr.Tqb("Download RVC Model"):
+        url = gr.Textbox(label="Your model URL")
+        dirname = gr.Textbox(label="Your Model name")
+        button_model = gr.Button("Download model")
+        button_model.click(fn=download_online_model, inputs=url, dirname, output=dirname)
     with gr.Tab("Audio Separation"):
         with gr.Row():
             input_audio = gr.Audio(source="upload", type="filepath", label="Upload Audio File")