Update app.py

app.py

@@ -1,7 +1,33 @@
 import gradio as gr
 import os, shutil
+import subprocess
+
+def convert(audio_picker,model_picker):
+    command = [
+        "python",
+        "tools/infer_cli.py",
+        "--f0up_key", "0",
+        "--input_path", f"audios/{audio_picker}",
+        "--index_path", "",
+        "--f0method", "rmvpe",
+        "--opt_path", "cli_output.wav",
+        "--model_name", f"{model_picker}",
+        "--index_rate", "0.8",
+        "--device", "cpu",
+        "--is_half", "False",
+        "--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}")
 
-'''import subprocess, os
 assets_folder = "assets"
 if not os.path.exists(assets_folder):
     os.makedirs(assets_folder)
@@ -19,28 +45,104 @@ for file, link in files.items():
         try:
             subprocess.run(['wget', link, '-O', file_path], check=True)
         except subprocess.CalledProcessError as e:
-            print(f"Error downloading {file}: {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']
-    file_name, file_extension = os.path.splitext(file.name)
-    if file_extension.lower() in audio_formats:
-        shutil.move(file.name,'audios')
-    elif file_extension.lower().endswith('.pth'):
-        shutil.move(file.name,'assets/weights')
-    elif file_extension.lower().endswith('.index'):
-        shutil.move(file.name,'logs')
+    _, 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:
-        print("Filetype not compatible")
-    return {"choices":show_available('audios'),"__type__": "update"}
+        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"}
+  
 with gr.Blocks() as app:
     with gr.Row():
-        dropbox = gr.File(label="Upload files")
-        audio_picker = gr.Dropdown(label="",choices=show_available('audios'))
-        dropbox.upload(fn=upload_file, inputs=[dropbox],outputs=[audio_picker])
-
-app.launch()  
+        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=[])
+        
+    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])
+        
+app.queue()
+app.launch()