from kokoro import KPipeline
import numpy as np


# Chargement unique du modèle
pipeline = KPipeline(lang_code="f")

VOICE = "ff_siwis"
SAMPLE_RATE = 24000


def text_to_audio(text):
    """
    Convertit un texte en audio.
    Retourne un tuple compatible avec gr.Audio
    """

    generator = pipeline(
        text,
        voice=VOICE
    )

    chunks = []

    for _, _, audio in generator:
        chunks.append(audio)

    audio = np.concatenate(chunks)

    return SAMPLE_RATE, audio