o lWi)@sddlZddlZddlZddlZddlmZddlmZddlm Z ddlm Z ddl mZddlmZeejdZeeZeGd d d ZeGd d d ZGd dde Z Gddde Z dS)N) dataclass)RequestOptions)AsyncMusicClient) MusicClient)'MusicComposeDetailedRequestOutputFormat) MusicPrompt.c@sBeZdZUeed<eed<ejeed<ejeed<eed<dS) SongMetadatatitle descriptiongenres languages is_explicitN)__name__ __module__ __qualname__str__annotations__typingListboolrr[/var/www/html/asistente-voz-ia/venv/lib/python3.10/site-packages/elevenlabs/music_custom.pyrs  rc@s2eZdZUejeejfed<eed<eed<dS)MultipartResponsejsonaudiofilenameN) rrrrDictrAnyrbytesrrrrrs  rceZdZdZdeeeedddejedejedeje deje dejej d d eje d e ffd d Zdejed e fddZZS)rz A client to handle ElevenLabs music-related functionality Extends the autogenerated client to include custom music methods N output_formatpromptcomposition_planmusic_length_msmodel_idrequest_optionsr!r"r#r$r%music_v1r&returncs"tj||||||d}||S)" Compose a song from a prompt or a composition plan with detailed response parsing. This method calls the original compose_detailed and then parses the stream response. Returns a MultipartResponse containing parsed JSON metadata, audio bytes, and filename. r )supercompose_detailed_parse_multipartselfr!r"r#r$r%r&stream __class__rrr+)s zMusicClient.compose_detailedr/c Cs6g}|D]}||qd|}|jddd}|d}|s#td|d}d}d } ttd t|D]D} d || vrz| d t|krz|| d } | rz| d rzzt | }Wnt j yw} zt d| WYd} ~ nd} ~ wwnq6ttdt|D]} d|| vrtd|| } | r| d} nq|d}d}d}tt|t|dD]} || | t||kr|dkr| }q|dkr| }nq|dkrtd|t|}|t|dkr||d kr||dd kr|d 7}n |d7}|t|dks||d}|stdt||| dS)a+ Reads a byte stream containing multipart data and parses it into JSON and audio parts. Args: stream: Iterator of bytes from ElevenLabs music API response Returns: MultipartResponse containing parsed JSON metadata, audio bytes, and filename utf-8ignoreerrors Empty response from music APIrNgenerated_music.mp3 Content-Type: application/json{z'Failed to parse composition plan JSON:  filename=filename="([^"]+)""Could not find audio part boundaryCould not parse JSON datarrrappendjoindecodesplit ValueErrorstriprangeminlen startswithrloadsJSONDecodeErrorloggererrorresearchgroupencoderr.r/chunkschunkresponse_bytes response_textlinesboundary json_datari json_lineematchboundary_bytesfirst_boundarysecond_boundary audio_start audio_bufferrrrr,Fsv            zMusicClient._parse_multipart)rrr__doc__OMITrOptionalrrrintLiteralrrr+Iteratorrr, __classcell__rrr0rr#0  rcr)rz An async client to handle ElevenLabs music-related functionality Extends the autogenerated async client to include custom music methods Nr r!r"r#r$r%r'r&r(cs*tj||||||d}||IdHS)r)r N)r*r+_parse_multipart_asyncr-r0rrr+s z!AsyncMusicClient.compose_detailedr/c sFg}|2z 3dHW}||q6d|}|jddd}|d}|s)td|d}d}d } ttd t|D]D} d || vr| d t|kr|| d } | r| d rzt | }Wnt j y}} zt d| WYd} ~ nd} ~ wwnqr?r@rArBrCrDrErFrYrrrrrsz           z'AsyncMusicClient._parse_multipart_async)rrrrjrkrrlrrrrmrnrrr+ AsyncIteratorrrrrprrr0rrrqr)rloggingrUr dataclassesrelevenlabs.core.request_optionsrelevenlabs.music.clientrAutogeneratedAsyncMusicClientrAutogeneratedMusicClientCelevenlabs.music.types.music_compose_detailed_request_output_formatrelevenlabs.types.music_promptrcastrrk getLoggerrrSrrrrrrs$       ~