|
|
import librosa |
|
|
import gradio as gr |
|
|
import subprocess |
|
|
import os |
|
|
import uuid |
|
|
|
|
|
MAX_DURATION = 60 |
|
|
|
|
|
def clean_audio(audio_path): |
|
|
duration = librosa.get_duration(path=audio_path) |
|
|
if duration > MAX_DURATION: |
|
|
raise gr.Error("❌ Audio too long. Max 60 seconds allowed.") |
|
|
|
|
|
uid = str(uuid.uuid4()) |
|
|
out_dir = f"/tmp/out_{uid}" |
|
|
os.makedirs(out_dir, exist_ok=True) |
|
|
|
|
|
cmd = [ |
|
|
"python", "-m", "demucs", |
|
|
"-n", "htdemucs", |
|
|
audio_path, |
|
|
"-o", out_dir |
|
|
] |
|
|
subprocess.run(cmd, check=True) |
|
|
|
|
|
model_dir = os.path.join(out_dir, "htdemucs") |
|
|
song_dir = os.listdir(model_dir)[0] |
|
|
vocals_path = os.path.join(model_dir, song_dir, "vocals.wav") |
|
|
|
|
|
return vocals_path |
|
|
|
|
|
gr.Interface( |
|
|
fn=clean_audio, |
|
|
inputs=gr.Audio(type="filepath", label="Upload Audio"), |
|
|
outputs=gr.Audio(type="filepath", label="Clean Vocals"), |
|
|
title="AI Voice Cleaner & Vocal Remover", |
|
|
description="Demucs-based vocal separation (HF Spaces compatible)" |
|
|
).launch() |
