tts.py

import subprocess
import json
from playsound import playsound
import os
import logging
from colorlog import ColoredFormatter

handler = logging.StreamHandler()
handler.setFormatter(ColoredFormatter(
    "%(log_color)s[%(levelname)s] [%(funcName)s] %(message)s",
    log_colors={
        'INFO':     'white'
    }
))

logger = logging.getLogger()
logger.setLevel(logging.INFO) # zmienić INFO na DEBUG żeby tts dawał więcej logów
logger.addHandler(handler)
logger.propagate = False


def tts(text: str, language: str = None, filename: str =None ):
    if not language or not filename:
        if not os.path.exists(text):
            raise ValueError("Missing parameters: either provide 'language' and 'filename', or pass a path to a JSON file.")
        with open(text, 'r', encoding='utf-8') as f:
            params = json.load(f)
            code = tts(**params)
            return code
    model = "./piper/voices/"
    language = language.lower()
    match language:
        case "en":
            model = model + "en_GB-alba-medium.onnx"
        case "pl":
            model = model + "pl_PL-gosia-medium.onnx"
        case _:
            model = model + "en_GB-alba-medium.onnx"

    filename = ".\\"+filename+".wav"

    code = tts_run_model(text, model, filename)

    if code == 0:
        playsound(filename)
    else:
        print("TTS failed. Error code:", code)
    return code



def tts_run_model(text, model, filename):
    try:
        process = subprocess.Popen(
            [
                "piper", "-m", model,
                "-f", filename,"--debug"
            ],
            stdin=subprocess.PIPE,
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE,
            text=True,
            encoding="utf-8"
        )
    except Exception as e:
        logging.exception("TTS failed, unhandled exception:", e)
    stdout, stderr = process.communicate(input=text)

    code = process.returncode

    logging.debug("TTS logs: %s", stderr)
    logging.info("Output: %s", stdout)
    logging.info("Return code: %s", code)

    return code