o 㥵im @sUddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z ddl mZmZddZedkreZejZedBed <edurqejZejZedur[gZnd d eDZedurigZn d d eDZngZgZ ejd d eeeDeejejej ej!ej"ej#ej$ej%ej&d Z'ed!ie'Z(ej)ej*ej+e(ej,dej$dej-dddZ.e.j/dkrpej$r/e0Z1ej2Z3e1j4e3ej5ej6ddZ7e4ej8ddZ9e9:ej5e9;e1z/e.j?ddD]Z@e@r e7Ae@e9Be@qe>se7CdZ>qWe7De1Ee9DdSe7De1Ee9Dwe.jFZGej8dejZHe4eHd ZIeIAeGWdn 1sQwYejJeHejdZKej rfe eKeLdeHddSeLd e.j/eLe.MdSdS)"N) AudioSegment)play)audio_to_bytes read_ref_text)ServeReferenceAudioServeTTSRequestcCstjdtjd}|jddtddd|jdd td d d |jd dtddd|jddtdddd|jddtdddd|jddtddd|jdtjd dd|jdtgd d!d"|jd#td$d$d%gd&d'|jd(td)d*d|jd+td,d-d|jd.td/d0d|jd1td2d3d|jd4td/d5d|jd6td7d8d|jd9td:d;d|jdd|jd?td@dAd@gdBd'|jdCtddDd|jdEtdFdGd| S)HNzCSend a WAV file and text to a server and receive synthesized audio.) descriptionformatter_classz--urlz-uzhttp://127.0.0.1:8080/v1/ttszURL of the server)typedefaulthelpz--textz-tTzText to be synthesized)r requiredr z--reference_idz-idzgID of the reference model to be used for the speech (Local: name of folder containing audios and files)z--reference_audioz-ra+zPath to the audio file)r nargsr r z--reference_textz-rtz"Reference text for voice synthesisz--outputz-ogenerated_audiozOutput audio file namez--playz*Whether to play audio after receiving data)actionr r z--format)wavmp3flacr)r choicesr z --latencynormalbalancedzUsed in api.fish.audio/v1/tts)r r rr z--max_new_tokensz2Maximum new tokens to generate. 0 means no limit.z--chunk_lengthi,zChunk length for synthesisz--top_pg?zTop-p sampling for synthesisz--repetition_penaltyg?z Repetition penalty for synthesisz --temperaturezTemperature for samplingz --streamingFzEnable streaming responsez --channelszNumber of audio channelsz--rateiDzSample rate for audioz--use_memory_cacheoffonz*Cache encoded references codes in memory. z--seedza`None` means randomized inference, otherwise deterministic. It can't be used for fixing a timbre.z --api_key YOUR_API_KEYzAPI key for authentication) argparseArgumentParserRawTextHelpFormatter add_argumentstrBooleanOptionalActionintfloatbool parse_args)parserr(D/home/ubuntu/.local/lib/python3.10/site-packages/tools/api_client.pyr&s  r&__main__idstrcCg|]}t|qSr()r).0 ref_audior(r(r) r/cCr,r()r)r-ref_textr(r(r)r/r0cCs(g|]\}}t|dur |nd|dqS)N)audiotext)r)r-r1r.r(r(r)r/s ) r4 references reference_idformatmax_new_tokens chunk_lengthtop_prepetition_penalty temperature streaminguse_memory_cacheseed)optionzBearer zapplication/msgpack) authorizationz content-type)datastreamheadersT)r7channelsrateoutputz.wavwbFr) chunk_size.)r7zAudio has been saved to 'z'.z Request failed with status code r()Nrbase64wave ormsgpackpyaudiorequestspydubrpydub.playbackrfish_speech.utils.filerrfish_speech.utils.schemarrr&__name__argsr6r+r!__annotations__reference_audio ref_audiosreference_text ref_texts byte_audiosr4zipr7r8r9r:r;r<r=r>r?rB pydantic_dataposturlpackbOPT_SERIALIZE_PYDANTICapi_keyresponse status_codePyAudioppaInt16 audio_formatopenrFrGrCrHwf setnchannels setsampwidthget_sample_size setframeratestream_stopped_flag iter_contentchunkwritewriteframesraw stop_streamclose terminatecontent audio_content audio_path audio_file from_filer3printjsonr(r(r(r)s    t