liqiang
b119135836
Build latest book artifacts / build (push) Canceled after 0s
dependency resolution / resolve (3.11) (push) Canceled after 0s
dependency resolution / resolve (3.13) (push) Canceled after 0s
deploy-pages / build (push) Canceled after 0s
deploy-pages / deploy (push) Canceled after 0s
i18n consistency check / check (push) Canceled after 0s
provider adoption tests / test (chapter2/context-compression) (push) Canceled after 0s
provider adoption tests / test (chapter2/prompt-injection) (push) Canceled after 0s
provider adoption tests / test (chapter2/system-hint) (push) Canceled after 0s
provider adoption tests / test (chapter3/log-sanitization) (push) Canceled after 0s
web-search-agent tests / test (push) Canceled after 0s
web-search-agent tests / agentbook (push) Canceled after 0s
40 lines
1.5 KiB
Python
40 lines
1.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""可选的语音合成（TTS）——把玩家公开发言合成为语音。
|
|
|
|
语音是本实验的**可选增强**，不是跑通的必需：默认文本模式即可完整跑完一局并
|
|
验证信息隔离。加 --voice 时才启用，用 OpenAI tts-1 把每条公开发言合成 mp3，
|
|
存到 audio/ 目录；在 macOS 上可用 afplay 顺带播放（--play）。
|
|
"""
|
|
|
|
import os
|
|
import subprocess
|
|
|
|
from .agent import get_client
|
|
|
|
|
|
# 给不同玩家分配不同音色，便于区分
|
|
_VOICES = ["alloy", "echo", "fable", "onyx", "nova", "shimmer", "coral"]
|
|
|
|
|
|
class TTS:
|
|
def __init__(self, out_dir: str, play: bool = False):
|
|
self.out_dir = out_dir
|
|
self.play = play
|
|
os.makedirs(out_dir, exist_ok=True)
|
|
self._idx = 0
|
|
|
|
def synth(self, speaker: str, text: str, round_no: int):
|
|
voice = _VOICES[(int(speaker.lstrip("P")) - 1) % len(_VOICES)]
|
|
path = os.path.join(self.out_dir, f"r{round_no}_{speaker}_{self._idx}.mp3")
|
|
self._idx += 1
|
|
try:
|
|
resp = get_client().audio.speech.create(
|
|
model="tts-1", voice=voice, input=text)
|
|
resp.stream_to_file(path)
|
|
print(f" [TTS] {speaker} 发言已合成语音（音色 {voice}）→ {path}")
|
|
if self.play:
|
|
# macOS 自带 afplay；其它平台请自行改播放器
|
|
subprocess.run(["afplay", path], check=False)
|
|
except Exception as e:
|
|
print(f" [TTS] 合成失败（不影响游戏进行）：{e}")
|