import torch
import torchaudio

if torch.cuda.is_available():
    device = "cuda"
else:
    device = "cpu"
from resemble_enhance.enhancer.inference import denoise, enhance, download

if path is None:
    return None, None

solver = solver.lower()
nfe = int(nfe)
lambd = 0.9 if denoising else 0.1

dwav, sr = torchaudio.load(path)
dwav = dwav.mean(dim=0)

download(RUN_DIR, safetensors=True)
wav1, new_sr = denoise(dwav, sr, device, run_dir=RUN_DIR / "denoiser")
wav2, new_sr = enhance(
    dwav, sr, device, nfe=nfe, solver=solver, lambd=lambd, tau=tau, run_dir=RUN_DIR / "enhancer", skip_download=True,
)

wav1 = wav1.cpu().numpy()
wav2 = wav2.cpu().numpy()

return (new_sr, wav1), (new_sr, wav2)