models/kokoro_model.py

import os
import tempfile
import soundfile as sf

from kokoro import KPipeline

print("Loading Kokoro...")

pipeline = KPipeline(
    lang_code = "f"
)

print("Kokoro Loaded")


def generate(sentence):

    generator = pipeline(
        sentence,
        voice = "ff_siwis"
    )

    output_wav = tempfile.mktemp(
        suffix = ".wav"
    )

    audio_data = []

    for _,_, audio in generator:
        audio_data.extend(audio)

    sf.write(
        output_wav,
        audio_data,
        24000
    )

    return output_wav