Nj4"TUddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl Z ddl Z ddl m Z mZmZmZmZddlmZddlmZmZddlmZddlmZdd lmZmZmZmZm Z m!Z!m"Z"m#Z#dd l$m%Z%m&Z&dd l'm(Z(dd l)m*Z*ejVe,Z-e j\d dZ/da0e%dze1d<e dddZ2e2jgegdddgdge2jidedZ5e2jiddZ6dZ7de8d e9fd!Z:e2jwd"ed#ededfd$e8d%e8dzd&edzfd'Zd?*fd/ee8dze j|d@d1fd2ee8dze j|d3*fdAee?e j|dB*fdCeeBe j|dD*fdEeeBe j|dF*fdGeeBe j|dH*fdIee?e j|dJ*fdKee8e j|dL*fdMee8e j|dN*fdOee?e j|dP*fd4ee@e j|d5*ffdQZCe/j{ dZdRee8e jdS*fdTee8e jdU*fdd?*fd/ee8dze j|dVd1fd2ee8dze j|d3*ff dWZEe,dXk(re/yy)[N)Path)Queue)FastAPIFileForm HTTPException UploadFile)CORSMiddleware) HTMLResponseStreamingResponse) Annotated)stream_audio_chunks)DEFAULT_EOS_THRESHOLDDEFAULT_FRAMES_AFTER_EOSDEFAULT_LSD_DECODE_STEPSDEFAULT_NOISE_CLAMPDEFAULT_TEMPERATUREMAX_TOKEN_PER_CHUNKget_default_text_for_languageget_default_voice_for_language)TTSModelexport_model_state)enable_logging)_ORIGINS_OF_PREDEFINED_VOICESz2Kyutai Pocket TTS - Text-to-Speech generation toolF)helppretty_exceptions_show_locals tts_modelzKyutai Pocket TTS APIzText-to-Speech generation APIz1.0.0)title descriptionversion)zhttp://localhost:3000z&https://pod1-10007.internal.kyutai.orgzhttps://kyutai.orgT*) allow_originsallow_credentials allow_methods allow_headers/)response_classcKttjdz dz }|j}t t t j|jdtt t j}|Sw)zServe the frontend.staticz index.htmlDEFAULT_TEXT_PROMPT) r__file__parent read_textprintstrroriginreplacer) static_pathcontents `/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/pocket_tts/main.pyrootr5>smx.''(2\AK##%G #i  oo.FileLikeToQueuec||_yN)queue)selfrAs r4__init__z0write_to_queue..FileLikeToQueue.__init__Ts DJr:c:|jj|yr@rAput)rBdatas r4writez-write_to_queue..FileLikeToQueue.writeW JJNN4 r:cyr@r9rBs r4flushz-write_to_queue..FileLikeToQueue.flushZs r:c:|jjdyr@rErKs r4closez-write_to_queue..FileLikeToQueue.close]rIr:N)__name__ __module__ __qualname__rCrHrLrNr9r:r4FileLikeToQueuer>Ss  !  !r:rR) model_statetext_to_generateN)ioIOBasergenerate_audio_streamrconfigmimi sample_rate)rArTrSrR audio_chunkss r4write_to_queuer\PsT !")) !222B3L. i>N>N>S>S>_>_`r:rTrSc#Kt}tjt|||f}|j d} |j }|n |dz }||j yw)N)targetargsr)r threadingThreadr\startgetjoin)rTrSrAthreadirGs r4generate_data_with_staterhfsn GE  ^5BRT_:` aF LLN A yy{ <  Q  KKMsA*A,z/tts.text voice_url voice_wavc|js tdd|$|"tttj }|| tdd|t|j ds7|j ds&|j ds|tvs tdd tj|}tjd |n||jrt|jjnd }tjd | 5}|j j#}|j%||j'|j(}ddd tj+td}t-j.|n tddt1t3||ddddS#1swYhxYw#t-j.wxYw)aJ Generate speech from text using the pre-loaded voice prompt or a custom voice. Args: text: Text to convert to speech voice_url: Optional built-in voice name (e.g., "alba"), or voice URL (http://, https://, or hf://) voice_wav: Optional uploaded voice file (mutually exclusive with voice_url) izText cannot be empty) status_codedetailNz+Cannot provide both voice_url and voice_wavzhttp://zhttps://zhf://z5voice_url must start with http://, https://, or hf://zUsing voice from URL: %sz.wavF)deletesuffixT)truncateizThis should never happen.z audio/wavz)attachment; filename=generated_speech.wavchunked)zContent-DispositionzTransfer-Encoding) media_typeheaders)striprrr/rr0 startswithr"_cached_get_state_for_audio_promptloggingwarningfilenamerrptempfileNamedTemporaryFilefilereadrHrLnameget_state_for_audio_promptosunlinkr rh)rirjrkrSrp temp_filer3temp_file_paths r4text_to_speechrys ::<4JKKY.23y7G7G3HI !64abb   +##J/##G,99(_  BB9M 2I>  4=4F4Fi(()00F  ( (f Enn))+G OOG $ OO &^^N F &#>>tN?S^b>cK IIn %4OPP  {3#N!*  F E IIn %sAG" G G  G#hostzHost to bind to)rportzPort to bind toreloadzEnable auto-reloadlanguagezLanguage for the TTS model. 'english_2026-01', 'english_2026-04', 'english', 'french_24l', 'german_24l', 'portuguese', 'italian', 'spanish'. Incompatible with the config argument. Default is 'english', which is the same model as 'english_2026-04'.)r show_defaultrXzPath to locally-saved model config .yaml file. Incompatible with the language argument. If not provided, will use the default English model.quantizez.Apply int8 quantization to reduce memory usagecftj|||atjd|||y)zStart the FastAPI server.)rrXrzpocket_tts.main:web_app)rrrN)r load_modelruvicornrun)rrrrrXrs r4servers+8##XfxXI KK)4Or:z./tts_output.wavcpuzText to generatevoicezPath to audio conditioning file (voice to clone). Defaults to a built-in voice chosen from the language: 'giovanni' for italian, 'lola' for spanish, 'juergen' for german, 'rafael' for portuguese, 'estelle' for french, 'alba' otherwise.quietz-qz--quietzDisable logging outputaLanguage for the TTS model. 'english_2026-01', 'english_2026-04', 'english', 'french_24l', 'spanish_24l','german_24l', 'portuguese_24l', 'italian_24l'. Incompatible with the config argument. Default is 'english', which is the same model as 'english_2026-04'. The '24l' variants are bigger models, not distilled yet and here only as preview. They're not the final models for those languages.lsd_decode_stepszNumber of generation steps temperaturezTemperature for generation noise_clampzNoise clamp value eos_thresholdz EOS thresholdframes_after_eosz&Number of frames to generate after EOS output_pathzOutput path for generated audiodevicez Device to use max_tokensz#Maximum number of tokens per chunk.c *|rtjntj}td|5| t |}|dk(rt j j}|js+tjdtjdtj||||||| }|j| | t!|}|j#|}|j%||| | }t'| ||j(j*j,| dk7rtj/d | tj/d tj/d tj/d dddy#1swYyxYw) z(Generate speech using Kyutai Pocket TTS. pocket_ttsN-zNo input received from stdin.r`)code)rrXtemprrrr)rSrTrrzResults written in %sz--------------------zPIf you want to try multiple voices and prompts quickly, try the `serve` command.zRIf you like Kyutai projects, comment, like, subscribe at https://x.com/kyutai_labs)rxERRORINFOrrsysstdinr~ruloggererrortyperExitrrtorrrWrrXrYrZinfo)rirrrrXrrrrrrrrr log_levelrmodel_state_for_voicer[s r4generatersQz"' GLLI  i 0 <0:D 3;99>>#Dzz| LL8 9**!$ $''-#'   V =28"' GLLI  i 0''&I ::)D;  ; 4 1 0 0s 7A//A8__main__) localhosti@FNNF)FNN)FrUrxrrr{rapathlibrrArrrfastapirrrrr fastapi.middleware.corsr fastapi.responsesr r typing_extensionsr pocket_tts.data.audiorpocket_tts.default_parametersrrrrrrrrpocket_tts.models.tts_modelrrpocket_tts.utils.logging_utilsrpocket_tts.utils.utilsr getLoggerrOrTypercli_appr__annotations__web_appadd_middlewarerdr5r;r\r/dictrhpostrcommandOptionintboolrfloatrArgumentrr9r:r4rs   BB2='5   E9@   8 $ %++ =]b " 8d?!  !/NX_  %%  S. /  Y!!a,s& fS  J#':; ;Tz;D ;;| ALAEGL   -P C+<== >P C+<== >P dLELL.BCC DP d  z    P  d  l   P( lell PQQ)PPJ BF [`   ! L_J_ ! AF  sh C+=>> ?h  d  S    h  T<5<<i>VWW Xh  d  .  h <  d  l   =h J  \U\\; <<Kh P |u||!=>>Qh V5,%,,4G"HHIWh XULELLo$FFGYh Z  \U\\G HH[h ` \U\\@ AAah f c<5<<_== >gh h \U\\D EEih n lell PQQoh h `  \a  5$5 ^U^^!P QQ$534N OOP $5 T<5<<i>VWW X $5  d  >    $5(  d  l   )$5$5N z Ir: