01/software/source/server/services/tts/openai/tts.py

import ffmpeg
import tempfile
from openai import OpenAI
import os

from source.server.utils.logs import logger
from source.server.utils.logs import setup_logging

setup_logging()

# If this TTS service is used, the OPENAI_API_KEY environment variable must be set
if not os.getenv("OPENAI_API_KEY"):
    logger.error("")
    logger.error(
        "OpenAI API key not found. Please set the OPENAI_API_KEY environment variable, or run 01 with the --local option."
    )
    logger.error("Aborting...")
    logger.error("")
    os._exit(1)

client = OpenAI()


class Tts:
    def __init__(self, config):
        pass

    def tts(self, text):
        response = client.audio.speech.create(
            model="tts-1",
            voice=os.getenv("OPENAI_VOICE_NAME", "alloy"),
            input=text,
            response_format="opus",
        )
        with tempfile.NamedTemporaryFile(suffix=".opus", delete=False) as temp_file:
            response.stream_to_file(temp_file.name)

            # TODO: hack to format audio correctly for device
            outfile = tempfile.gettempdir() + "/" + "raw.dat"
            ffmpeg.input(temp_file.name).output(
                outfile, f="s16le", ar="16000", ac="1", loglevel="panic"
            ).run()

            return outfile