Spaces:
Paused
Paused
File size: 3,352 Bytes
9ae08c5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 | import argparse
import time
from chatter import ChatterWrapper
from tts import synthesize_text
from audio_utils import to_portable_file
from audio_file_utils import AudioFileUtils
from effects.chain_processor import EffectChainProcessor
import json
def main():
parser = argparse.ArgumentParser(description="Test TTS Synthesis with Piper")
parser.add_argument("text", help="Text to synthesize")
parser.add_argument("--model", default="piper", help="Model used to generate audio (default: Piper) - options: piper, chatterbox")
# parameters for Piper
parser.add_argument("--local", default="fr_FR", help="Local voice file (default: fr_FR)")
parser.add_argument("--voice", default="siwis-medium", help="Voice model (default: siwis-medium)")
parser.add_argument("--silence", type=int, default=1, help="Silence length (default: 1)")
parser.add_argument("--speed", type=float, default=1.0, help="Speech speed (default: 1.0)")
parser.add_argument("--noise_w", type=float, default=0.8, help="Noise weight (default: 0.8)")
# parameters for Chatterbox exaggeration float 0-1 (default: 0.5), cfg_weight float 0-1 (default: 0.5), prompt string (default: None)
parser.add_argument("--exaggeration", type=float, default=0.5, help="Exaggeration factor (default: 0.5)")
parser.add_argument("--cfg_weight", type=float, default=0.5, help="CFG weight (default: 0.5)")
parser.add_argument("--prompt", type=str, default=None, help="Prompt for Chatterbox (default: None) - Value must be a path to a wav file.")
# add a param --lite-file without value
parser.add_argument("--lite-file", action='store_true', help="Convert to portable file format")
# optionnal param of json data called --effects
parser.add_argument("--effects", type=str, default=None, help="Effects to apply to the audio")
args = parser.parse_args()
try:
if args.model == "chatterbox":
# generate random file name
chatter_wrapper = ChatterWrapper()
output_file = chatter_wrapper.run_synchronously(
text=args.text,
exaggeration=args.exaggeration,
cfg_weight=args.cfg_weight,
voice=args.prompt
)
else:
output_file = synthesize_text(
text=args.text,
local=args.local,
voice=args.voice,
silence=args.silence,
speed=args.speed,
noise_w=args.noise_w
)
if args.effects:
# Parse effects JSON
effects_data = json.loads(args.effects)
if not isinstance(effects_data, list):
raise ValueError("Effects should be a JSON array of effect steps")
# Load audio
audio, framerate = AudioFileUtils.wav_to_audio(output_file)
# Process chain
processor = EffectChainProcessor()
audio = processor.apply_chain(audio, framerate, effects_data)
# Save result
AudioFileUtils.audio_to_wav(audio, framerate, output_file)
if args.lite_file:
to_portable_file(output_file, output_file)
print(f"Audio file generated: {output_file}")
except Exception as e:
print(f"Error: {e}")
if __name__ == "__main__":
main()
|