output device for tts

Author: dborodin836

main.py

@@ -9,7 +9,7 @@
 
 from config import config
 from modules.bot_state import state_manager
-from modules.chat import parse_console_logs_and_build_conversation_history
+from modules.setup import parse_console_logs_and_build_conversation_history
 from modules.commands.gui.openai import gpt3_cmd_handler
 from modules.gui.log_window import (
     CopyStdoutToSocket,

modules/builder/loaders.py

@@ -151,7 +151,8 @@ def __load_rcon_command(self):
             "model",
             "voice",
             "speed",
-            "volume"
+            "volume",
+            "output_device"
         }
     )
 }

modules/commands/base.py

@@ -42,7 +42,7 @@ def func(logline: LogLine, shared_dict: InitializerConfig) -> Optional[str]:
             msg = remove_args(logline.prompt)
             result = get_tts(msg, cls.settings)
 
-            pygame.mixer.init()
+            pygame.mixer.init(devicename=cls.settings.get("output_device", None))
             sound = pygame.mixer.Sound(io.BytesIO(result.content))
             sound.set_volume(cls.settings.get('volume', 1.0))
             sound.play()

modules/commands/gui/audio.py

@@ -0,0 +1,13 @@
+from modules.command_controllers import InitializerConfig
+from modules.utils.audio import get_devices
+from modules.logs import get_logger
+
+gui_logger = get_logger("gui")
+
+
+def handle_list_devices(command: str, shared_dict: InitializerConfig):
+    gui_logger.info("Available devices:")
+    gui_logger.info("#" * 10)
+    for device in get_devices():
+        gui_logger.info(f"- {device}")
+    gui_logger.info("#" * 10)

modules/commands/gui/invoke.py

@@ -1,5 +1,5 @@
 from config import config
-from modules.chat import controller
+from modules.setup import controller
 from modules.lobby_manager import lobby_manager
 from modules.logs import get_logger
 from modules.typing import LogLine

modules/gui/controller.py

@@ -1,6 +1,7 @@
 import sys
 
 from modules.command_controllers import GuiCommandController
+from modules.commands.gui.audio import handle_list_devices
 from modules.commands.gui.bans import handle_ban, handle_list_bans, handle_unban
 from modules.commands.gui.config import handle_config
 from modules.commands.gui.invoke import invoke
@@ -29,3 +30,4 @@
 )
 command_controller.register_command("quit", lambda *args: sys.exit(0), "Quit the program.")
 command_controller.register_command("config", handle_config, config_command_description)
+command_controller.register_command("audio-devices", handle_list_devices, "List output audio devices.")

modules/server.py

@@ -12,7 +12,7 @@
 
 from config import Config, ValidatableConfig, config
 from modules.builder.utils import create_command_from_dict
-from modules.chat import controller
+from modules.setup import controller
 from modules.gui.controller import command_controller as gui_cmd_controller
 from modules.logs import get_logger
 from modules.set_once_dict import ModificationOfSetKey

modules/chat.py modules/setup.py

@@ -22,7 +22,7 @@
 controller = CommandController(InitializerConfig())
 
 
-def setup() -> None:
+def init() -> None:
     """
     Initializes the program.
     """
@@ -49,7 +49,7 @@ def parse_console_logs_and_build_conversation_history() -> None:
     """
     Processes the console logs and builds a conversation history, filters banned usernames.
     """
-    setup()
+    init()
 
     # Commands
     controller.register_command("!gh", handle_gh_command)

modules/utils/audio.py

@@ -0,0 +1,14 @@
+from typing import Tuple
+
+import pygame
+import pygame._sdl2.audio as sdl2_audio
+
+
+def get_devices(capture_devices: bool = False) -> Tuple[str, ...]:
+    init_by_me = not pygame.mixer.get_init()
+    if init_by_me:
+        pygame.mixer.init()
+    devices = tuple(sdl2_audio.get_audio_device_names(capture_devices))
+    if init_by_me:
+        pygame.mixer.quit()
+    return devices

schemas/commands/openai-tts.schema.json

@@ -24,6 +24,11 @@
         "volume": {
           "title": "Volume (0.00 - 1.00)",
           "type": "number"
+        },
+        "output_device": {
+          "type": "string",
+          "title": "Output device",
+          "description": "Use 'audio-devices' command to get all available devices."
         }
       },
       "required": [