wyoming-piper 1.5.3__py3-none-any.whl → 2.1.2__py3-none-any.whl

wyoming_piper/process.py

@@ -1,171 +0,0 @@
-#!/usr/bin/env python3
-import argparse
-import asyncio
-import json
-import logging
-import tempfile
-import time
-from dataclasses import dataclass
-from typing import Any, Dict, Optional
-
-from .download import ensure_voice_exists, find_voice
-
-_LOGGER = logging.getLogger(__name__)
-
-
-@dataclass
-class PiperProcess:
-    """Info for a running Piper process (one voice)."""
-
-    name: str
-    proc: "asyncio.subprocess.Process"
-    config: Dict[str, Any]
-    wav_dir: tempfile.TemporaryDirectory
-    last_used: int = 0
-
-    def get_speaker_id(self, speaker: str) -> Optional[int]:
-        """Get speaker by name or id."""
-        return _get_speaker_id(self.config, speaker)
-
-    @property
-    def is_multispeaker(self) -> bool:
-        """True if model has more than one speaker."""
-        return _is_multispeaker(self.config)
-
-
-def _get_speaker_id(config: Dict[str, Any], speaker: str) -> Optional[int]:
-    """Get speaker by name or id."""
-    speaker_id_map = config.get("speaker_id_map", {})
-    speaker_id = speaker_id_map.get(speaker)
-    if speaker_id is None:
-        try:
-            # Try to interpret as an id
-            speaker_id = int(speaker)
-        except ValueError:
-            pass
-
-    return speaker_id
-
-
-def _is_multispeaker(config: Dict[str, Any]) -> bool:
-    """True if model has more than one speaker."""
-    return config.get("num_speakers", 1) > 1
-
-
-# -----------------------------------------------------------------------------
-
-
-class PiperProcessManager:
-    """Manager of running Piper processes."""
-
-    def __init__(self, args: argparse.Namespace, voices_info: Dict[str, Any]):
-        self.voices_info = voices_info
-        self.args = args
-        self.processes: Dict[str, PiperProcess] = {}
-        self.processes_lock = asyncio.Lock()
-
-    async def get_process(self, voice_name: Optional[str] = None) -> PiperProcess:
-        """Get a running Piper process or start a new one if necessary."""
-        voice_speaker: Optional[str] = None
-        if voice_name is None:
-            # Default voice
-            voice_name = self.args.voice
-
-        if voice_name == self.args.voice:
-            # Default speaker
-            voice_speaker = self.args.speaker
-
-        assert voice_name is not None
-
-        # Resolve alias
-        voice_info = self.voices_info.get(voice_name, {})
-        voice_name = voice_info.get("key", voice_name)
-        assert voice_name is not None
-
-        piper_proc = self.processes.get(voice_name)
-        if (piper_proc is None) or (piper_proc.proc.returncode is not None):
-            # Remove if stopped
-            self.processes.pop(voice_name, None)
-
-            # Start new Piper process
-            if self.args.max_piper_procs > 0:
-                # Restrict number of running processes
-                while len(self.processes) >= self.args.max_piper_procs:
-                    # Stop least recently used process
-                    lru_proc_name, lru_proc = sorted(
-                        self.processes.items(), key=lambda kv: kv[1].last_used
-                    )[0]
-                    _LOGGER.debug("Stopping process for: %s", lru_proc_name)
-                    self.processes.pop(lru_proc_name, None)
-                    if lru_proc.proc.returncode is None:
-                        try:
-                            lru_proc.proc.terminate()
-                            await lru_proc.proc.wait()
-                        except Exception:
-                            _LOGGER.exception("Unexpected error stopping piper process")
-
-            _LOGGER.debug(
-                "Starting process for: %s (%s/%s)",
-                voice_name,
-                len(self.processes) + 1,
-                self.args.max_piper_procs,
-            )
-
-            ensure_voice_exists(
-                voice_name,
-                self.args.data_dir,
-                self.args.download_dir,
-                self.voices_info,
-            )
-
-            onnx_path, config_path = find_voice(voice_name, self.args.data_dir)
-            with open(config_path, "r", encoding="utf-8") as config_file:
-                config = json.load(config_file)
-
-            wav_dir = tempfile.TemporaryDirectory()
-            piper_args = [
-                "--model",
-                str(onnx_path),
-                "--config",
-                str(config_path),
-                "--output_dir",
-                str(wav_dir.name),
-                "--json-input",  # piper 1.1+
-            ]
-
-            if voice_speaker is not None:
-                if _is_multispeaker(config):
-                    speaker_id = _get_speaker_id(config, voice_speaker)
-                    if speaker_id is not None:
-                        piper_args.extend(["--speaker", str(speaker_id)])
-
-            if self.args.noise_scale:
-                piper_args.extend(["--noise-scale", str(self.args.noise_scale)])
-
-            if self.args.length_scale:
-                piper_args.extend(["--length-scale", str(self.args.length_scale)])
-
-            if self.args.noise_w:
-                piper_args.extend(["--noise-w", str(self.args.noise_w)])
-
-            _LOGGER.debug(
-                "Starting piper process: %s args=%s", self.args.piper, piper_args
-            )
-            piper_proc = PiperProcess(
-                name=voice_name,
-                proc=await asyncio.create_subprocess_exec(
-                    self.args.piper,
-                    *piper_args,
-                    stdin=asyncio.subprocess.PIPE,
-                    stdout=asyncio.subprocess.PIPE,
-                    stderr=asyncio.subprocess.DEVNULL,
-                ),
-                config=config,
-                wav_dir=wav_dir,
-            )
-            self.processes[voice_name] = piper_proc
-
-        # Update used
-        piper_proc.last_used = time.monotonic_ns()
-
-        return piper_proc

