BATCH11 0.1.1__py3-none-any.whl

BATCH11/__init__.py

@@ -0,0 +1,3 @@
+from .tts import generate, load_model
+
+__all__ = ["generate", "load_model"]

BATCH11/tts.py

@@ -0,0 +1,100 @@
+from pathlib import Path
+from threading import Lock
+
+_model = None
+_model_device = None
+_model_lock = Lock()
+
+
+def _resolve_device(device):
+    import torch
+
+    if device == "auto":
+        return "cuda" if torch.cuda.is_available() else "cpu"
+
+    if device == "cuda" and not torch.cuda.is_available():
+        raise RuntimeError("CUDA was requested, but no CUDA GPU is available.")
+
+    if device not in ("cpu", "cuda"):
+        raise ValueError("device must be 'auto', 'cpu', or 'cuda'.")
+
+    return device
+
+
+def load_model(device="auto"):
+    """Load the Chatterbox model once and reuse it."""
+
+    global _model, _model_device
+
+    resolved_device = _resolve_device(device)
+
+    with _model_lock:
+        if _model is None or _model_device != resolved_device:
+            from chatterbox.tts import ChatterboxTTS
+
+            _model = ChatterboxTTS.from_pretrained(
+                device=resolved_device
+            )
+            _model_device = resolved_device
+
+    return _model
+
+
+def generate(
+    text,
+    output_path="output.wav",
+    audio_prompt_path=None,
+    device="auto",
+    **generation_kwargs
+):
+    """Generate speech using Chatterbox and save it as a WAV file."""
+
+    if not isinstance(text, str) or not text.strip():
+        raise ValueError("Text must be a non-empty string.")
+
+    output_path = Path(output_path)
+
+    if output_path.suffix.lower() != ".wav":
+        raise ValueError("Output file must have a .wav extension.")
+
+    if audio_prompt_path is not None:
+        audio_prompt_path = Path(audio_prompt_path)
+
+        if not audio_prompt_path.is_file():
+            raise FileNotFoundError(
+                f"Voice reference audio not found: {audio_prompt_path}"
+            )
+
+    model = load_model(device=device)
+
+    kwargs = dict(generation_kwargs)
+
+    if audio_prompt_path is not None:
+        kwargs["audio_prompt_path"] = str(audio_prompt_path)
+
+    audio = model.generate(
+        text=text.strip(),
+        **kwargs
+    )
+
+    import torch
+    import torchaudio
+
+    if not isinstance(audio, torch.Tensor):
+        raise TypeError("Chatterbox returned an unexpected audio format.")
+
+    if audio.ndim == 1:
+        audio = audio.unsqueeze(0)
+
+    if audio.ndim != 2:
+        raise ValueError("Generated audio must be a 1D or 2D tensor.")
+
+    output_path.parent.mkdir(parents=True, exist_ok=True)
+
+    torchaudio.save(
+        str(output_path),
+        audio.detach().cpu(),
+        model.sr
+    )
+
+    return output_path

batch11-0.1.1.dist-info/METADATA

@@ -0,0 +1,33 @@
+Metadata-Version: 2.4
+Name: BATCH11
+Version: 0.1.1
+Summary: Python Text-to-Speech toolkit powered by Chatterbox
+Requires-Python: <3.14,>=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: chatterbox-tts==0.1.7
+Requires-Dist: torchvision==0.21.0
+
+# BATCH11
+
+A Python Text-to-Speech toolkit powered by the Chatterbox engine.
+
+## Installation
+
+    pip install BATCH11
+
+## Basic usage
+
+    import BATCH11
+
+    BATCH11.generate(
+        text="Hello from BATCH11!",
+        output_path="output.wav"
+    )
+
+## Voice cloning
+
+    BATCH11.generate(
+        text="Hello from my reference voice!",
+        audio_prompt_path="reference.wav",
+        output_path="cloned.wav"
+    )

batch11-0.1.1.dist-info/RECORD

@@ -0,0 +1,6 @@
+BATCH11/__init__.py,sha256=hXN7AirSjhbmvaQwHO0zY-PBfrqoqgBnVqO_RpJG7Vc,76
+BATCH11/tts.py,sha256=a4OJ_vukOglG6oYw-KSZjxnfxBvvZ6jy_W3s18ZrFY8,2471
+batch11-0.1.1.dist-info/METADATA,sha256=ayc9IIxDnaU6xqHIhF0wb30zW9BQjU6AxwhW_lSFBfg,678
+batch11-0.1.1.dist-info/WHEEL,sha256=mOdXR9C3wMVov7cSY8PY-XCzo3SmmWCHvGQr5csJoao,91
+batch11-0.1.1.dist-info/top_level.txt,sha256=V6izC4u8jX2xfamqYOTbuC-SZgYV3RDvN4bvbcoOcII,8
+batch11-0.1.1.dist-info/RECORD,,

batch11-0.1.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (84.2.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

batch11-0.1.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+BATCH11