toolnet-tts 0.2.0__py3-none-any.whl

toolnet/__init__.py

@@ -0,0 +1,3 @@
+from .engine import TTS
+
+__all__ = ["TTS"]

toolnet/engine.py

@@ -0,0 +1,11 @@
+class TTS:
+    def __init__(self, voice="giong1", device="auto"):
+        from vieneu import Vieneu
+
+        self.voice = "Thiện Minh" if voice == "giong1" else voice
+        self.engine = Vieneu(mode="v3turbo", device=device)
+
+    def say(self, text, path="out.wav"):
+        audio = self.engine.infer(text, voice=self.voice)
+        self.engine.save(audio, str(path))
+        return path

toolnet/text.py

@@ -0,0 +1,50 @@
+import re
+
+_D = ["không", "một", "hai", "ba", "bốn", "năm", "sáu", "bảy", "tám", "chín"]
+_ABBR = {"TP.HCM": "thành phố Hồ Chí Minh", "UBND": "uỷ ban nhân dân", "VND": "đồng"}
+
+
+def _read_3(n, full=False):
+    h, t, u = n // 100, (n % 100) // 10, n % 10
+    out = []
+    if h or full:
+        out += [_D[h], "trăm"]
+    if t > 1:
+        out += [_D[t], "mươi"]
+    elif t == 1:
+        out.append("mười")
+    elif (h or full) and u:
+        out.append("lẻ")
+    if u:
+        if t > 1 and u == 1:
+            out.append("mốt")
+        elif t >= 1 and u == 5:
+            out.append("lăm")
+        elif t > 1 and u == 4:
+            out.append("tư")
+        else:
+            out.append(_D[u])
+    return " ".join(out)
+
+
+def number_to_words(n):
+    if n == 0:
+        return "không"
+    parts, units = [], ["", "nghìn", "triệu", "tỷ"]
+    groups = []
+    while n:
+        groups.append(n % 1000)
+        n //= 1000
+    for i in range(len(groups) - 1, -1, -1):
+        g = groups[i]
+        if g:
+            parts.append(_read_3(g, full=(i < len(groups) - 1)))
+            parts.append(units[i])
+    return " ".join(p for p in parts if p)
+
+
+def clean(text):
+    for k, v in _ABBR.items():
+        text = text.replace(k, v)
+    text = re.sub(r"\d+", lambda m: number_to_words(int(m.group())), text)
+    return re.sub(r"\s+", " ", text).strip().lower()

toolnet_tts/__init__.py

@@ -0,0 +1,3 @@
+from .client import Communicate, ToolnetError
+
+__all__ = ["Communicate", "ToolnetError"]

toolnet_tts/cli.py

@@ -0,0 +1,27 @@
+import argparse
+from .client import Communicate, ToolnetError
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Chuyển văn bản tiếng Việt thành giọng nói ToolNet.")
+    source = parser.add_mutually_exclusive_group(required=True)
+    source.add_argument("--text", help="Văn bản cần đọc")
+    source.add_argument("--file", help="File văn bản UTF-8")
+    parser.add_argument("--write-media", required=True, help="Đường dẫn file WAV")
+    parser.add_argument("--server", help="URL máy chủ (hoặc biến TOOLNET_TTS_SERVER)")
+    parser.add_argument("--voice", default="Thiện Minh")
+    args = parser.parse_args()
+    try:
+        if args.file:
+            with open(args.file, encoding="utf-8-sig") as file:
+                text = file.read()
+        else:
+            text = args.text
+        Communicate(text, args.voice, server=args.server).save_sync(args.write_media)
+    except (ToolnetError, ValueError, OSError) as exc:
+        parser.exit(1, f"toolnet-tts: {exc}\n")
+    print(f"Đã tạo {args.write_media}")
+
+
+if __name__ == "__main__":
+    main()

toolnet_tts/client.py

