TTS

# TTS 활용 예제 - 다양한 방법
import os

# ===== 1. OpenAI TTS API =====
from openai import OpenAI

client = OpenAI()

def openai_tts(text: str, voice: str = "alloy", output_file: str = "speech.mp3"):
    """
    OpenAI TTS API 사용
    voices: alloy, echo, fable, onyx, nova, shimmer
    """
    response = client.audio.speech.create(
        model="tts-1",  # tts-1 (빠름) 또는 tts-1-hd (고품질)
        voice=voice,
        input=text
    )
    response.stream_to_file(output_file)
    print(f"생성 완료: {output_file}")

# 사용 예시
# openai_tts("안녕하세요, KAITRUST AI 백과사전입니다.", "nova")

# ===== 2. Google Cloud TTS =====
from google.cloud import texttospeech

def google_tts(text: str, language: str = "ko-KR", output_file: str = "output.mp3"):
    """Google Cloud TTS - 고품질 한국어 지원"""
    client = texttospeech.TextToSpeechClient()

    synthesis_input = texttospeech.SynthesisInput(text=text)

    voice = texttospeech.VoiceSelectionParams(
        language_code=language,
        ssml_gender=texttospeech.SsmlVoiceGender.FEMALE
    )

    audio_config = texttospeech.AudioConfig(
        audio_encoding=texttospeech.AudioEncoding.MP3,
        speaking_rate=1.0,  # 0.25 ~ 4.0
        pitch=0.0  # -20.0 ~ 20.0
    )

    response = client.synthesize_speech(
        input=synthesis_input, voice=voice, audio_config=audio_config
    )

    with open(output_file, "wb") as out:
        out.write(response.audio_content)
    print(f"생성 완료: {output_file}")

# ===== 3. 오픈소스 TTS (Coqui TTS) =====
# pip install TTS
from TTS.api import TTS

def coqui_tts(text: str, output_file: str = "coqui_output.wav"):
    """무료 오픈소스 TTS - 로컬 실행"""
    # 사용 가능한 모델 목록: TTS.list_models()
    tts = TTS(model_name="tts_models/ko/css10/vits")  # 한국어 모델

    tts.tts_to_file(
        text=text,
        file_path=output_file
    )
    print(f"생성 완료: {output_file}")

# ===== 4. Edge TTS (무료, Microsoft) =====
# pip install edge-tts
import edge_tts
import asyncio

async def edge_tts_generate(text: str, voice: str = "ko-KR-SunHiNeural"):
    """Microsoft Edge TTS - 무료, 고품질"""
    communicate = edge_tts.Communicate(text, voice)
    await communicate.save("edge_output.mp3")
    print("생성 완료: edge_output.mp3")

# 한국어 음성: ko-KR-SunHiNeural (여성), ko-KR-InJoonNeural (남성)
# asyncio.run(edge_tts_generate("안녕하세요, 테스트입니다."))

# ===== 5. 비교 테이블 출력 =====
print("=" * 60)
print("TTS 서비스 비교")
print("=" * 60)
print(f"{'서비스':<15} {'비용':<20} {'품질':<10} {'한국어'}")
print("-" * 60)
print(f"{'OpenAI TTS':<15} {'$15/1M chars':<20} {'우수':<10} {'O'}")
print(f"{'Google TTS':<15} {'$4~16/1M chars':<20} {'최상':<10} {'O'}")
print(f"{'Edge TTS':<15} {'무료':<20} {'우수':<10} {'O'}")
print(f"{'Coqui TTS':<15} {'무료(로컬)':<20} {'양호':<10} {'O'}")