fcyai/ChatTTS-Story-Telling

1import argparse2import os3from tts_model import load_chat_tts_model, tts4from config import DEFAULT_SPEED, DEFAULT_ORAL, DEFAULT_LAUGH, DEFAULT_BK, DEFAULT_SEG_LENGTH, DEFAULT_BATCH_SIZE5 6if __name__ == "__main__":7    parser = argparse.ArgumentParser(description="Generate TTS audio from text file.")8    parser.add_argument("--text_file", type=str, required=True, help="Path to the text file to convert.")9    parser.add_argument("--seed", type=int,10                        help="Specific seed for generating audio. If not provided, seeds will be random.")11    parser.add_argument("--speed", type=int, default=DEFAULT_SPEED, help="Speed of generated audio.")12    parser.add_argument("--oral", type=int, default=DEFAULT_ORAL, help="Oral")13    parser.add_argument("--laugh", type=int, default=DEFAULT_LAUGH, help="Laugh")14    parser.add_argument("--bk", type=int, default=DEFAULT_BK, help="Break")15    parser.add_argument("--seg", type=int, default=DEFAULT_SEG_LENGTH, help="Max len of text segments.")16    parser.add_argument("--batch", type=int, default=DEFAULT_BATCH_SIZE, help="Batch size for TTS inference.")17    parser.add_argument("--source", type=str, default="huggingface", help="Model source: 'huggingface' or 'local'.")18    parser.add_argument("--local_path", type=str, help="Path to local model if source is 'local'.")19 20    args = parser.parse_args()21    chat = load_chat_tts_model(source=args.source, local_path=args.local_path)22    # chat = None23    tts(chat, args.text_file, args.seed, args.speed, args.oral, args.laugh, args.bk, args.seg,24        args.batch)25