96 lines
3.0 KiB
Python
96 lines
3.0 KiB
Python
#!/usr/bin/env python3
|
|
"""Text-to-speech via Supertonic local server.
|
|
|
|
CLI:
|
|
python3 tools/tts.py --text "Salut Marius" [--voice M1] [--lang ro]
|
|
→ stdout: {"ok": true, "path": "/tmp/echo-tts-xxx.wav", "size_bytes": 12345}
|
|
→ stdout: {"ok": false, "error": "..."}
|
|
|
|
Module:
|
|
from tools.tts import synthesize
|
|
result = synthesize("text", voice="M1", lang="ro")
|
|
"""
|
|
|
|
import argparse
|
|
import json
|
|
import sys
|
|
import tempfile
|
|
|
|
import httpx
|
|
|
|
SUPERTONIC_URL = "http://127.0.0.1:7788"
|
|
VOICES = {"M1", "M2", "M3", "M4", "M5", "F1", "F2", "F3", "F4", "F5"}
|
|
DEFAULT_VOICE = "M1"
|
|
DEFAULT_LANG = "ro"
|
|
|
|
|
|
def synthesize(text: str, voice: str = DEFAULT_VOICE, lang: str = DEFAULT_LANG) -> dict:
|
|
"""Call Supertonic server and save audio to a temp WAV file.
|
|
|
|
Returns:
|
|
{"ok": True, "path": "/tmp/echo-tts-xxx.wav", "size_bytes": N}
|
|
{"ok": False, "error": "mesaj eroare"}
|
|
"""
|
|
if not text or not text.strip():
|
|
return {"ok": False, "error": "Text gol."}
|
|
|
|
voice = voice.upper()
|
|
if voice not in VOICES:
|
|
voice = DEFAULT_VOICE
|
|
|
|
try:
|
|
resp = httpx.post(
|
|
f"{SUPERTONIC_URL}/v1/audio/speech",
|
|
json={
|
|
"model": "supertonic-3",
|
|
"input": text,
|
|
"voice": voice,
|
|
"response_format": "wav",
|
|
},
|
|
timeout=60.0,
|
|
)
|
|
resp.raise_for_status()
|
|
except httpx.ConnectError:
|
|
return {
|
|
"ok": False,
|
|
"error": (
|
|
"Serverul Supertonic nu rulează pe :7788. "
|
|
"Pornește cu: systemctl --user start supertonic-tts"
|
|
),
|
|
}
|
|
except httpx.HTTPStatusError as e:
|
|
body = e.response.text[:300]
|
|
# Fallback: dacă lang=ro eșuează, încearcă na (language-agnostic)
|
|
if lang != "na":
|
|
return synthesize(text, voice=voice, lang="na")
|
|
return {"ok": False, "error": f"HTTP {e.response.status_code}: {body}"}
|
|
except Exception as e:
|
|
return {"ok": False, "error": str(e)}
|
|
|
|
# Salvează în fișier temp
|
|
try:
|
|
fd, path = tempfile.mkstemp(prefix="echo-tts-", suffix=".wav")
|
|
with open(fd, "wb") as f:
|
|
f.write(resp.content)
|
|
return {"ok": True, "path": path, "size_bytes": len(resp.content)}
|
|
except Exception as e:
|
|
return {"ok": False, "error": f"Scriere fișier: {e}"}
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Supertonic TTS CLI")
|
|
parser.add_argument("--text", required=True, help="Text de convertit în audio")
|
|
parser.add_argument(
|
|
"--voice", default=DEFAULT_VOICE,
|
|
help="Voce: M1-M5 (masculin) sau F1-F5 (feminin). Default: M1"
|
|
)
|
|
parser.add_argument(
|
|
"--lang", default=DEFAULT_LANG,
|
|
help="Limbă (ro, en, na). Default: ro. Fallback automat la na dacă ro eșuează."
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
result = synthesize(args.text, voice=args.voice, lang=args.lang)
|
|
print(json.dumps(result, ensure_ascii=False))
|
|
sys.exit(0 if result.get("ok") else 1)
|