@@ -0,0 +1,51 @@
+"""A small HTTP client for a ToolNet TTS server."""
+import asyncio
+import json
+import os
+from pathlib import Path
+from urllib.error import HTTPError, URLError
+from urllib.request import Request, urlopen
+
+
+class ToolnetError(RuntimeError):
+    """The speech service could not fulfil a request."""
+
+
+class Communicate:
+    def __init__(self, text, voice="Thiện Minh", *, server=None, api_key=None, timeout=180):
+        if not isinstance(text, str) or not text.strip():
+            raise ValueError("text must be a non-empty string")
+        self.text = text
+        self.voice = voice
+        self.server = (server or os.getenv("TOOLNET_TTS_SERVER", "")).rstrip("/")
+        if not self.server.startswith(("https://", "http://")):
+            raise ValueError("Set server= or TOOLNET_TTS_SERVER to the speech server URL")
+        self.api_key = api_key if api_key is not None else os.getenv("TOOLNET_TTS_API_KEY")
+        self.timeout = timeout
+
+    def save_sync(self, path):
+        path = Path(path)
+        if path.suffix.lower() != ".wav":
+            raise ValueError("Output must have a .wav extension")
+        payload = json.dumps({"text": self.text, "voice": self.voice}, ensure_ascii=False).encode("utf-8")
+        headers = {"Content-Type": "application/json", "Accept": "audio/wav"}
+        if self.api_key:
+            headers["Authorization"] = f"Bearer {self.api_key}"
+        request = Request(f"{self.server}/v1/speech", data=payload, headers=headers, method="POST")
+        try:
+            with urlopen(request, timeout=self.timeout) as response:
+                content_type = response.headers.get_content_type()
+                if content_type not in ("audio/wav", "audio/x-wav"):
+                    raise ToolnetError(f"Unexpected response type: {content_type}")
+                audio = response.read()
+        except HTTPError as exc:
+            raise ToolnetError(f"Speech server returned HTTP {exc.code}") from exc
+        except (URLError, TimeoutError, OSError) as exc:
+            raise ToolnetError(f"Cannot reach speech server: {exc}") from exc
+        if len(audio) < 44 or audio[:4] != b"RIFF" or audio[8:12] != b"WAVE":
+            raise ToolnetError("Speech server returned invalid WAV data")
+        path.write_bytes(audio)
+        return str(path)
+
+    async def save(self, path):
+        return await asyncio.to_thread(self.save_sync, path)

toolnet_tts/server.py

@@ -0,0 +1,51 @@
+"""Self-hosted VieNeu inference service. Install with toolnet-tts[server]."""
+import hmac
+import io
+import os
+import threading
+
+from fastapi import FastAPI, Header, HTTPException
+from fastapi.responses import Response
+from pydantic import BaseModel, Field
+from starlette.concurrency import run_in_threadpool
+
+app = FastAPI(title="ToolNet TTS")
+_engine = None
+_lock = threading.Lock()
+
+
+class SpeechRequest(BaseModel):
+    text: str = Field(min_length=1, max_length=10000)
+    voice: str = "Thiện Minh"
+
+
+def _synthesize(text: str, voice: str) -> bytes:
+    global _engine
+    import soundfile as sf
+    from vieneu import Vieneu
+
+    with _lock:
+        if _engine is None:
+            _engine = Vieneu(mode="v3turbo", device=os.getenv("TOOLNET_TTS_DEVICE", "auto"))
+        audio = _engine.infer(text, voice=voice)
+        output = io.BytesIO()
+        sf.write(output, audio, _engine.sample_rate, format="WAV", subtype="PCM_16")
+        return output.getvalue()
+
+
+@app.get("/health")
+def health():
+    return {"status": "ok", "model_loaded": _engine is not None}
+
+
+@app.post("/v1/speech")
+async def speech(request: SpeechRequest, authorization: str = Header(default="")):
+    key = os.getenv("TOOLNET_TTS_API_KEY")
+    if key and not hmac.compare_digest(authorization, f"Bearer {key}"):
+        raise HTTPException(status_code=401, detail="Unauthorized")
+    if not request.text.strip():
+        raise HTTPException(status_code=422, detail="Text is empty")
+    if request.voice not in ("Thiện Minh", "giong1"):
+        raise HTTPException(status_code=400, detail="Voice is not supported")
+    audio = await run_in_threadpool(_synthesize, request.text, "Thiện Minh")
+    return Response(audio, media_type="audio/wav")

toolnet_tts-0.2.0.dist-info/METADATA

