models/manager.py

from models.omnivoice_model import generate as generate_omnivoice
from models.kokoro_model import generate as generate_kokoro
from models.piper_model import generate as generate_piper

def generate_audio(model_name, sentence):

    if model_name == "OmniVoice":
        return generate_omnivoice(sentence)
    
    elif model_name == "Piper":
        return generate_piper(sentence)

    elif model_name == "Kokoro":
        return generate_kokoro(sentence)
    
    raise ValueError(
        f"Unknown model: {model_name}"
    )