Update app.py

app.py

@@ -27,27 +27,30 @@ main_dir = Path().resolve()
 print(main_dir)
 
 os.chdir(main_dir)
-models_dir = "rvc_models"
+models_dir = main_dir / "rvc_models"
 audio_separat_dir = main_dir / "audio_input"
 
-rvc_models_dir = os.path.join(models_dir)
-
 AUDIO_DIR = main_dir / 'audio_input'
 
 
+# Function to list all folders in the models directory
+def get_folders():
+    if models_dir.exists() and models_dir.is_dir():
+        return [folder.name for folder in models_dir.iterdir() if folder.is_dir()]
+    return []
+
+# Function to refresh and return the list of folders
+def refresh_folders():
+    return gr.Dropdown.update(choices=get_folders())
 
 
 
-def get_current_models(models_dir):
-    models_list = os.listdir(models_dir)
-    items_to_remove = ['hubert_base.pt', 'readme.txt', 'rmvpe.pt']
-    return [item for item in models_list if item not in items_to_remove]
 
 
 
 # Function to get the list of audio files in the specified directory
 def get_audio_files():
-    if not os.path.exists(AUDIO_DIR):
+    if not os.path.exists(QUDIO_DIR):
         os.makedirs(AUDIO_DIR)
     # List all supported audio file formats
     return [f for f in os.listdir(AUDIO_DIR) if f.lower().endswith(('.mp3', '.wav', '.flac', '.ogg', '.aac'))]
@@ -70,13 +73,6 @@ def play_audio(file_path):
 
 
 
-def update_models_list():
-    models_l = get_current_models(rvc_models_dir)
-    dropdown_instance = gr.Dropdown(choices=models_l)
-    return dropdown_instance
-
-
-voice_models = get_current_models(rvc_models_dir)
 
 
 def download_audio(url):
@@ -261,8 +257,16 @@ with gr.Blocks(title="Hex RVC", theme=gr.themes.Default(primary_hue="red", secon
     gr.Markdown(" join [AIHub](https://discord.gg/aihub) to get the rvc model!")
     
     with gr.Tab("Inference"):
-        MODEL_NAME = gr.Dropdown(voice_models, label='Voice Models', info='Models folder "AICoverGen --> rvc_models". After new models are added into this folder, click the refresh button')
-        ref_btn = gr.Button('Refresh Models 🔁', variant='primary')
+        MODEL_NAME = gr.Dropdown(
+            label="Select a Model Folder",
+            choices=get_folders(),
+            interactive=True,
+            elem_id="model_folder"
+        )
+        # Button to refresh the list of folders
+        refresh_button = gr.Button("Refresh Folder List")
+        # Action to refresh folder list on button click
+        refresh_button.click(refresh_folders, outputs=folder_dropdown)
         with gr.Row():
         #     = gr.Textbox(label="Model Name", placeholder="Enter model name")
         #    SOUND_PATH = gr.Textbox(label="Audio Path (Optional)", placeholder="Leave blank to upload audio")
@@ -327,9 +331,12 @@ with gr.Blocks(title="Hex RVC", theme=gr.themes.Default(primary_hue="red", secon
         )
 
     with gr.Tab("Download RVC Model"):
-        url = gr.Textbox(label="Your model URL")
-        dirname = gr.Textbox(label="Your Model name")
+        with gr.Row():
+            url = gr.Textbox(label="Your model URL")
+            dirname = gr.Textbox(label="Your Model name")
+        outout_pah = gr.Textbox(label="output download", interactive=False)
         button_model = gr.Button("Download model")
+        
         button_model.click(fn=download_online_model, inputs=[url, dirname], outputs=[dirname])
     with gr.Tab("Audio Separation"):
         with gr.Row():