o U6i @sdZddlZddlZddlmZddlmZddlmZddl m Z m Z m Z dde d e eBdBd e d efd d Zde ded eefddZdde d e eBdBd efddZdeeded e fddZdde ded eefddZdS)z.Text-to-Speech engine with timing information.N)Path)gTTS) AudioSegment) TTS_LANGUAGETTS_SPEED_SLOW OUTPUT_DIRtext output_pathlanguagereturncCs|pt}t||td}tjddd}||j|j}Wdn1s&wYt|}t |}|durEt dt |dd}nt |}|j jddd t||dt||||d S) a( Generate speech audio from text using Google TTS. Args: text: Text to convert to speech output_path: Path to save the audio file (optional) language: Language code (default: from config) Returns: dict with audio info including duration and path )r langZslowz.mp3F)suffixdeleteNZspeech_iT)parentsexist_ok)successpath duration_msr r )rrrtempfileNamedTemporaryFilesavenamerZfrom_mp3lenrhashrparentmkdirosrenamestr)r r r ZttstmpZtmp_pathaudiorr"M/var/www/tkim.planitai.co.jp/blog/20251208-make-apng-tool/./src/tts_engine.pygenerate_speech s&    r$total_duration_msc Csddlm}||}|sgStdd|D}|dkrgSg}d}|D]'}|d}|||} ||d|dt|t|| t| d || 7}q#|S) a Estimate phoneme timings based on text length and total audio duration. This is a simple estimation - for more accurate results, use a forced alignment tool like Montreal Forced Aligner. Args: text: The text being spoken total_duration_ms: Total audio duration in milliseconds Returns: List of dicts with phoneme, start_ms, end_ms, viseme r)korean_text_to_visemescss|]}|dVqdS)durationNr").0itemr"r"r# Osz+estimate_phoneme_timings..rr'charviseme)r+r,start_msend_msr)text_to_visemer&sumappendint) r r%r&Z viseme_dataZtotal_relativetimings current_timer)Zrelative_durationZactual_durationr"r"r#estimate_phoneme_timings8s*    r5cCs4t||}|ds |St||d}i|d|iS)z Generate speech and phoneme timing information. Args: text: Text to convert to speech output_path: Path to save the audio file Returns: dict with audio path, duration, and phoneme timings rrr3)r$r5)r r speech_resultr3r"r"r#generate_speech_with_timingis r7r3time_mscCs:|D]}|d|kr|dkrnq|dSqdS)z Get the viseme that should be displayed at a given time. Args: timings: List of timing dicts from estimate_phoneme_timings time_ms: Time in milliseconds Returns: Viseme code r-r.r,restr")r3r8Ztimingr"r"r#get_viseme_at_times   r:fpsc Cst|}|ds gS|d}|d}d|}g}d}d}||kr)NN)N)r;)__doc__rrpathlibrZgttsrZpydubrconfigrrrrdictr$r2listr5r7r:r>r"r"r"r#s   $. 1