Update app.py

app.py

@@ -92,9 +92,11 @@ def download_audio(url):
 
 
 
-# Define a function to handle the entire separation process
+
+# Define the audio separation function
 def separate_audio(input_audio, model_voc_inst, model_deecho, model_back_voc):
-    output_dir = audio_separat_dir
+    output_dir = audio_separat_dir 
+    os.makedirs(output_dir, exist_ok=True)
     separator = Separator(output_dir=output_dir)
 
     # Define output files
@@ -123,7 +125,18 @@ def separate_audio(input_audio, model_voc_inst, model_deecho, model_back_voc):
     os.rename(os.path.join(output_dir, backing_voc[0]), backing_vocals)  # Rename to “Backing Vocals.wav”
     os.rename(os.path.join(output_dir, backing_voc[1]), lead_vocals)     # Rename to “Lead Vocals.wav”
 
-    return "separation done..."
+    return [
+        instrumental, vocals, vocals_no_reverb, vocals_reverb, lead_vocals, backing_vocals
+    ]
+
+# Gradio Interface
+def gradio_interface(input_audio, model_voc_inst, model_deecho, model_back_voc):
+    # Separate audio and get paths to output files
+    results = separate_audio(input_audio, model_voc_inst, model_deecho, model_back_voc)
+    
+    # Convert file paths to Gradio downloadable links
+    return [gr.File(file) for file in results]
+
 
 # Main function to process audio (Inference)
 def process_audio(MODEL_NAME, SOUND_PATH, F0_CHANGE, F0_METHOD, MIN_PITCH, MAX_PITCH, CREPE_HOP_LENGTH, INDEX_RATE, 
@@ -282,6 +295,7 @@ with gr.Blocks(title="Hex RVC", theme=gr.themes.Base(primary_hue="red", secondar
         with gr.Row():
             input_audio = gr.Audio(type="filepath", label="Upload Audio File")
             
+            
         with gr.Row():
             with gr.Accordion("Separation by Link", open = False):
                 with gr.Row():
@@ -305,16 +319,16 @@ with gr.Blocks(title="Hex RVC", theme=gr.themes.Base(primary_hue="red", secondar
             model_deecho = gr.Textbox(value='UVR-DeEcho-DeReverb.pth', label="DeEcho-DeReverb Model", visible=False)
             model_back_voc = gr.Textbox(value='mel_band_roformer_karaoke_aufr33_viperx_sdr_10.1956.ckpt', label="Backing Vocals Model", visible=False)
         
-        separate_button = gr.Button("Separate Audio")
-        
+        submit_button = gr.Button("Separate Audio")
+    
         with gr.Row():
-            outout_paht = gr.Textbox(label="output download", interactive=False)
-      
-        separate_button.click(
-            separate_audio,
-            inputs=[input_audio, model_voc_inst, model_deecho, model_back_voc],
-            outputs=[outout_paht]
-        )
+            output_files = gr.File(label="Download Separated Files", multiple=True)
+    
+        submit_button.click(
+        fn=gradio_interface, 
+        inputs=[input_audio, model_voc_inst, model_deecho, model_back_voc],
+        outputs=output_files
+    )
 
 
 # Launch the Gradio app