import gradio as gr
import tempfile
from ttsmms import TTS
import soundfile as sf

tts = TTS("data/ory")

def generate_audio(text):
    wav = tts.synthesis(text)
    audio_file = tempfile.NamedTemporaryFile(suffix=".wav", delete=False)
    audio_path = audio_file.name
    sf.write(audio_path, wav["x"], wav["sampling_rate"])
    return audio_path

inputs = gr.Textbox(label="Input", max_lines=3)
outputs = gr.Audio(label="Output")

title = "Text-to-Speech App"
description = "Enter your text and generate audio."

gr.Interface(fn=generate_audio, inputs=inputs, outputs=outputs, title=title, description=description).launch(share=True)