Update app.py

app.py

@@ -6,6 +6,16 @@ import random
 import logging
 import time
 import soundfile as sf
+from infer_rvc_python.main import download_manager
+import zipfile
+
+logging.getLogger("infer_rvc_python").setLevel(logging.ERROR)
+
+converter = BaseLoader(only_cpu=False, hubert_path=None, rmvpe_path=None)
+
+title = "<center><strong><font size='7'>RVC⚡ZERO</font></strong></center>"
+description = "This demo is provided for educational and research purposes only. The authors and contributors of this project do not endorse or encourage any misuse or unethical use of this software. Any use of this software for purposes other than those intended is solely at the user's own risk. The authors and contributors shall not be held responsible for any damages or liabilities arising from the use of this demo inappropriately."
+theme = "aliabid94/new-theme"
 
 PITCH_ALGO_OPT = [
     "pm",
@@ -15,13 +25,72 @@ PITCH_ALGO_OPT = [
     "rmvpe+",
 ]
 
-logging.getLogger("infer_rvc_python").setLevel(logging.ERROR)
 
-converter = BaseLoader(only_cpu=False, hubert_path=None, rmvpe_path=None)
+def find_files(directory):
+    file_paths = []
+    for filename in os.listdir(directory):
+        # Check if the file has the desired extension
+        if filename.endswith('.pth') or filename.endswith('.zip') or filename.endswith('.index'):
+            # If yes, add the file path to the list
+            file_paths.append(os.path.join(directory, filename))
 
-title = "<center><strong><font size='7'>RVC⚡ZERO</font></strong></center>"
-description = "This demo is provided for educational and research purposes only. The authors and contributors of this project do not endorse or encourage any misuse or unethical use of this software. Any use of this software for purposes other than those intended is solely at the user's own risk. The authors and contributors shall not be held responsible for any damages or liabilities arising from the use of this demo inappropriately."
-theme = "aliabid94/new-theme"
+    return file_paths
+
+
+def unzip_in_folder(my_zip, my_dir):
+    with zipfile.ZipFile(my_zip) as zip:
+        for zip_info in zip.infolist():
+            if zip_info.is_dir():
+                continue
+            zip_info.filename = os.path.basename(zip_info.filename)
+            zip.extract(zip_info, my_dir)
+
+
+def find_my_model(a_, b_):
+
+    if a_ is None or a_.endswith(".pth"):
+        return a_, b_
+
+    txt_files = []
+    for base_file in [a_, b_]:
+        if base_file is not None and base_file.endswith(".txt"):
+            txt_files.append(base_file)
+    
+    directory = os.path.dirname(a_)
+    
+    for txt in txt_files:
+        with open(txt, 'r') as file:
+            first_line = file.readline()
+    
+        download_manager(
+            url=first_line.strip(),
+            path=directory,
+            extension="",
+        )
+    
+    for f in find_files(directory):
+        if f.endswith(".zip"):
+            unzip_in_folder(f, directory)
+    
+    model = None
+    index = None
+    end_files = find_files(directory)
+    
+    for ff in end_files:
+        if ff.endswith(".pth"):
+            model = os.path.join(directory, ff)
+            gr.Info(f"Model found: {ff}")
+        if ff.endswith(".index"):
+            index = os.path.join(directory, ff)
+            gr.Info(f"Index found: {ff}")
+
+    if not model:
+        gr.Error(f"Model not found in: {end_files}")
+    
+    if not index:
+        gr.Warning("Index not found")
+    
+    return model, index
 
 
 @spaces.GPU()
@@ -51,6 +120,10 @@ def run(
     if isinstance(audio_files, str):
         audio_files = [audio_files]
 
+    if file_m is not None and file_m.endswith(".txt"):
+        file_m, file_index = find_my_model(file_m, file_index)
+        print(file_m, file_index)
+
     random_tag = "USER_"+str(random.randint(10000000, 99999999))
 
     converter.apply_conf(
@@ -221,6 +294,29 @@ def get_gui(theme):
                     0.25,
                     0.50,
                 ],
+                [
+                    ["./example2/test2.ogg"],
+                    "./example2/model_link.txt",
+                    "rmvpe+",
+                    0,
+                    "./example2/index_link.txt",
+                    0.75,
+                    3,
+                    0.25,
+                    0.50,
+                ],
+                [
+                    ["./example3/test3.wav"],
+                    "./example3/zip_link.txt",
+                    "rmvpe+",
+                    0,
+                    None,
+                    0.75,
+                    3,
+                    0.25,
+                    0.50,
+                ],
+                
             ],
             fn=run,
             inputs=[