Audiobook Producer
104 lines
4.9 KiB
Python
104 lines
4.9 KiB
Python
"""Voice-tuning clips: one paragraph, many voice variants, so a human can pick by ear.
|
|
Standard library only. Usage:
|
|
python3 tune_clips.py --text-file <chapter-01.txt> --paragraph 3 --out <dir> [--chatterbox URL] [--kokoro URL]
|
|
Writes <out>/NN-name.wav (and .mp3 if ffmpeg exists) plus <out>/clips.md describing each clip."""
|
|
import argparse
|
|
import json
|
|
import os
|
|
import shutil
|
|
import subprocess
|
|
import time
|
|
import urllib.request
|
|
|
|
SEED = 7 # same seed everywhere so differences come from the settings, not luck
|
|
|
|
# Chatterbox: stock male voices, then setting changes on one voice. exaggeration = expressiveness (lower = calmer),
|
|
# cfg_weight = pacing (lower = slower, more deliberate), temperature = variation (lower = steadier).
|
|
CHATTERBOX = [
|
|
("henry-default", "Henry", dict(exaggeration=0.5, cfg_weight=0.5, temperature=0.8), "Henry, server defaults."),
|
|
("henry-calm", "Henry", dict(exaggeration=0.3, cfg_weight=0.3, temperature=0.6), "Henry, calmer and slower: low expressiveness, low pace weight, steadier."),
|
|
("gabriel-calm", "Gabriel", dict(exaggeration=0.3, cfg_weight=0.3, temperature=0.6), "Gabriel, same calm settings as Henry calm."),
|
|
("adrian-calm", "Adrian", dict(exaggeration=0.3, cfg_weight=0.3, temperature=0.6), "Adrian, same calm settings."),
|
|
("everett-calm", "Everett", dict(exaggeration=0.3, cfg_weight=0.3, temperature=0.6), "Everett, same calm settings."),
|
|
("henry-warmer", "Henry", dict(exaggeration=0.4, cfg_weight=0.4, temperature=0.7), "Henry, between default and calm: a bit more life."),
|
|
]
|
|
# Kokoro: male voices. speed < 1 is slower.
|
|
KOKORO = [
|
|
("kokoro-adam", "am_adam", 0.92, "Kokoro am_adam at 0.92 speed."),
|
|
("kokoro-michael", "am_michael", 0.92, "Kokoro am_michael at 0.92 speed."),
|
|
("kokoro-george", "bm_george", 0.92, "Kokoro bm_george (British) at 0.92 speed."),
|
|
]
|
|
|
|
|
|
def post(url, body, timeout=600):
|
|
req = urllib.request.Request(url, json.dumps(body).encode(), {"Content-Type": "application/json"})
|
|
with urllib.request.urlopen(req, timeout=timeout) as r:
|
|
return r.read()
|
|
|
|
|
|
def get_json(url):
|
|
with urllib.request.urlopen(url, timeout=30) as r:
|
|
return json.loads(r.read())
|
|
|
|
|
|
def chatterbox_clip(base, text, voice, p):
|
|
body = dict(text=text, voice_mode="predefined", predefined_voice_id=voice + ".wav" if not voice.endswith(".wav") else voice,
|
|
output_format="wav", split_text=True, chunk_size=240, seed=SEED, **p)
|
|
return post(base.rstrip("/") + "/tts", body)
|
|
|
|
|
|
def kokoro_clip(base, text, voice, speed):
|
|
return post(base.rstrip("/") + "/v1/audio/speech",
|
|
dict(model="kokoro", input=text, voice=voice, speed=speed, response_format="wav"))
|
|
|
|
|
|
def main():
|
|
ap = argparse.ArgumentParser()
|
|
ap.add_argument("--text-file", required=True)
|
|
ap.add_argument("--paragraph", type=int, default=3, help="1-based paragraph number")
|
|
ap.add_argument("--out", required=True)
|
|
ap.add_argument("--chatterbox", default="http://127.0.0.1:8090")
|
|
ap.add_argument("--kokoro", default="http://127.0.0.1:8091")
|
|
a = ap.parse_args()
|
|
paras = [p.strip() for p in open(a.text_file, encoding="utf-8").read().split("\n\n") if p.strip()]
|
|
text = paras[a.paragraph - 1]
|
|
os.makedirs(a.out, exist_ok=True)
|
|
rows = ["# Voice-tuning clips", "", "Text: Clark, Chapter I, paragraph %d (%d words). Same text in every clip." % (a.paragraph, len(text.split())), "", "> " + text, ""]
|
|
ff = shutil.which("ffmpeg")
|
|
n = 0
|
|
jobs = [(nm, "Chatterbox", lambda v=v, p=p: chatterbox_clip(a.chatterbox, text, v, p), d) for nm, v, p, d in CHATTERBOX]
|
|
try:
|
|
have = {x if isinstance(x, str) else x.get("id", x.get("name")) for x in get_json(a.kokoro.rstrip("/") + "/v1/audio/voices").get("voices", [])}
|
|
except Exception as e:
|
|
print("kokoro voice list failed:", e)
|
|
have = None
|
|
for nm, v, sp, d in KOKORO:
|
|
if have is not None and v not in have:
|
|
print("skip", v, "(not on server)")
|
|
continue
|
|
jobs.append((nm, "Kokoro", lambda v=v, sp=sp: kokoro_clip(a.kokoro, text, v, sp), d))
|
|
for nm, eng, fn, d in jobs:
|
|
n += 1
|
|
base = os.path.join(a.out, "%02d-%s" % (n, nm))
|
|
t0 = time.time()
|
|
try:
|
|
data = fn()
|
|
except Exception as e:
|
|
print("FAILED", nm, e)
|
|
rows.append("- %02d-%s: FAILED (%s)" % (n, nm, e))
|
|
continue
|
|
with open(base + ".wav", "wb") as f:
|
|
f.write(data)
|
|
final = base + ".wav"
|
|
if ff:
|
|
subprocess.run([ff, "-y", "-loglevel", "error", "-i", base + ".wav", "-ac", "1", "-ar", "44100", "-b:a", "192k", base + ".mp3"], check=True)
|
|
final = base + ".mp3"
|
|
print("ok %s %.1fs render, %s" % (nm, time.time() - t0, final))
|
|
rows.append("- **%02d-%s** (%s): %s" % (n, nm, eng, d))
|
|
with open(os.path.join(a.out, "clips.md"), "w", encoding="utf-8") as f:
|
|
f.write("\n".join(rows) + "\n")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|