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

import ffmpeg
import tempfile
from openai import OpenAI
import os
import subprocess
import tempfile

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(f"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