import spaces
import gradio as gr
from f5_tts.infer.utils_infer import remove_silence_for_generated_wav
from f5_tts.api import F5TTS
import tempfile
import os
f5tts = F5TTS()


@spaces.GPU
def run_tts(ref_audio, ref_text, gen_text, remove_silence=False):
    output_wav_path = tempfile.mktemp(suffix=".wav")

    wav, sr, _ = f5tts.infer(
        ref_file=ref_audio,
        ref_text=ref_text,
        gen_text=gen_text,
        file_wave=output_wav_path,
        remove_silence=remove_silence,
    )

    return output_wav_path

demo = gr.Interface(
    fn=run_tts,
    inputs=[
        gr.Audio(label="Reference Audio", type="filepath"),
        gr.Textbox(label="Reference Text", placeholder="some call me nature, others call me mother nature."),
        gr.Textbox(label="Generation Text", placeholder="I don't really care what you call me..."),
        gr.Checkbox(label="Remove Silence from Output?", value=False)
    ],
    outputs=gr.Audio(label="Generated Speech"),
    title="🗣️ F5-TTS Demo",
    description="Upload a reference voice, give reference and generation text, and hear it in the same voice!",
)

if __name__ == "__main__":
    demo.launch()