o 0iM=@sdZddlmZddlZddlZddlmZddlmZm Z ddl Z ddl m Z mZmZmZmZmZmZmZmZmZmZmZmZddlmZdd lmZmZmZm Z m!Z!dd l"m#Z#e$e%Z&Gd d d Z'dS) zHigh-level TTS interface for Supertonic. This module provides the main TTS class for easy text-to-speech synthesis with automatic model loading and voice style management. ) annotationsN)Path)OptionalUnion) AVAILABLE_LANGUAGESAVAILABLE_MODELSDEFAULT_LANGUAGEDEFAULT_MAX_CHUNK_LENGTHDEFAULT_MAX_CHUNK_LENGTH_KO DEFAULT_MODELDEFAULT_SILENCE_DURATION DEFAULT_SPEEDDEFAULT_TOTAL_STEPSMAX_TEXT_LENGTHMAX_TOTAL_STEPSMIN_TOTAL_STEPSis_multilingual_model)Style) get_cache_dir list_available_voice_style_names load_modelload_voice_style_from_json_fileload_voice_style_from_name) chunk_textc@steZdZdZeddddfd.d dZd/ddZd0ddZee de e dfd1d$d%Z d2d*d+Z ee de e dfd1d,d-ZdS)3TTSaHigh-level interface for Supertonic text-to-speech synthesis. Args: model: Model name to use ("supertonic" or "supertonic-2"). Default is "supertonic-2" (multilingual support). model_dir: Directory containing model files. If None, uses default cache directory based on model name. auto_download: If True, automatically downloads model files from HuggingFace Hub if they're missing intra_op_num_threads: Number of threads for intra-op parallelism. If None (default), ONNX Runtime automatically determines optimal value based on your system. Can also be set via SUPERTONIC_INTRA_OP_THREADS environment variable inter_op_num_threads: Number of threads for inter-op parallelism. If None (default), ONNX Runtime automatically determines optimal value based on your system. Can also be set via SUPERTONIC_INTER_OP_THREADS environment variable Attributes: model (supertonic.core.Supertonic): The underlying Supertonic engine model_name (str): Name of the loaded model model_dir (pathlib.Path): Path to the model directory sample_rate (int): Audio sample rate in Hz voice_style_names (list[str]): List of available voice style names is_multilingual (bool): Whether the model supports multiple languages Example: ```python from supertonic import TTS # Use default model (supertonic-2 with multilingual support) tts = TTS() style = tts.get_voice_style("M1") wav, dur = tts.synthesize("Hello!", voice_style=style, lang="en") # Use specific model version tts_v1 = TTS(model="supertonic") # English only tts_v2 = TTS(model="supertonic-2") # Multilingual ``` NTmodelstr model_dirOptional[Union[Path, str]] auto_downloadboolintra_op_num_threads Optional[int]inter_op_num_threadscCs|tvrtd|ddt||_t||_|dur!t|}t|ts*t|}t ||||||_ ||_ |j j |_ t ||_dS)aInitialize the TTS engine. Args: model: Model name ("supertonic" or "supertonic-2"). Default: "supertonic-2" model_dir (Union[Path, str]): Directory containing model files. If None, uses default cache directory based on model name auto_download: If True, automatically downloads missing model files intra_op_num_threads: Number of threads for intra-op parallelism. If None (default), ONNX Runtime automatically determines optimal value based on your system. Can also be set via SUPERTONIC_INTRA_OP_THREADS environment variable inter_op_num_threads: Number of threads for inter-op parallelism. If None (default), ONNX Runtime automatically determines optimal value based on your system. Can also be set via SUPERTONIC_INTER_OP_THREADS environment variable zInvalid model: 'z'. Available models: , N)r ValueErrorjoin model_nameris_multilingualr isinstancerrrr sample_raterZvoice_style_names)selfrrr r"r$r-M/home/kim/smarthome/.venv/lib/python3.10/site-packages/supertonic/pipeline.py__init__Ts     z TTS.__init__ voice_namereturnrcCs t|j|S)a-Load a voice style by name. Avaliable voice style names can be listed with `list_available_voice_style_names()`. Args: voice_name: Name of the voice style (e.g., 'M1', 'F1', 'M2', 'F2') Returns: Style object containing voice style vectors )rr)r,r0r-r-r.get_voice_styles zTTS.get_voice_stylevoice_style_pathUnion[Path, str]cCst|S)zLoad a voice style from a JSON file path. Args: voice_style_path: Path to the voice style JSON file (str or Path) Returns: Style object containing voice style vectors )r)r,r3r-r-r.get_voice_style_from_paths zTTS.get_voice_style_from_pathFtext voice_style total_stepsintspeedfloatmax_chunk_lengthsilence_durationlangverbosetuple[np.ndarray, np.ndarray]c Cs|r|s td|jr!|tvrtd|ddt|} n|dkr3|r3td|jd|dd } |rUtd t|d |jrLtd |n td |jdt|tkrhtdt|dtdt |t sxt dt |j d|tkr|tkstdtdtd|d|dkrtd||jj|\} } | stdt| d| |d ur| dkrtnt}t||} |rtdt| dt| dkrt| d d D]\} }td!| dd"|d d#t|d#krd$nd%qt| d krtd&t| d d'td(|d)|d*d+|jd,g}g}t| D]]\} }|r@td-| dd.t| d/d%d0d1td2| dd.t| ||g|||| \}}|ritd3|dd*d4|jddkrytd5|j||||q't|dkrt|dksJd6tjdt||jftj d7}g}t|D]\} }||| t|dkr||qtj!|dd8}t"|}|t|d}||}|r|jd}td9td:|dd*d;td<|d=td>|j||fS)?uSynthesize speech from text. This method automatically chunks long text into smaller segments and concatenates them with silence in between. Args: text: Text to synthesize voice_style: Voice style object total_steps: Number of synthesis steps (default: 5) speed: Speech speed multiplier (default: 1.05) max_chunk_length: Max characters per chunk. If None, automatically determined based on language (300 for most, 120 for Korean) silence_duration: Silence between chunks in seconds (default: 0.3) lang: Language code for synthesis. Supported languages: - "en": English (default) - "ko": Korean - "es": Spanish - "pt": Portuguese - "fr": French verbose: If True, print detailed progress information (default: False) Returns: Tuple of (waveform, duration): - waveform: Audio array of shape (1, num_samples) - duration: Total duration in seconds Example: ```python tts = TTS() style = tts.get_voice_style("M1") wav, dur = tts.synthesize("Hello, world!", voice_style=style, lang="en") wav_ko, dur_ko = tts.synthesize("안녕하세요!", voice_style=style, lang="ko") print(f"Generated {dur[0]:.2f}s of audio") ``` zText cannot be emptyzInvalid language: 'z'. Supported languages: r%enu⚠️ Model 'z"' is English-only. Ignoring lang='z'.Nu📝 Input text length: z charactersu🌐 Language: u 🌐 Model: z (English only)z Text length (z") exceeds maximum allowed length (z.). Please split your text into smaller chunks.z(voice_style must be a Style object, got z(. Use get_voice_style() to load a style.ztotal_steps must be between z and z, got z,. Higher values = better quality but slower.rz+silence_duration must be non-negative, got zFound z unsupported character(s): koz Split into z chunk(s)rzChunk z: <z...z... and z more chunk(s)z&Synthesizing audio... Settings: steps=z, speed=z.2fzx, sample_rate=ZHzz [/z] Processing chunk... T)endflushzProcessing chunk u✓ (zs)z%Expected wav shape (1, samples), got zNo audio generated)Zdtype)ZaxiszGeneration complete!zTotal duration: szTotal samples: ,z Array shape: )#stripr&r)rr'printr(lenrr*r TypeErrortype__name__rrrZtext_processorZ validate_textr r r enumerater+loggerdebugshape RuntimeErrorappendnpZzerosr9Zfloat32Z concatenatesum)r,r6r7r8r:r<r=r>r?Zeffective_langZis_valid unsupportedZ text_chunksichunkZwav_listZdur_listZ text_chunkwavZdur_onnxZsilenceZarrays_to_concatZwav_catZtotal_audio_durZtotal_silence_durZdur_catZ total_samplesr-r-r. synthesizes /     6$  $   zTTS.synthesizer\ np.ndarray output_pathNonec Cszddl}Wnty}z tdtd|d}~wwt|}|jjdddt|jtj s9t d|jt d|| t |||jt d dS) zSave synthesized audio to a WAV file. Args: wav: Audio waveform array from synthesize() output_path: Path where to save the WAV file rNzsoundfile not installedzSsoundfile library is required to save audio. Install it with: pip install soundfileT)parentsexist_okz#No write permission for directory: zSaving audio to zAudio saved successfully)Z soundfile ImportErrorrRerrorrparentmkdirosaccessW_OKPermissionErrorinfowriterZsqueezer+)r,r\r_ZsfeZoutput_path_objr-r-r. save_audio>s$  zTTS.save_audioc Cs|j||||||||dS)uShorthand for synthesize(). Allows using tts(...) instead of tts.synthesize(...). Args: text: Text to synthesize voice_style: Voice style object total_steps: Number of synthesis steps (default: 5) speed: Speech speed multiplier (default: 1.05) max_chunk_length: Max characters per chunk. If None, automatically determined based on language (300 for most, 120 for Korean) silence_duration: Silence between chunks in seconds (default: 0.3) lang: Language code for synthesis (default: "en"). Supported: "en", "ko", "es", "pt", "fr" verbose: If True, print detailed progress information (default: False) Returns: Tuple of (waveform, duration): - waveform: Audio array of shape (1, num_samples) - duration: Total duration in seconds Example: ```python tts = TTS() style = tts.get_voice_style("M1") wav, dur = tts("Hello, world!", voice_style=style, lang="en") wav_ko, dur_ko = tts("안녕하세요!", voice_style=style, lang="ko") print(f"Generated {dur[0]:.2f}s of audio") ``` )r6r7r8r:r<r=r>r?)r]) r,r6r7r8r:r<r=r>r?r-r-r.__call___s'z TTS.__call__) rrrrr r!r"r#r$r#)r0rr1r)r3r4r1r)r6rr7rr8r9r:r;r<r#r=r;r>rr?r!r1r@)r\r^r_rr1r`)rP __module__ __qualname____doc__r r/r2r5rrr r r]rnror-r-r-r.r,s4) ,   (%r)(rr __future__rloggingrgpathlibrtypingrrnumpyrWconfigrrr r r r r rrrrrrcorerloaderrrrrrutilsr getLoggerrPrRrr-r-r-r.s  <