mirror of
https://github.com/Significant-Gravitas/AutoGPT.git
synced 2026-01-09 15:17:59 -05:00
bfbe613960
Additional changes: * Improve typing * Modularize message history memory & fix/refactor lots of things * Fix summarization * Move memory relevance calculation to MemoryItem & improve test * Fix import warnings in web_selenium.py * Remove `memory_add` ghost command * Implement overlap in `split_text` * Move memory tests into subdirectory * Remove deprecated `get_ada_embedding()` and helpers * Fix used token calculation in `chat_with_ai` * Replace Message TypedDict by dataclass * Fix AgentManager singleton issues in tests --------- Co-authored-by: Auto-GPT-Bot <github-bot@agpt.co>
47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
""" Text to speech module """
|
|
import threading
|
|
from threading import Semaphore
|
|
|
|
from autogpt.config.config import Config
|
|
from autogpt.speech.base import VoiceBase
|
|
from autogpt.speech.brian import BrianSpeech
|
|
from autogpt.speech.eleven_labs import ElevenLabsSpeech
|
|
from autogpt.speech.gtts import GTTSVoice
|
|
from autogpt.speech.macos_tts import MacOSTTS
|
|
|
|
_QUEUE_SEMAPHORE = Semaphore(
|
|
1
|
|
) # The amount of sounds to queue before blocking the main thread
|
|
|
|
|
|
def say_text(text: str, voice_index: int = 0) -> None:
|
|
"""Speak the given text using the given voice index"""
|
|
cfg = Config()
|
|
default_voice_engine, voice_engine = _get_voice_engine(cfg)
|
|
|
|
def speak() -> None:
|
|
success = voice_engine.say(text, voice_index)
|
|
if not success:
|
|
default_voice_engine.say(text)
|
|
|
|
_QUEUE_SEMAPHORE.release()
|
|
|
|
_QUEUE_SEMAPHORE.acquire(True)
|
|
thread = threading.Thread(target=speak)
|
|
thread.start()
|
|
|
|
|
|
def _get_voice_engine(config: Config) -> tuple[VoiceBase, VoiceBase]:
|
|
"""Get the voice engine to use for the given configuration"""
|
|
default_voice_engine = GTTSVoice()
|
|
if config.elevenlabs_api_key:
|
|
voice_engine = ElevenLabsSpeech()
|
|
elif config.use_mac_os_tts == "True":
|
|
voice_engine = MacOSTTS()
|
|
elif config.use_brian_tts == "True":
|
|
voice_engine = BrianSpeech()
|
|
else:
|
|
voice_engine = GTTSVoice()
|
|
|
|
return default_voice_engine, voice_engine
|