Files
dify-handleidingen/05-transcriptietool/maak_opname_drie_stemmen.py
T
2026-09-20 21:46:31 +02:00

65 lines
2.9 KiB
Python

#!/usr/bin/env python3
"""Maakt van sectieoverleg-script.txt een testopname met drie verschillende computerstemmen,
zodat de sprekerherkenning (gpt-4o-transcribe-diarize) iets te onderscheiden heeft.
Gebruik: OPENAI_API_KEY=sk-... python3 maak_opname_drie_stemmen.py
Vereist: ffmpeg (aanwezig) en een OpenAI-sleutel in de omgevingsvariabele; de sleutel wordt
nergens opgeslagen. Kosten: circa 4.000 tekens bij tts-1, ongeveer 6 cent.
Uitvoer: sectieoverleg-drie-stemmen.mp3 (128 kbps, om te beluisteren)
sectieoverleg-drie-stemmen-klein.mp3 (32 kbps mono, onder de 5 MB-grens van de tool)
"""
import json
import os
import re
import subprocess
import sys
import tempfile
import urllib.request
from pathlib import Path
HERE = Path(__file__).resolve().parent
SCRIPT = HERE / "sectieoverleg-script.txt"
STEMMEN = {"Karin": "nova", "Bram": "echo", "Fatima": "alloy"} # drie duidelijk verschillende stemmen
VERTELLER = "onyx" # voor de openingsregel
key = os.environ.get("OPENAI_API_KEY")
if not key:
sys.exit("Zet de sleutel in de omgevingsvariabele OPENAI_API_KEY en start opnieuw.")
def spreek(tekst: str, stem: str, doel: Path) -> None:
req = urllib.request.Request(
"https://api.openai.com/v1/audio/speech",
data=json.dumps({"model": "tts-1", "voice": stem, "input": tekst, "response_format": "mp3"}).encode(),
headers={"Authorization": f"Bearer {key}", "Content-Type": "application/json"},
)
with urllib.request.urlopen(req, timeout=120) as r:
doel.write_bytes(r.read())
regels = [r.strip() for r in SCRIPT.read_text(encoding="utf-8").splitlines() if r.strip()]
with tempfile.TemporaryDirectory() as tmp:
tmp = Path(tmp)
delen = []
for i, regel in enumerate(regels):
m = re.match(r"^(Karin|Bram|Fatima):\s*(.+)$", regel)
if m:
stem, tekst = STEMMEN[m.group(1)], m.group(2)
else:
stem, tekst = VERTELLER, regel
deel = tmp / f"{i:03d}.mp3"
spreek(tekst, stem, deel)
delen.append(deel)
print(f"{i + 1}/{len(regels)} {stem}: {tekst[:50]}...")
lijst = tmp / "lijst.txt"
lijst.write_text("".join(f"file '{d}'\n" for d in delen))
# Korte stilte tussen sprekers komt vanzelf door de pauze aan het eind van elke zin.
subprocess.run(["ffmpeg", "-y", "-loglevel", "error", "-f", "concat", "-safe", "0", "-i", str(lijst),
"-c:a", "libmp3lame", "-b:a", "128k", str(HERE / "sectieoverleg-drie-stemmen.mp3")], check=True)
subprocess.run(["ffmpeg", "-y", "-loglevel", "error", "-i", str(HERE / "sectieoverleg-drie-stemmen.mp3"),
"-ac", "1", "-ar", "16000", "-b:a", "32k", str(HERE / "sectieoverleg-drie-stemmen-klein.mp3")], check=True)
for naam in ("sectieoverleg-drie-stemmen.mp3", "sectieoverleg-drie-stemmen-klein.mp3"):
p = HERE / naam
print(f"{naam}: {p.stat().st_size // 1024} kB")