_LiZ%dZddlZddlZdZdZdZdZdZdZdZ dZ dZ dZ dZ d Zd Zd Zd Zd ZdZdZdZdefdZdZeeeeeee e e eeddefdedededededededededededed ed!ed"ed#ed$ed%ef$d&Zd'Zed(k(reyy))u)Auto-correct cue start / end timecodes by scanning a reference waveform. Primary algorithm — **hysteresis edge detection** (requires a clean voice-only stem, e.g. HTDemucs `voice_waveform.json`): - Cue START = silence → voice rising edge closest to the claimed start. - Cue END = voice → silence falling edge closest to the claimed end. Thresholds are derived dynamically from the search window's peak distribution so the detector auto-scales across quiet dialogue and shouted scenes. Fallback — **nearest local minimum** (for intra-phrase cue splits where the voice is continuous across the bound, so no rising/falling edge exists): - Smooth the peaks (box filter), enumerate local minima in the window, rank by depth + proximity to the claimed bound, return the best one. The fallback is what v1 used exclusively when the waveform was mixed audio and music made edge detection unreliable. Now that we read voice-only stems, edges are usable and preferred. Waveform structure: { "sampleRate": 100, "peaks": [0..1, ...] } at 10 ms steps. Nii xig333333?(g?c btdtt|dz |z}|dz}t|}dg|z}t |dt ||dz}t ||dz}t |D]<} ||z || <| |zdz} | |z } | |kr ||| z }|dz }| dk\s0||| z}|dz}>|S)Nr)maxintroundlensumminrange) peakssr smoothing_msswhalfNout window_sumcountiadd_irem_is >/home/booth/atri-vox-suite/vo-booth/scripts/autocorrect_cue.py _smoothedr6s QE,-234 5B 7D E A %!)CU1SD1H-./J 4!8 E 1X3e#AD1 D 19 %, &J  A: %, &J 3 Jc g}d}tt|dzdt|t|dz D]P}||||dz ks||||dzks!|||z |kr||||kr||d<|}>|j ||}R|S)Nr)rr rrappend)smoothi0i1 min_spacingminimaprevks r _local_minimar+Hs F D 3rAvq>3r3v;?#; <' !9va!e} $fQUm)CQX$<!9vd|+!"F2J1D MM! Qd ' Mr c rtt||z}tt||z} td|| z } tt ||| z} | | dzkryt |||} t| | | } | dkrytdtt|dz |z}t | | | |}| |z}|Dcgc] }| ||ks |}}|sytdtt|dz |z}d}td}|D]*}t||z }||z | z}| ||z}||ks'|}|},|Scc}w)Nrrrinf) r r r rrrr+floatabs)rrtarget_swindow_srmin_spacing_ms depth_factordistance_bias_mstarget_iwinr%r&r$ window_maxr' candidates depth_capr* bias_samplesbest best_scoredist dist_penaltyscores r_find_nearest_minrAUsi5B'(H eHrM" #C Q3 B SZC (B R!V| ub, /FVBr]#JQaU>D#82#=>?@Kvr2{;J\)I'B6!9 +A!BJB q#e$4t$;b$@ABCL DuJ )1x< |+z9 q L( : Jq ) KCs ; D4 D4g?ct|}|dz}||z }|}||kr9d} t|||zD]} || |ks | dz } | |k\r||z S|dz }||kr9||z S)u_Walk forward from onset_i; return number of samples until sustained silence (≥85% of `silence_min` samples below silence_thresh) appears. Used for duration-aware candidate filtering: a 200 ms burst at the snap point is almost certainly an onomatopoeia / breath / interjection when the cue's expected line duration is 2 seconds. 333333?rr)rr) rronset_isilence_thresh silence_minr threshold end_limitrsilentjs r_voice_burst_duration_samplesrKs E Ad"IKIA i-q!k/* AQx.(!   Y w;  Q i- w;r c |||}|syt|}|tt|dz tt|dz}t d|t z}t d|t z}||fS)N)g{Gz?gQ?rg?g{Gz?g{Gz?)sortedrrr r DEFAULT_VOICE_FACTORDEFAULT_SILENCE_FACTOR)rr%r&windowsorted_wp90 voice_threshrEs r_dynamic_thresholdsrTss 2b\F f~H 3s8}q(#c(mc.A*BC DCtS#778L|.DDEN  ''r c \"tt||z"td"tt||zz } tt |"tt||zz} | | z dkryt || | \} } tdtt|dz |z} tdtt|dz |z}g}t| |z|}t| | z t || z }t ||D]m}d}t ||| zD]}||| kDs |dz }|| dzkr0d}t ||z |D]}||| ks |dz }||dzkr]|j|o|syd}||dkrt|"fd  }nr||z}d}td }d }|D]R}t|||| |}t|"z |z }|dkDr||z nd }||kr ||z |zdz}nd }||z} | |ksM| }|}|}T||krd}|}!|!dkDr!||!dz | kDr|!dz}!|!dkDr ||!dz | kDr|!|fS)uFind silence → voice transition index closest to target_s within the asymmetric window [target - back_s, target + forward_s]. Returns (onset_idx, low_confidence) tuple or (None, False) if no candidate. Two-pass detection: 1. Anchor scan — collect ALL positions where sustained voice follows sustained silence (robust against mid-phrase peaks / HTDemucs bleed). 2. Candidate selection: - Without expected_duration_s: pick anchor closest to target (legacy). - With expected_duration_s: score each anchor by (distance from target + onomatopoeia penalty if its voice-burst duration is much shorter than the cue's expected line length). Rejects grunts/breaths/ad-libs that would otherwise win on proximity alone. If no candidate clears the duration floor, the lowest-cost (closest plausible) is returned with low_confidence=True. 3. Onset snap — from the chosen anchor, walk BACKWARD through every non-silent sample to find the true first audible sample of that voice burst. Removes the "lands late" bias on gradual ramps. r)NFrrgffffff?FNc t|z S)N)r0)cr6s rz#_find_rising_edge..sSX->r )keyr.r g?g@T) r r r rrrTrr#r/rKr0)#rrr1back_s forward_s voice_min_mssilence_min_msexpected_duration_sduration_floor_ratior%r&rSrE voice_minrFr9lohir voice_countrJ silence_countlow_confidencer<expected_samplesr= best_ratioanchorburst distance_sratio penalty_sr@onsetr6s# @r_find_rising_edgeros05B'(H Q3uVb[122 3B SZCi"n(=$>> ?B Bw|#6ub"#E L.As5!4r!9:;D#82#=>?@KJ R+ { +B R)^SZ)3 4B 2r] q!i-( !AQx,&q  ! S (  q;* #AQx.("  # ;, , !" N"&9Q&>:#>?.35\    #F1r6>;EVh./"4J0@10DE,,#E++1E9=PPSVV  *Ez!" " # #* , ,!N E !)eai(>9   !)eai(>9 .  r c tt||z}td|tt||zz }tt ||tt||zz} | |z dkryt ||| \} } tdtt|dz |z} tdtt|dz |z} d}t d}t|| }t| | z t || z }t||D]{}d}t||| zD]}||| ks |dz }|| dzkr0d}ttd|| z |D]}|||kDs ||}|| krdt||z }||ksx|}|}}|yt |}|}||kr||| kDr|dz }||kr ||| kDrtd|| z }||kDr!||dz | kr|dz}||kDr ||dz | kr|S) uFind where voice gives way to sustained silence, closest to target_s. Mirror of `_find_rising_edge` for the trailing boundary: 1. Anchor scan — find a position where sustained silence follows recent voice activity. Voice endings TAPER (loud → quiet → silent over ~500 ms), so there's no sharp edge. Anchor detects: (a) sustained silence of `silence_min_ms` starting at index i, and (b) voice activity somewhere in the preceding `voice_lookback_ms` window. Max-based lookback tolerates taper — the phrase can be quiet at i-1 as long as it was loud earlier in the lookback. 2. Offset snap — from the anchor, refine so the returned index is exactly "one past the last audible sample". The 85%-silent rule in the anchor tolerates a few stray tail peaks inside [i, i+silence_min], and the anchor may land a few samples after the true tail because the silence needed to sustain. Snap: (a) walk FORWARD while peaks[i] > silence_thresh — captures any straggler tail peaks the 85% rule allowed past the anchor; (b) walk BACKWARD while peaks[i-1] ≤ silence_thresh — pulls back past any silent samples inside the anchor region so the result lands immediately after the last audible peak. The back-walk is bounded to the lookback window so it can't cross a real silence gap into a previous voice burst. rrVNrrr.rCr ) r r r rrrTr/rr0)rrr1r[r\r^voice_lookback_msr6r%r&rSrErFvoice_lookbackr< best_distrbrcrrerJprev_maxr>roffset back_limits r_find_voice_endrwsQ65B'(H Q3uVb[122 3B SZCi"n(=$>> ?B Bw|#6ub"#E L.aU>D#82#=>?@KC&7$&>&C DEFN De I R B R+ s5zK7 8B 2r]' q!k/* #AQx.("  # ;- - s1a.0115 $AQx(" 8 $ l " 1x<  ) IQd''* | E A F 1*v7!  1*v7 Q~-.J : % "3~"E!  : % "3~"E Mr r waveformoriginal_start_soriginal_end_s start_back_msstart_forward_ms end_back_msend_forward_msr]r^rqrr3r4r5gate_thresholdr_r`returnct|jdxsd}|d}|r|dkDr|Dcgc] }||k\r|nd }}|dz }|dz }|dz }|dz }||zdz }t||||||||| \}}|d nd}|t||||| | | | }|d nd }t ||||||| }|d nd}|t||||| | | | }|d nd }|||z nd}|||z nd}||n|} ||n|}!|!| krAt | |z }"t |!|z }#|"|#kDr |} |r|d nd }n |}!|r|d nd }|!| kr|} |}!||| |!tt | |z dztt |!|z dz|du|dut|||d||||||| | | | | |||dd Scc}w)N sampleRatedrrr g@@g@)r_r` rising_edge local_minnone voice_end _revertedr)startendr{r|r}r~r]r^rqrr3r4r5rr_r`) original_start original_endcorrected_start corrected_enddelta_start_ms delta_end_ms onset_found offset_foundrfmethodparams)r getrorArwr0r bool)$rxryrzr{r|r}r~r]r^rqrr3r4r5rr_r`rrp start_back_sstart_forward_s end_back_s end_forward_sfallback_window_s start_idxstart_low_conf start_methodend_idx end_methodrr final_start final_end start_delta end_deltas$ rautocorrect_cuer[s X\\, ' .3 /B W E .1,)s2BB#/L&/O!/J$/M%73>!2 r#on/1 !I~%.$9=tL%eR1ACT&2N&24DF '0&;{ eR(-,.?AG!( 3J#E2~?P$0.$02BD%,$7[V *3*?y2~TO&-&9Wr\tM%4%@/FVK%2%@-nIK+(889 )67  "*K9El^956L&I5?J%At$KLM)5'5~.(<* 0&,(,!2(,( 0,#6$8  wCsFc6tjtjd}|j d|j dt d|j dt d|j dt t |j d t t |j d t t |j d t t |j d t t |j dt t |j dt t |j dt t |j dt t |j dt t |j dt t" |j dt dd|j dt dd|j dt t$d|j'}t)|j*5}t-j.|}dddt1|j2|j4f|j6|j8|j:|j<|j>|j@|jB|jD|jF|jH|jJ|jL|jN|jPd}tSt-jT|dy#1swYxYw)Nr) descriptionrxz--startT)typerequiredz--endz--start-back-ms)rdefaultz--start-forward-msz --end-back-msz--end-forward-msz--voice-min-msz--silence-min-msz--voice-lookback-msz--smoothing-msz--min-spacing-msz--depth-factorz--distance-bias-msz--gate-thresholdr z4pre-zero peaks below this value (0..1); 0 = disabled)rrhelpz--expected-duration-szucue's expected line duration in seconds; enables duration-aware candidate scoring that rejects onomatopoeia / breathsz--duration-floor-ratiozfreject candidate bursts whose duration is less than this fraction of expected_duration_s (default 0.4)rr )indent)+argparseArgumentParser__doc__ splitlines add_argumentr/r DEFAULT_START_BACK_MSDEFAULT_START_FORWARD_MSDEFAULT_END_BACK_MSDEFAULT_END_FORWARD_MSDEFAULT_VOICE_MIN_MSDEFAULT_SILENCE_MIN_MSDEFAULT_VOICE_LOOKBACK_MSDEFAULT_SMOOTHING_MSDEFAULT_MIN_SPACING_MSDEFAULT_DEPTH_FACTORDEFAULT_DISTANCE_BIAS_MSDEFAULT_DURATION_FLOOR_RATIO parse_argsopenrxjsonloadrrrr{r|r}r~r]r^rqrr3r4r5rr_r`printdumps)apargsfwfresults r_clirsv  W-?-?-A!-D EBOOJOOIEDO9OOGEDO9OO%s>SOTOO(s>VOWOOOs>QOROO&s>TOUOO$s>ROSOO&s>TOUOO)s>WOXOO$s>ROSOO&s>TOUOO$u>ROSOO(s>VOWOO&ucOQOO+%QROO,5B^@A ==?D dmm  YYq\  DJJ((..$$**&&**00&&**&&..** 44!66F" $**VA &''s LL__main__)rrrrrrrrrrrNrOrrrrrr+rArrKrTrorwdictr/r rr__name__r rrs.   $ P #.(+/+Gb!JMj*?,D':*@(<*@-F(<*@*>,D,/152NldlelUl#&l&)l"%l%( l #& l %( l(+l#&l%(l#(l'*l%*l*/l+0lTXl^,(^ zFr