import gradio as gr
from bark import SAMPLE_RATE, generate_audio, preload_models
from IPython.display import Audio

# Preload models if needed
preload_models()

def create_audio(text):
    audio_array = generate_audio(text)
    return audio_array, SAMPLE_RATE

iface = gr.Interface(
    fn=create_audio,
    inputs=gr.inputs.Textbox(lines=5, placeholder="Digite seu texto aqui..."),
    outputs=[
        gr.outputs.Audio(type="numpy", label="Audio Gerado"),
        gr.outputs.Textbox(label="Taxa de Amostragem")
    ],
    title="Gerador de Áudio com IA",
    description="Digite um texto para gerar áudio usando IA."
)

iface.launch()