EjB xdZddlZddlZddlmZddlZddlmZmZm Z m Z m Z m Z m Z mZmZmZdZdedefdZd edefd Zd ed edeeeeffd ZdeededefdZd ededeeeeffdZedk(rHddlZdZ ee d\Z!Z"Z#e$de!de#dde$de"e$e"jKddyy)zDGenerate voiceover audio + word-timed ASS subtitles from story text.N)Path) AUDIO_DIRSUB_COLOR_HIGHLIGHTSUB_COLOR_NORMALSUB_FONT SUB_FONT_SIZE SUB_MARGIN_VSUB_OUTLINE_COLORSUB_OUTLINE_SIZESUB_WORDS_PER_GROUP TTS_VOICEiticksreturnc|tz S)N)_TICKS_PER_SEC)rs /root/workspace/ytshorts/tts.py_ticks_to_secondsrs > !!tcztd|}t|dz}t|dzdz}|dz}|d|dd|dS)u3Convert float seconds → ASS timestamp H:MM:SS.ccgi<:02dz05.2f)maxint)rhmss r_seconds_to_assrsQ C A AIA QX" A BAS!C!E ##rtextvoicec \Ktj||dd}g}g}|j23d{}|ddk(r|j|d&|ddk(s/|j|dt |d t |d |d zd f7a6d j ||fSw) zAStream edge-tts and collect audio bytes + per-word timing events.z+8% WordBoundary)rateboundaryNtypeaudiodatar offsetduration)wordstartendr)edge_tts Communicatestreamappendrjoin)r r! communicate audio_chunkswordschunks r _synthesiser7&s&&tUXK "LE"))+  e =G #   f . 6]n , LL!&M.uX?,U8_uZ?P-PQ  + 88L !5 ((s+.B,BBB&B,7B,BB,r5total_durationc ~dtdtdtdtdtdt d }g}t dt|tD]}|j|||tzg}t|D]-\}}|dd}|d zt|kr||d zdd} n|d z} t|D]\} } | r| dn|} | d zt|kr || d zdn| } | dk(rd nd }g}t|D]g\}}tjd d |dj}|| k(r$|jdtd|dtdW|j|i|dj|z}|jdt!| dt!| d|0|dj|zdzS)a Group words into fixed-size groups and emit one ASS event per group. Within each event the current word is highlighted in yellow; the rest are white. Each word gets its own event overlapping the group so the highlight moves word-by-word. ar[Script Info] ScriptType: v4.00+ PlayResX: 1080 PlayResY: 1920 WrapStyle: 1 ScaledBorderAndShadow: yes [V4+ Styles] Format: Name, Fontname, Fontsize, PrimaryColour, SecondaryColour, OutlineColour, BackColour, Bold, Italic, Underline, StrikeOut, ScaleX, ScaleY, Spacing, Angle, BorderStyle, Outline, Shadow, Alignment, MarginL, MarginR, MarginV, Encoding Style: Default,,z ,&H000000FF,z#,&H64000000,-1,0,0,0,100,100,0,0,1,z ,0,2,40,40,z],1 [Events] Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text rr,g333333?z){\fscx80\fscy80\t(0,90,\fscx100\fscy100)}z[^\w\'\-\,\.\!\?]r+z{\cz&\fscx112\fscy112}z&\fscx100\fscy100} z Dialogue: 0,z,Default,,0,0,0,, )rrrr r r rangelenr r1 enumerateresubupperrr2r)r5r8headergroupsieventsg_idxgroup group_start group_endw_idxr+w_startw_endpoppartsjwclean line_texts r _build_assrV;s2 z=/+;*.mp3 AUDIO_DIR/.ass Returns (mp3_path, ass_path, audio_duration_seconds). z.mp3z.assz)edge-tts returned no word-boundary eventsr-g?zutf-8)encoding) rasynciorunr7r write_bytes RuntimeError write_textrV)r rWmp3_pathass_path audio_bytesr5r8s r generate_ttsrc|sdV4=(HdV4=(H [y%ABK % FGG2Yu%+N  5.9GL X~ --r__main__aI never expected my coworker to steal my lunch for three months. So I put a tiny tracker inside a sealed container of leftovers. At noon the signal led straight to my manager's office. He had been taking food because he thought nobody would say a word. I showed HR the location history and suddenly he had a lot to explain. The next day he brought in sandwiches for everyone. Nobody touched them.test_ttszAudio: z (z.1fzs)zASS: iX)&__doc__r[rBpathlibrr.configrrrrrr r r r r rrfloatrstrrtuplebyteslistdictr7rVrc__name__syssamplemp3assdurprint read_textrrrxs*J    "S"U"$u$$)C))eT$Z6G0H)*>-d4j>-%>-C>-B.s.#.%dE0A*B.0 z S !4MCc GC5C9B '( GC5/ #--/$3  r