|
|
|
try: |
|
import dotenv |
|
dotenv.load_dotenv() |
|
except ImportError: |
|
pass |
|
|
|
import argparse |
|
import os |
|
import pysbd |
|
import queue |
|
import sys |
|
import tempfile |
|
import threading |
|
import shutil |
|
import sys |
|
import tempfile |
|
import contextlib |
|
|
|
import openai |
|
|
|
try: |
|
from playsound import playsound |
|
except ImportError: |
|
print("Error: missing required package 'playsound'. !pip install playsound") |
|
sys.exit(1) |
|
|
|
@contextlib.contextmanager |
|
def tempdir(): |
|
path = tempfile.mkdtemp() |
|
try: |
|
yield path |
|
finally: |
|
try: |
|
shutil.rmtree(path) |
|
except IOError: |
|
sys.stderr.write('Failed to clean up temp dir {}'.format(path)) |
|
|
|
class SimpleAudioPlayer: |
|
def __init__(self): |
|
self._queue = queue.Queue() |
|
self.running = True |
|
self._thread = threading.Thread(target=self.__play_audio_loop, daemon=True) |
|
self._thread.start() |
|
|
|
def put(self, file): |
|
self._queue.put(file) |
|
|
|
def stop(self): |
|
self.running = False |
|
self._thread.join() |
|
try: |
|
while True: |
|
file = self._queue.get_nowait() |
|
if os.path.exists(file): |
|
os.unlink(file) |
|
except queue.Empty as e: |
|
pass |
|
|
|
def __play_audio_loop(self): |
|
while self.running: |
|
try: |
|
while True: |
|
file = self._queue.get(block=True, timeout=0.01) |
|
|
|
try: |
|
playsound(file) |
|
finally: |
|
os.unlink(file) |
|
|
|
except queue.Empty as e: |
|
continue |
|
|
|
class OpenAI_tts: |
|
def __init__(self, model, voice, speed, base_dir): |
|
self.base_dir = base_dir |
|
self.openai_client = openai.OpenAI( |
|
|
|
|
|
api_key = os.environ.get("OPENAI_API_KEY", "sk-ip"), |
|
base_url = os.environ.get("OPENAI_BASE_URL", "http://localhost:8000/v1"), |
|
) |
|
|
|
self.params = { |
|
'model': model, |
|
'voice': voice, |
|
'speed': speed |
|
} |
|
|
|
def speech_to_file(self, text: str) -> None: |
|
with self.openai_client.audio.speech.with_streaming_response.create( |
|
input=text, response_format='opus', **self.params |
|
) as response: |
|
tf, output_filename = tempfile.mkstemp(suffix='.wav', prefix="audio_reader_", dir=self.base_dir) |
|
response.stream_to_file(output_filename) |
|
return output_filename |
|
|
|
|
|
if __name__ == "__main__": |
|
parser = argparse.ArgumentParser( |
|
description='Text to speech player', |
|
formatter_class=argparse.ArgumentDefaultsHelpFormatter) |
|
|
|
parser.add_argument('-m', '--model', action='store', default="tts-1", help="The OpenAI model") |
|
parser.add_argument('-v', '--voice', action='store', default="alloy", help="The voice to use") |
|
parser.add_argument('-s', '--speed', action='store', default=1.0, help="How fast to read the audio") |
|
|
|
args = parser.parse_args() |
|
|
|
try: |
|
with tempdir() as base_dir: |
|
player = SimpleAudioPlayer() |
|
reader = OpenAI_tts(voice=args.voice, model=args.model, speed=args.speed, base_dir=base_dir) |
|
seg = pysbd.Segmenter(language='en', clean=True) |
|
|
|
for raw_line in sys.stdin: |
|
for line in seg.segment(raw_line): |
|
if not line: |
|
continue |
|
|
|
print(line) |
|
player.put(reader.speech_to_file(line)) |
|
|
|
player.stop() |
|
|
|
except KeyboardInterrupt: |
|
pass |
|
|
