patch_tts.py

import os
import sys
from unittest.mock import patch
from io import StringIO
from TTS.api import TTS
import logging

# Configure logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

# Force license acceptance
os.environ["COQUI_TTS_ACCEPT_LICENSE"] = "y"

# Mock input to return 'y' for license prompt
def mock_input(prompt):
    logger.info("Mocking input for license prompt")
    return "y"

# Patch input function
with patch('builtins.input', mock_input):
    try:
        logger.info("Initializing TTS with XTTS-v2 model")
        tts = TTS(model_name="tts_models/multilingual/multi-dataset/xtts_v2", progress_bar=False)
        logger.info("TTS initialized successfully")
    except Exception as e:
        logger.error(f"Failed to initialize TTS: {str(e)}")
        raise