@@ -0,0 +1,64 @@
+Metadata-Version: 2.4
+Name: toolnet-tts
+Version: 0.2.0
+Summary: Lightweight client for ToolNet Vietnamese text-to-speech
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Provides-Extra: server
+Requires-Dist: fastapi<1,>=0.115; extra == "server"
+Requires-Dist: uvicorn<1,>=0.30; extra == "server"
+Requires-Dist: vieneu==3.8.4; extra == "server"
+Requires-Dist: onnxruntime<1.23,>=1.20; extra == "server"
+
+# toolnet-tts
+
+Client Python và Node.js gửi văn bản đến máy chủ ToolNet TTS để tạo WAV bằng giọng Thiện Minh của VieNeu v3 Turbo. Client không tải model; máy chủ cần cài model và các phụ thuộc riêng. Cần có URL máy chủ để sử dụng.
+
+## Python
+
+```bash
+pip install toolnet-tts
+export TOOLNET_TTS_SERVER="https://your-server.example"
+# Nếu máy chủ yêu cầu khóa:
+export TOOLNET_TTS_API_KEY="your-api-key"
+toolnet-tts --text "Xin chào các bạn" --write-media hello.wav
+```
+
+```python
+import asyncio
+from toolnet_tts import Communicate
+
+asyncio.run(Communicate("Xin chào các bạn").save("hello.wav"))
+# Hoặc đồng bộ: Communicate("Xin chào").save_sync("hello.wav")
+```
+
+Có thể đặt URL bằng `server="https://your-server.example"`. Chỉ hỗ trợ WAV; hiện không hỗ trợ MP3, word boundaries hoặc phát trực tiếp.
+
+## Node.js
+
+```bash
+npm install toolnet-tts
+export TOOLNET_TTS_SERVER="https://your-server.example"
+npx toolnet-tts --text "Xin chào các bạn" --write-media hello.wav
+```
+
+```js
+import { Communicate } from 'toolnet-tts';
+await new Communicate('Xin chào các bạn').save('hello.wav');
+```
+
+## Chạy máy chủ riêng
+
+```bash
+pip install 'toolnet-tts[server]'
+export TOOLNET_TTS_API_KEY="replace-with-a-long-random-key"
+uvicorn toolnet_tts.server:app --host 127.0.0.1 --port 8000
+```
+
+Đặt `TOOLNET_TTS_DEVICE=cpu` để chạy CPU. Lần đầu cần tải model VieNeu; nếu dùng GPU cần thêm các phụ thuộc CUDA tương thích. `/health` báo tiến trình và trạng thái nạp model; `POST /v1/speech` nhận JSON `{"text":"...","voice":"Thiện Minh"}` và trả WAV. Đặt `TOOLNET_TTS_API_KEY` ở cả client và server để xác thực Bearer. Nếu mở máy chủ cho client ở xa, cần cấu hình HTTPS và mạng phù hợp; lệnh trên chỉ lắng nghe tại localhost.
+
+Giọng và model: [VieNeu-TTS v3 Turbo](https://huggingface.co/pnnbao-ump/VieNeu-TTS-v3-Turbo), Apache-2.0. Dự án này sử dụng model gốc; không đưa checkpoint Piper thử nghiệm vào gói client.
+
+## Chạy cục bộ kiểu cũ
+
+`from toolnet import TTS` vẫn có thể dùng sau khi cài `toolnet-tts[server]`. API này tải và chạy VieNeu trên chính máy gọi, không dùng máy chủ.

toolnet_tts-0.2.0.dist-info/RECORD

@@ -0,0 +1,12 @@
+toolnet/__init__.py,sha256=rGUnFxpGA_sOO8efCNjNFOPyk3ipp4_nGMlqX_znkjE,43
+toolnet/engine.py,sha256=5INBZY2u_eXc7w6zIuUAVoJ80-utpHv1qEWyhMXWIHw,392
+toolnet/text.py,sha256=xqiom_AlLmmBufUdKq0NFPERa7eUGC8t715Eqj4EBk4,1391
+toolnet_tts/__init__.py,sha256=-jZGmk2Xf2cZ16aD101001U5X89Eb0J5Uk2zogJB8ME,89
+toolnet_tts/cli.py,sha256=RCpcDQ7-Nef8ktVOq0wtLCPTfQEMP_NlDd9zqLJA_uU,1105
+toolnet_tts/client.py,sha256=PD_0YXSc6CvNsJzOXLn0FsgWicVMJvlm3xSlsXaHB_8,2382
+toolnet_tts/server.py,sha256=QzKFSIfFjmLbNGMmQFZx77ySXwPQU2dRKa_FGPESHK8,1725
+toolnet_tts-0.2.0.dist-info/METADATA,sha256=iE1YIhf9RxzK5AgUucssa0gAZ4RGMPaF-g-zjumqXus,2745
+toolnet_tts-0.2.0.dist-info/WHEEL,sha256=mOdXR9C3wMVov7cSY8PY-XCzo3SmmWCHvGQr5csJoao,91
+toolnet_tts-0.2.0.dist-info/entry_points.txt,sha256=n1ZK89GzaXROtCodI2Br3RcMHEy_cphJ92Ew3tROcV4,53
+toolnet_tts-0.2.0.dist-info/top_level.txt,sha256=jhs3eUnPhSz_aMq3JQHiW-Cc6ZNUZSv44mD6WkUXqSM,20
+toolnet_tts-0.2.0.dist-info/RECORD,,

toolnet_tts-0.2.0.dist-info/WHEEL

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

toolnet_tts-0.2.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+toolnet-tts = toolnet_tts.cli:main

toolnet_tts-0.2.0.dist-info/top_level.txt

@@ -0,0 +1,2 @@
+toolnet
+toolnet_tts