Update app.py

app.py

@@ -1,11 +1,29 @@
 import gradio as gr
 import os, sys
 from colorama import Fore
-
+from rvcsynch import download_from_url
 now_dir = os.getcwd()
 sys.path.append(now_dir)
 
 
+def show_available(filepath,format=None):
+    if format:
+        print(f"Format: {format}")
+        files = []
+        for file in os.listdir(filepath):
+            if file.endswith(format):
+                print(f"Matches format: {file}")
+                files.append(file)
+            else:
+                print(f"Does not match format: {file}")
+        print(f"Matches: {files}")
+        if len(files) < 1:
+            return ['']
+        return files
+    if len(os.listdir(filepath)) < 1:
+        return ['']
+    return os.listdir(filepath)
+
 # Function to detect the .pth and .index files
 def detect_files(model_name):
     model_dir = f"{now_dir}/assets/weights/{model_name}"
@@ -72,30 +90,41 @@ def process_audio(model_name, pitch, input_path, f0_method, save_as, index_rate,
         return "Error in processing audio.", None
 
 # Gradio interface
-with gr.Blocks() as demo:
-    gr.Markdown("# 🔊 **LISTEN TO YOUR MODEL**")
-
-    model_name = gr.Textbox(label="Model Name", value="Ren")
-    pitch = gr.Slider(minimum=-12, maximum=12, step=1, label="Pitch", value=0)
-    input_path = gr.Dropdown(label="",choices=show_available('audios'),value='',interactive=True)
-    f0_method = gr.Radio(choices=["rmvpe", "pm", "crepe"], label="F0 Method", value="rmvpe")
-    save_as = gr.Textbox(label="Save As", value="/content/RVC/audios/cli_output.wav")
-    index_rate = gr.Slider(minimum=0, maximum=1, step=0.01, label="Index Rate", value=0.5)
-    volume_normalization = gr.Slider(minimum=0, maximum=1, step=0.01, label="Volume Normalization", value=0)
-    consonant_protection = gr.Slider(minimum=0, maximum=1, step=0.01, label="Consonant Protection", value=0.5)
-
-    output_text = gr.Textbox(label="Output")
-    output_audio = gr.Audio(label="Processed Audio")
-
-    # Button to detect files
-    detect_btn = gr.Button("Detect Files")
-    detect_btn.click(fn=detect_files, inputs=[model_name], outputs=output_text)
-
-    # Button to process the audio and return audio output
-    submit_btn = gr.Button("Submit")
-    submit_btn.click(fn=process_audio, 
-                     inputs=[model_name, pitch, input_path, f0_method, save_as, index_rate, volume_normalization, consonant_protection], 
-                     outputs=[output_text, output_audio])
-
+with gr.Blocks(theme="gradio/soft") as demo:
+    gr.Markdown("# 🔊 ** RVC GUI**")
+
+    with gr.Tabs():
+        model_name = gr.Textbox(label="Model Name", value="Ren")
+        pitch = gr.Slider(minimum=-12, maximum=12, step=1, label="Pitch", value=0)
+        with gr.Tab("Infernece"):
+            input_path = gr.Dropdown(label="",choices=show_available('audios'),value='',interactive=True)
+            f0_method = gr.Radio(choices=["rmvpe", "pm", "crepe"], label="F0 Method", value="rmvpe")
+            save_as = gr.Textbox(label="Save As", value="/content/RVC/audios/cli_output.wav")
+            index_rate = gr.Slider(minimum=0, maximum=1, step=0.01, label="Index Rate", value=0.5)
+            volume_normalization = gr.Slider(minimum=0, maximum=1, step=0.01, label="Volume Normalization", value=0)
+            consonant_protection = gr.Slider(minimum=0, maximum=1, step=0.01, label="Consonant Protection", value=0.5)
+
+        output_text = gr.Textbox(label="Output")
+        output_audio = gr.Audio(label="Processed Audio")
+        # Button to detect files
+        detect_btn = gr.Button("Detect Files")
+        detect_btn.click(fn=detect_files, inputs=[model_name], outputs=output_text)
+        # Button to process the audio and return audio output
+        submit_btn = gr.Button("Submit")
+        submit_btn.click(fn=process_audio, 
+                         inputs=[model_name, pitch, input_path, f0_method, save_as, index_rate, volume_normalization, consonant_protection], 
+                         outputs=[output_text, output_audio])
+        
+    with gr.Tabs():
+        with gr.Tab("Download a model"):
+
+            url = gr.Texbox(label="url models...")
+            url_name = gr.Texbox(label="models names...")
+            download_fm = gr.Button("download model")
+            download_fm.click(fn=download_from_url, inputs=[url, url_name], outputs=url_name)
+            
+
+    
+        
 # Launch the app
 demo.launch()