wyoming_piper-1.5.3.dist-info/RECORD

@@ -1,14 +0,0 @@
-wyoming_piper/__init__.py,sha256=jq4RWRoEFQNm73oe9FNeVxdHTbu-MdElAnW1fbodESo,138
-wyoming_piper/__main__.py,sha256=knqXEoX_oxJr69oKitAILZKioKsn14HSrTpa3lCT-nA,7186
-wyoming_piper/const.py,sha256=04sCdtJ2QGuF1BQGkOuQW10og61PgH3fCnPhaYu-YoU,1015
-wyoming_piper/download.py,sha256=n3oKXN0fLk7Dn7x4wnR17b3QzXilFgF-QSxtN0QE10s,6185
-wyoming_piper/file_hash.py,sha256=HMuwrgEIg-bCOXHG0wE3vtjrqGD7QaA_UNfvBMXeUcY,1107
-wyoming_piper/handler.py,sha256=0bcllkoR6iNsseWQ1Y-Uj4sMV_az6mn4ivS8AAo4RHw,4936
-wyoming_piper/process.py,sha256=L_qqxQcQawrC940fwlv4u6KM9KjCq6N6ym-OADSZcrM,5794
-wyoming_piper/voices.json,sha256=3uEoyiBYbQlJnFYUEKnFHYK0Wq40sbaHiD9AsgRWKzc,202976
-wyoming_piper-1.5.3.dist-info/licenses/LICENSE.md,sha256=E3RtUJ105V6iJl--8gS7fNv4SoMVsCB-mIMmy1Q4cCg,1071
-wyoming_piper-1.5.3.dist-info/METADATA,sha256=9fYLRK0GOJCnaf9FBwlV1zoYUoDtyb0h66tOilKf2YU,2463
-wyoming_piper-1.5.3.dist-info/WHEEL,sha256=A8Eltl-h0W-qZDVezsLjjslosEH_pdYC2lQ0JcbgCzs,91
-wyoming_piper-1.5.3.dist-info/entry_points.txt,sha256=n2UgsOCQitQ5Itr20aITTWZLL2dAtaVKn5pdecXdDHE,61
-wyoming_piper-1.5.3.dist-info/top_level.txt,sha256=t7U7-u1sK_4xy_qbTJhxQRbxle3cLQfPq2oVLezHVNU,14
-wyoming_piper-1.5.3.dist-info/RECORD,,