import gradio as gr
from huggingface_hub import snapshot_download
import os
import spaces
import torch
import torchaudio

print(f"torch: {torch.__version__}")
print(f"torachaudio: {torchaudio.__version__}")

huggingface_token = os.getenv("hf_token")

snapshot_download(
    repo_id="pollyai/denoiser", 
    local_dir="model",
    token=huggingface_token, # type a new token-id.
)
from model.spaces_api import process as api_process

@spaces.GPU
def process(path, model_config):
    return api_process(path, model_config)

# Create interface with pitch shift options
demo = gr.Interface(
    fn=process,
    inputs=[
        gr.Audio(type='filepath', label='input'),
        gr.Radio(["Denoise", "Noise Extract"], label="Choose mode", value = "Denoise"),
        
    ],
    outputs=[gr.Audio(show_share_button=False, label='output')],
    allow_flagging="never",
    clear_btn = None
)

demo.launch()