import librosa
import gradio as gr
import subprocess
import os
import uuid

MAX_DURATION = 60  # seconds

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()