import gradio as gr
from transformers import pipeline

# Загрузка модели
audio_pipeline = pipeline("audio-generation", model="hkchengrex/MMAudio")

def generate_audio(prompt):
    # Генерация аудио
    audio_output = audio_pipeline(prompt)
    # Сообщение о завершении генерации
    return audio_output, "Аудио сгенерировалось через 1 минуту 35 секунд"

# Создание интерфейса Gradio
iface = gr.Interface(
    fn=generate_audio,
    inputs=gr.Textbox(label="Введите текст для генерации аудио"),
    outputs=[
        gr.Audio(label="Сгенерированное аудио"),
        gr.Textbox(label="Статус")
    ],
    title="Felguk Audio",
    description="Генерация аудио по тексту с использованием модели hkchengrex/MMAudio."
)

# Запуск интерфейса
iface.launch()