o 0i:@sdZddlmZddlZddlZddlZddlZddlmZddl m Z m Z ddl m Z eeZdd Zd d Zd d ZddZddZddZdddZddZedkr[edSdS)zCommand-line interface for Supertonic TTS. This module provides a command-line interface for easy text-to-speech synthesis, batch processing, and model management. ) annotationsN) __version__)AVAILABLE_MODELS DEFAULT_MODEL)TTSc Cs|jr tjtjddntjtjddzddl}Wnty4tdtdtdt dYnw|jrEtd |j dd d z td |j d t }t |j d}t |}td|dd|jrtdt }|j j|j \}}|j j|j }t |} td| ddtd|j ddt|j dkrd nd|std|dd||j krtd|ddt|dkrd ndtd|jp|jd t } |jr||j} n||j} t | } td| ddtd|jd t } |j|j | |j|j|j|j|j|jd\}}t | }td|ddtd |ddd!|||j|td"WdSt y{}ztd#||jrkt!"d$t dWYd}~dSd}~ww)%z;Generate speech and play it directly without saving a file.%(levelname)s: %(message)slevelformatrNu9❌ Error: sounddevice is required for the 'say' command.z4 Install it with: pip install supertonic[playback]z Or: pip install sounddevicer🎤 Generating speech: 2...Loading model ()...model -> Model loaded in .2fsProcessing text... -> Text processed in .3f Original: P ⚠️ Unsupported chars:  Preprocessed: Loading voice style ( -> Voice style loaded in Generating speech (lang= voice_styleZ total_stepsspeedmax_chunk_lengthsilence_durationlangverbose -> Speech generated in zPlaying z s audio...z -> Audio played ❌ Error: z#TTS playback failed with exception:)#r(logging basicConfigINFOWARNINGZ sounddevice ImportErrorprintsysexittextrtimertext_processor validate_text_preprocess_textlencustom_style_pathvoiceget_voice_style_from_pathget_voice_styler' synthesizestepsr$r%r&ZplayZsqueeze sample_ratewait Exceptionlogger exception)argssd load_starttts load_time text_startis_valid unsupported preprocessed text_time style_startr# style_time start_timewavduration elapsed_timeerUH/home/kim/smarthome/.venv/lib/python3.10/site-packages/supertonic/cli.pycmd_says     . $    rWc Cs|jr tjtjddntjtjdd|jr$td|jdddztd|jdt}t |jd }t|}td |d d |jrtd t}|jj |j\}}|jj |j}t|}td|dd td|jddt |jdkrdnd|std|dd||jkrtd|ddt |dkrdndtd|jp|jdt} |jr||j} n||j} t| } td| dd td|jdt} |j|j| |j|j|j|j|j|jd\} }t| }td|d d td|dd d|jd|| |jtd|jWdSty\}ztd||jrLtd td!WYd}~dSd}~ww)"z$Generate speech from text using TTS.rr r Nr rrrrrrrrrrrrrrrrrr r!r"r)zSaving rz s audio to z -> Audio saved to r*z%TTS generation failed with exception:r)r(r+r,r-r.r0r3rr4rr5r6r7r8r9r:r;r<r'r=r>r$r%r&outputZ save_audiorArBrCr1r2)rDrFrGrHrIrJrKrLrMrNr#rOrPrQrRrSrTrUrUrVcmd_ttsmsp   . $    rYc Cs~zt}|j}tdt|d|D] }td|qWdSty>}ztd|tdWYd}~dSd}~ww)zList available voice styles.u📢 Available voice styles (z): u • r*rN)rvoice_style_namesr0r8rAr1r2)rDrGstylesstylerTrUrUrVcmd_list_voicessr]c Csz&t}tdtd|jtd|jdtdd|jWd StyE}ztd|tdWYd }~d Sd }~ww) zShow model information.u%ℹ️ Supertonic Model Information zModel directory: z Sample rate: z Hzz Available voice styles: z, r*rN) rr0Z model_dirr?joinrZrAr1r2)rDrGrTrUrUrVcmd_infosr_c Cszddlm}m}tdz|}||td|WdSty<}ztd|tdWYd}~dSd}~ww)z Download model from HuggingFace.r)download_model get_cache_diru$📥 Downloading Supertonic model...u✅ Model downloaded to: u❌ Download failed: N)loaderr`rar0rAr1r2)rDr`ra cache_dirrTrUrUrV cmd_downloadsrdcCstdtdS)zShow version information.z supertonic N)r0r)rDrUrUrV cmd_versionsrereturnargparse.ArgumentParserc Cstjddtjdd}|jddd}dd }|jd d d }|jd dd |jdtttdtd|jdddd|jdtddd|jdtdgddd|jdt dd d|jd!t d"d#d|jd$t dd%d|jd&t d'd(d|||j t d)|jd*d+gd,d-}|jd d.d |jd/d0d1d2d3|jdtttdtd|jdddd|jdtddd|jdtdgddd|jdt dd d|jd!t d"d#d|jd$t dd%d|jd&t d'd(d|||j t d)|jd4d5gd6d-}|jd d.d |jd/d0d1d2d3|jdddd|jdt dd d|||j t d)|jd7d8gd9d-}|j td)|jd:d;gdgd?d-}|j td)|jd@dAgdBd-} | j td)|S)CzCreate and return the CLI argument parser. This function is separated to allow documentation generation tools to extract CLI arguments automatically. Returns: ArgumentParser configured with all Supertonic CLI commands Z supertonicz2Supertonic - High-quality Text-to-Speech synthesisu Examples: # Generate and play speech directly (no file saved) supertonic say 'Hello, welcome to the world!' # Generate speech from text and save to file supertonic tts 'Hello, welcome to the world!' -o output.wav # Use different voice and quality supertonic say 'This is a female voice style.' --voice F1 --steps 10 supertonic tts 'This is a female voice style.' -o hello.wav --voice F1 --steps 10 # Multilingual support (Korean, Spanish, Portuguese, French) supertonic say '안녕하세요! 반갑습니다.' --lang ko supertonic tts 'Bonjour le monde!' -o french.wav --lang fr supertonic tts 'Hola, bienvenido!' -o spanish.wav --lang es # Use custom voice style from JSON file supertonic say 'This is a custom voice test.' --custom-style-path ./my_voice.json # Long text with custom chunking supertonic tts 'This is a very long text.' -o output.wav --max-chunk-length 200 # List available voices supertonic list-voices )prog descriptionformatter_classepilogcommandzAvailable commands)desthelpcSs|jddddddS)Nz-vz --verbose store_truez+Enable verbose output with detailed logging)actionrn) add_argument)prUrUrVadd_common_argss  z&create_parser..add_common_argsZsayz:Generate speech and play it directly without saving a file)rnr3zText to synthesize and playz--modelzQModel to use: supertonic (English only) or supertonic-2 (multilingual). Default: )typedefaultchoicesrnz--voiceZM1zVoice style (default: M1))rurnz--custom-style-pathNzDPath to custom voice style JSON file (overrides --voice if provided))rtrurnz--langen)rwkoesptfrzaLanguage code: en (English), ko (Korean), es (Spanish), pt (Portuguese), fr (French). Default: enz--stepsz)Quality steps (default: 5, higher=better)z--speedg?z4Speech speed (0.7-2.0, default: 1.05, 2.0=2x faster)z--max-chunk-lengthz>Maximum characters per chunk (default: auto based on language)z--silence-durationg333333?z0Silence between chunks in seconds (default: 0.3))funcrGtzGenerate speech from text)aliasesrnzText to synthesizez-oz--outputTzOutput WAV file)requiredrnr=rz/(Deprecated: use tts) Generate speech from textz list-voiceslvzList available voice stylesinfoizShow model informationdownloaddzDownload model from HuggingFaceversionvzShow version information)argparseArgumentParserRawDescriptionHelpFormatteradd_subparsers add_parserrqstrrrintfloat set_defaultsrWrYr]r_rdre) parserZ subparsersrsZ parser_sayZ parser_ttsZ parser_synthZ parser_voicesZ parser_infoZparser_downloadZparser_versionrUrUrV create_parsers          rc Cst}|}|js|tdz||WdSty.tdtdYdSt yL}ztd|tdWYd}~dSd}~ww)zMain CLI entry point.ru ⚠️ Interrupted by userr*N) r parse_argsrl print_helpr1r2r}KeyboardInterruptr0rA)rrDrTrUrUrVmains  r__main__)rfrg)__doc__ __future__rrr+r1r4rrconfigrrZpipeliner getLogger__name__rBrWrYr]r_rdrerrrUrUrUrVs,    XN >