Cambiar interfaz para subir .wav en lugar de URL de YouTube

app.py

@@ -1,34 +1,40 @@
+import os
 import gradio as gr
 from audio_pipeline import (
-    descargar_audio,
     separar_audio_demucs,
     limpiar_stems,
     combinar_stems_sin_vocales,
     reducir_ruido
 )
 
-def generar_base(url: str):
-    # 1) Descarga el audio
-    descargar_audio(url, output_filename="audio")
-    # 2) Separa stems
-    separar_audio_demucs("audio.wav", output_dir="separated")
-    stems_dir = "separated/audio"
-    # 3) Limpia stems
+def procesar_wav(input_wav_path):
+    # 1) Separa stems usando Demucs
+    #    output_dir creará carpeta como separated/<nombre_input>
+    separar_audio_demucs(input_wav_path, output_dir="separated")
+    base_name = os.path.splitext(os.path.basename(input_wav_path))[0]
+    stems_dir = os.path.join("separated", base_name)
+
+    # 2) Limpia cada stem (reducción de ruido)
     limpiar_stems(stems_dir)
-    # 4) Genera base instrumental
+
+    # 3) Combina stems sin vocales
     combinar_stems_sin_vocales(stems_dir)
-    # 5) Reduce ruido
-    resultado = f"{stems_dir}/base_instrumental_clean.wav"
-    reducir_ruido(f"{stems_dir}/base_instrumental.wav", resultado)
-    return resultado
 
+    # 4) Reduce ruido en la base instrumental
+    wav_base = os.path.join(stems_dir, "base_instrumental.wav")
+    wav_clean = os.path.join(stems_dir, "base_instrumental_clean.wav")
+    reducir_ruido(wav_base, wav_clean)
+
+    return wav_clean
+
+# Definimos la interfaz con un input tipo "upload" que nos da la ruta al archivo
 demo = gr.Interface(
-    fn=generar_base,
-    inputs=gr.Text(label="URL de YouTube"),
-    outputs=gr.Audio(label="Base instrumental limpia"),
-    title="Generador de Base Instrumental",
-    description="Pega la URL de un vídeo de YouTube y espera la creación de la instrumental."
+    fn=procesar_wav,
+    inputs=gr.Audio(source="upload", type="filepath", label="Sube un archivo .wav"),
+    outputs=gr.Audio(type="filepath", label="Base instrumental limpia"),
+    title="Procesador de WAV a Base Instrumental",
+    description="Sube tu archivo WAV, separo stems, limpio vocales y reduzco ruido."
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(server_name="0.0.0.0", server_port=7860)