)i5ddlmZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddl mZe GddZe ejZed z Zed z Zd;dZddZd?dZd@d!ZdAd#ZdBd&ZdCd+ZdDd-ZdEd0ZdFd7Z dGd9Z!e"d:kre#e!dS)H) annotationsN) dataclass)datetime)Path)AnycTeZdZUded<ded<ded<ded<dZded <dZd ed <dS) SegmentfloatstartendstrspeakertextN float | None confidence str | Nonesource)__name__ __module__ __qualname____annotations__rr#scripts/transcribe_with_voice_id.pyr r sXLLLJJJLLL III#J####Frr z_fw_transcribe.pyz_sb_voice_match.pycmd list[str]return subprocess.CompletedProcess[str]c tj|dd}|jdkr=td|jdd|d|jd|j|S) NT)capture_outputrrzCommand failed (z):  z stdout: z stderr: ) subprocessrun returncode RuntimeErrorjoinstdoutstderr)rresults rr#r#s ^C4 @ @ @F A (v0 ( (SXXc]] ( (  ( (  ( (   Mrpathrdict[str, Any]cxtjtdddddddt|gjS)Nffprobez-verrorz -show_streamsz -show_formatz-ofjson)r/loadsr#r r')r*s rr-r-*sK : D       r input_path output_pathchannelrr rr Nonec ddg}| |d|dgz }|dt|gz }|td||pdz }|d|dgz }|d kr|d d gz }n |d kr|d d gz }|dddddt|gz }t|dS)Nffmpegz-yz-ssz.3f-ig?z-tleft-afzpan=mono|c0=c0rightzpan=mono|c0=c1z-vnz-ac1z-ar16000)r maxr#)r1r2r3r r rdurations r extract_audior@;s T C  %~~&&D#j// ""C tSELS122 ''((& '(( G   '((E5#ugs;/?/? @@CHHHHHrr c tddddt|ddddd g }tjd |j}|sd St |d S) Nr6z -hide_bannerz-nostatsr7r: volumedetectz-fnull-z%mean_volume:\s*(-?\d+(?:\.\d+)?)\s*dBg8)r#r researchr(r group)r*r)matchs rmean_volume_dbrJQsu      II        F I> N NE v Q  rvalueslist[tuple[str, Path]]c&g}|D]}d|vrtd||dd\}}||t |f|S)N=zExpected NAME=PATH, got: rE) SystemExitsplitappendstripr expanduserresolve)rKpairsvaluenamer*s rparse_name_pathrXfs$&EHH e  @@@AA A[[a(( d djjllDJJ$9$9$;$;$C$C$E$EFGGGG Lrlist[tuple[str, str]]cg}|D]l}d|vrtd||dd\}}|||fm|S)NrNzExpected NAME=VALUE, got: rE)rOrPrQrR)rKrUrVrW raw_values rparse_name_valuer\ps#%E88 e  A%AABB B++c1--i djjllIOO$5$567777 Lrsecondsr ct|}t|d\}}t|d\}}|r |dd|dd|dS|dd|dS)Ni<02d:)intdivmod)r]totalhoursremminutessecss r format_tsrizsx LLEt$$JE33OOMGT 7666g666D6666 & & &D & & &&r audio_pathmodellanguagelist[dict[str, Any]]c tjddd5}t|j}dddn #1swxYwY t t jttt|t|d|d|gtj | d| d S#| d wxYw) Nfw_transcript_.jsonFprefixsuffixdelete--model --languageutf-8encodingT missing_ok) tempfileNamedTemporaryFilerrWr#sys executabler FW_HELPERr/r0 read_textunlink)rjrkrlhandle output_jsons rtranscribe_audiors  $,3y>>3z?? ;C55h $333344 :c#hho & &&rwindowsdict[str, dict[str, Any]]cptjddd5}t|j}dddn #1swxYwY |t j|ddtjttt|d t|g}|D]\}}|d |d |gz }t j t|j }d |d gD|dS#|dwxYw)Nvoice_windows_rpFrqindentrwrxz--windows-jsonrrNc:i|]}t|d|S)id)r ).0items r z(voice_match_segments..s4    T OOT   rresultsTrz)r|r}rrW write_textr/dumpsr~rr rr0r#r'getr) rjrrr windows_pathrrWrpayloads rvoice_match_segmentsrs  $   ) FK(( )))))))))))))))- 71 = = =PPP N  NN  OO      # 9 9ND( Md#7#7X#7#78 8CC*SXX_--   Ir22   t,,,, t,,,,s9==CDD5 output_stem source_pathmodenotessegments list[Segment]c |d}|d}t|tjd||d|Dd}|t j|dd d |jd d |dd|dg}|D]} | d| |gdz }|D]i} | j d| j ddnd } | dt| j d| j d| d| j| d j|d|dzd dS)N.transcript.json.transcript.mdr])timespecc \g|])}|j|j|j|j|j|jd*S)r r rrrrr)rsegs r z!write_outputs..sO    w;!n*       r)r created_atrrrrrrwrxu# Transcript — z - Source: ``z - Mode: `z- Note: )rz---rz (.2f)[z] **z**z:  ) with_suffixr rnow isoformatrr/rrWrQrrir rrr&rstrip) rrrrr json_pathmd_pathrlinesnoterscores r write_outputsrs''(:;;I%%&677Gk""lnn.. .BB        G"GA666III /K,.. $k$$$D  E (( &&&'''' ___E.1n.H*S^*****b U39--UU3;UU%UU38UUVVV R tyy''..0047'JJJJJrrbctjd}|dd|dd|dd d |d d d |ddd|ddgd|ddgd|dtdd|dtdd|}t |j}| std |t std!t|j r0t std!t|jr8t |jn|d"}t!|j }d#t#|jD}t'|}d$|d%gD}|std&||d'}t+|d(d)} t|d*id+p|d+pd,} g} g} d-} t-jd./5}t |}|d0z }t1|||jr| d1krd2} |d3z |d4z d5}|D]\}}t1|||6|D]\}}t7|}||jkr| |d7|d8d9=|||}d}|r||vrt?||}t|d:|j krItC|d;}t|d:}| |d<|d=|d>n/| |d?t|d:d>d@tE||j#|j$D]c}| tKt|dAt|dB|tC|dC||Ddn"tE||j#|j$}i}|r,|r*dEtM|D}tO|||}tM|D]\}}dF}d}|tC|}|r9t|d:}|}||j krtC|d;ndG}| tKt|dAt|dB|tC|dC|d-D| s| dHn/| d'kr)| dIj(| d1zkr| dJdddn #1swxYwY| )dKLtU||| | | tWdM|dNtWdM|dO| r&tWdP| D]}tWdQ|d'S)RNzATranscribe media files and optionally label speakers by voice ID.) descriptioninputzPath to input media file)helpz --output-stemz-Output stem path; defaults next to input fileruzsmall.enz-faster-whisper model name (default: small.en))defaultrrvenz-Language code for transcription (default: en)z--split-stereo store_truezDSplit stereo channels and transcribe each audible channel separately)actionrz--channel-labelrQz3Channel label mapping like left=Karl or right=Sarah)rrrrz4Reference voice clip mapping like Karl=refs/karl.wavz--voice-thresholdg?z8Cosine similarity threshold for voice ID (default: 0.45))typerrz--silence-threshold-dbgKz@Treat channels below this mean dB level as silent (default: -55)zInput file not found: zMissing helper: rci|]\}}|| Srr)rrWlabels rrzmain..,sZZZkdEdEZZZrcDg|]}|ddk|S) codec_typeaudio)r)rstreams rrzmain../s9  ::l # #w . .  . . .rstreamszNo audio stream found in rchannelsrEformatr?r8mono voice_id_)rrzmono.wavr split_stereozleft.wavz right.wav)r9r;)r3z channel treated as silent (z.1fz dB mean volume) best_score best_namez channel matched z at rz+ channel did not clear voice-ID threshold (rr r rrcg|]>\}}t|t|dt|dd?S)r r )rr r )r r )rindexrs rrzmain..ws\ $t "%jj!&tG}!5!5$T%[11rSpeakerUnknownz No transcript segments produced.zzmain..sDJ$,#Gr)keyzWrote rrzNotes:z- ),argparseArgumentParser add_argumentr parse_argsrrrSrTexistsrOr voice_refrrrrXr\ channel_labelr-rrbr|TemporaryDirectoryr@ritemsrJsilence_threshold_dbrQtitlervoice_thresholdr rrkrlr enumeraterr sortrprint)parserargsr1rrchannel_labels stream_info audio_streams audio_streamrr?rrr tmpdir_nametmpdirmono_wav channel_wavs channel_namewav_path channel_dbrrrIrtranscript_itemsmatchesrrrrs rmainrs  $WF &@AAA  <  <   <   S   B    C     G     O      Ddj!!,,..6688J     @>*>>???     97I77888 ~9i..0097I77888   (T ))++33555  # #B ' ' 4> * *DZZ5EdFX5Y5YZZZN*%%K!ooi44M CAZAABBB #L<##J2233H"%%))*55    J ' '  H E H D  $K 8 8 8XYKk""J&j(+++  N Q!D++-L+7*<*<*>*> J J& hj(LIIIII*6*<*<*>*>! ! & h+H55  999LL'eeZeeee(,,\<;M;M;O;OPP!  L>>,Xt<%?%?  +[[g[[:[[[ +yyX]^cdp^qXrXryyyy-Xtz4=QQ  DOO"'W "6"6 %d5k 2 2$+!$T&\!2!2'1#/      /! F 0$*dmTT 13G H( H (11A'B'B /xwGG()9::   t#!  CJJ//!% "566E!&J!D$888E+.///& #DM22!$u+.. ' f..#-%      Y LL; < < < < \\hrl.A== LLW X X XqXYXYXYXYXYXYXYXYXYXYXYXYXYXYXYt MMGGMHHH+z4AAA >;**+;<< > >??? @;**+=>> @ @AAA  h  D +t++     1s/NZ;;Z?Z?__main__)rrrr)r*rrr+)NNN) r1rr2rr3rr rr rrr4)r*rrr )rKrrrL)rKrrrY)r]r rr )rjrrkr rlr rrm)rjrrrLrr+)rjrrrLrrmrr) rrrrrr rrrrrr4)rrb)$ __future__rrr/rFr"r~r| dataclassesrrpathlibrtypingrr __file__rTparent SCRIPT_DIRrrr#r-r@rJrXr\rirrrrrrrOrrrrs5""""""  !!!!!!  T(^^ # # % % , , , - - (      ,!!!!*'''',,,,*''''----:*K*K*K*KZy y y y x z *TTVV  r