Update app_multi.py

app_multi.py

@@ -74,8 +74,48 @@ app_css = '''
 }
 '''
 
+
+
+rvc_models_dir = os.path.join(BASE_DIR, 'model')
+
+
+def download_online_model(url, dir_name, progress=gr.Progress()):
+    try:
+        progress(0, desc=f'[~] Downloading voice model with name {dir_name}...')
+        zip_name = url.split('/')[-1]
+        extraction_folder = os.path.join(rvc_models_dir, dir_name)
+        if os.path.exists(extraction_folder):
+            raise gr.Error(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}'
+
+        urllib.request.urlretrieve(url, zip_name)
+
+        progress(0.5, desc='[~] Extracting zip...')
+        extract_zip(extraction_folder, zip_name)
+        return f'[+] {dir_name} Model successfully downloaded!'
+
+    except Exception as e:
+        raise gr.Error(str(e))
+
+
+def upload_local_model(zip_path, dir_name, progress=gr.Progress()):
+    try:
+        extraction_folder = os.path.join(rvc_models_dir, dir_name)
+        if os.path.exists(extraction_folder):
+            raise gr.Error(f'Voice model directory {dir_name} already exists! Choose a different name for your voice model.')
+
+        zip_name = zip_path.name
+        progress(0.5, desc='[~] Extracting zip...')
+        extract_zip(extraction_folder, zip_name)
+        return f'[+] {dir_name} Model successfully uploaded!'
+
+    except Exception as e:
+        raise gr.Error(str(e))
+
 app = gr.Blocks(
-    theme=gr.themes.Soft(primary_hue="orange", secondary_hue="slate"),
+    theme=gr.themes.Soft(primary_hue="indigo", secondary_hue="slate"),
     css=app_css,
     analytics_enabled=False
 )
@@ -487,7 +527,31 @@ with app:
         outputs=[model_info],
         show_progress=False,
         queue=False
-    )
+    ),
+    with gr.Tab('Download model'):
+
+            with gr.Tab('From HuggingFace/Pixeldrain URL'):
+                with gr.Row():
+                    model_zip_link = gr.Text(label='Download link to model', info='Should be a zip file containing a .pth model file and an optional .index file.')
+                    model_name = gr.Text(label='Name your model', info='Give your new model a unique name from your other voice models.')
+
+                with gr.Row():
+                    download_btn = gr.Button('Download 🌐', variant='primary', scale=19)
+                    dl_output_message = gr.Text(label='Output Message', interactive=False, scale=20)
+
+                download_btn.click(download_online_model, inputs=[model_zip_link, model_name], outputs=dl_output_message)
+
+                gr.Markdown('## Input Examples')
+                gr.Examples(
+                    [
+                        ['https://huggingface.co/phant0m4r/LiSA/resolve/main/LiSA.zip', 'Lisa'],
+                        ['https://pixeldrain.com/u/3tJmABXA', 'Gura'],
+                        ['https://huggingface.co/Kit-Lemonfoot/kitlemonfoot_rvc_models/resolve/main/AZKi%20(Hybrid).zip', 'Azki']
+                    ],
+                    [model_zip_link, model_name],
+                    [],
+                    download_online_model,
+                )
 
 app.queue(
     concurrency_count=1,