hgT tddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddlm Z ddl mZddlZddlZddlZddlZddlmZmZddlmZddlZej4j7j9dj;dZej>d e Z d Z!d Z"d Z#ejHd6de%de%de%fdZ&dZ'de de%fdZ(de%de%de%fdZ)de%de%de%fdZ*dZ+de%fdZ,de%de%fdZ-d7de%de.d e%fd!Z/d"ej`d#e1dej`fd$Z2 d8d%Z3d&e%d'e%d(e%d)e%d*e4f d+Z5d,e6de%d'e%d(e%d-ee%d e%f d.Z7e jqejrjud/ejrjud0g1ejvd23 d9de%d4e1d e%d-e%fd5Z? ?$ 5 *d # q A /LA  tt!5affX>? ?%??!ooh377 \)D1 N  *' {.A   A 443AFF8<==J  LLj+I; 7 tt!5affX>? ?ffh 95 Z 6W#4  99; #c )<= = 1 # &7 }EFF  *' {2BC   A 449!&&BCCImmI&=)--*=G DEE N_ = =  s):M",M/"M,/M9r8out_dirc\tj|}|jddt|dz }|dddd}t |5}|j |d}ttj|j |jdcdddS#1swYyxYw)NT)parentsexist_okz%(id)s.%(ext)szmp4/best)outtmplformatquiet no_warnings)download.mp4)pathlibrmkdirrr extract_infoprepare_filename with_suffix)r8rbtmplydl_optsydlinfos r%download_videorusll7#G MM$M. w)) *D H 8 Qd37<< 4 4T :;GGOPQQQs AB""B+ video_pathcttj}ddd|dddddd t|d |d g }tj|d tj tj t |d |d S)Nr z-yz-iz-ac1z-ar16000z-c:a pcm_s16le/z.wavT)checkstdoutstderr)ruuiduuid4 subprocessrunPIPEr)rvrbr5cmds r% extract_audiors TZZ\ B      wiqD !" CNN3d:??:??S 7)1RD% &&r-2rLcrtj}d}|sCtjtd|dt j } |dr|S |sC|rty#t $rH}t|tjdtj|z tkD}Yd}~Xd}~wwxYw)NF /api/gen/z/aligned_lyrics/v2/rDaligned_lyrics?) rUrMrRrOr,r3rprintrWTIMEOUTrV)rLtimer is_timeoutlyricsr$s r% get_lyricsrs IIKEJ!j '*= >  $&  9&' ( "  9 !H JJsO))+-8J 9sA%% B6.>B11B6 style_genreciddddddddddd d d d d d dd dd dddddddddddd}|jj}||vr||Sd|d|dS)z Generate detailed tags that explicitly mention vocals and describe the style Args: style_genre: Basic genre like "drum and bass", "house music", etc. Returns: Detailed tag description mentioning vocals z drum and bassa This drum n bass cover launches straight into rapid-fire percussion and a driving bassline and clear vocals from the first second. Syncopated synth stabs and crisp hi-hats layer with energetic vocals, while doubled harmonies and subtle FX sweeps keep momentum high.z drum n bassz house musicaThis house music cover features a four-on-the-floor beat with clear vocals prominently mixed. Deep basslines and warm synth pads support the vocal performance, while classic house piano stabs and filtered sweeps create movement. The vocals remain crisp and upfront throughout.housedubstepaThis dubstep cover opens with clear vocals over sparse atmospherics before the massive sub-bass drops. Syncopated rhythms and wobbling basslines frame the vocal performance, while glitchy effects and dramatic builds maintain energy. The vocals cut through the heavy bass with clarity.jazzaThis smooth jazz cover features clear vocals supported by warm piano chords and walking basslines. Brushed drums and subtle horn sections complement the vocal melody, while sophisticated harmonies add depth. The intimate vocal delivery remains the centerpiece throughout.z smooth jazzlofia This lofi cover wraps the clear vocals in warm vinyl crackle and nostalgic atmospherics. Mellow hip-hop beats and dusty piano samples support the vocal performance, while subtle pitch variations add character. The vocals remain intelligible through the cozy production.z lofi beatszchill lofi beats orchestralaThis orchestral cover features clear vocals supported by sweeping string sections and dramatic brass. Timpani rolls and woodwind flourishes punctuate the vocal phrases, while the full orchestra swells during emotional peaks. The vocals soar above the symphonic arrangement.zepic orchestralaThis epic orchestral cover features clear vocals supported by sweeping string sections and dramatic brass. Timpani rolls and woodwind flourishes punctuate the vocal phrases, while the full orchestra swells during emotional peaks. The vocals soar above the symphonic arrangement. classicalaThis classical cover presents clear vocals with refined orchestral accompaniment. Elegant string arrangements and delicate woodwinds support the vocal line, while dynamic contrasts highlight the emotional content. The vocals maintain prominence throughout the sophisticated arrangement.zclassical symphonya'This classical symphony cover presents clear vocals with refined orchestral accompaniment. Elegant string arrangements and delicate woodwinds support the vocal line, while dynamic contrasts highlight the emotional content. The vocals maintain prominence throughout the sophisticated arrangement.z heavy metalaThis heavy metal cover features powerful vocals over crushing guitar riffs and thunderous drums. Palm-muted chugs and soaring lead guitars frame the vocal performance, while double-bass drumming drives the intensity. The vocals cut through the wall of sound with commanding presence.metalzThis z cover features clear and prominent vocals throughout. The instrumental arrangement supports and enhances the vocal performance while maintaining the characteristic elements of zV. The vocals remain crisp and intelligible, perfectly balanced with the backing track.)lowerstrip)rgenre_descriptions style_lowers r%generate_detailed_tagsrscec n h  r  b  i ` f l j t v H u o!(##%++-K((!+..{m$UVaUbbxy yr- audio_weightgenrec Bt|}|dDcgc]}|d }}dj|}tjtd|dt tjtd|ddd it |d t |d d d||dddddd}tjtd|t }|jstd|j|j} | jd} g} tj} tjdtjtd| t }|jstd|j|j} | rdT&\ >E >''%.K MMXJiy@(*U MM *IgY&89g  !&u-4  !- #(+    $  *$%,>  A 4409::vvxH \\$ FKJ  1 LLj3F8 ? ?ffh 3{+a/$QJ^^Hi8FEE(nnWb9  99; #c )56 6) ,$$0 =>> q>$ Q!5 55S ?s Harrwindowctjj|}|dkDrat||kDrStj||z }|dz}tj |||fd}tj ||d|| }|S)z,Smooth array while maintaining monotonicity.redge)modesame)npmaximum accumulateronespadconvolve)rrkernelrpaddeds r%smooth_monotonicrYsy **   $C zc#h'6)kc3Zf5kk&&v6sC4@ Jr-c~ t|d}tj|Dcgc]}|d c} tj|Dcgc]}|d c} tjdgtj dkD} | | t | t |}|rt|}|j} n d} ddkDr1tj dd tj d d d| kr/tj |  tj d fd } |j| j| } |r| j} | j|td |rd nd d |j|rjy y cc}wcc}w)aw Warp video timing based on timestamp pairs and optionally add audio. Args: video_path: Path to input video output_path: Path to output video timestamps: List of tuples (original_time, warped_time) in seconds audio_path: Optional path to audio file to add (unwarped) smooth_win: Window size for smoothing the time mapping c |dS)Nr)xs r%z,warp_video_with_timestamps..rs !A$r-)r?rrTư>rcDttj|S)z)Map from output time to input video time.)floatrinterp)toriginal_times warped_timess r%map_timez,warp_video_with_timestamps..map_timesRYYq,?@@r-libx264aacN)fpscodec audio_codecthreads)sortedrarrayappenddiffrrrdurationinsertfl_time set_duration set_audiowrite_videofileFPSclose)rv output_path timestampsr. smooth_winrmaskvideor>output_durationr warped_videorrs @@r%warp_video_with_timestampsrds 7JXXZ8qt89N88:6aQqT67L 99dVRWW\2T9 :D%L#D)N&njAN * %Ej)..&r*Ayyq#6 >1nQ6GHB/)yy? >>"3EFA ==*77HL#--e4    'ET ! KKM k96s F5 F:song_idmp4temp_dirrwarp_timestampsc : d|d}tjj|j|d}t j |}|j t|d5}|j|jddd t||jd|d||jd|dt|jd|dd |dtd d |dy#1swYlxYw#t$r} t| Yd} ~ nd} ~ wwxYw) Nrr@z_song_audio.mp3wbz/songify_video_rkr{)rvrrr.songify_video_zUploaded video to S3: )rpathrr rMrRraise_for_statusrSwritecontentrrrr&) rrrrrsong_urlrresponser[r$s r%generate_videor s%  .wit"? @A * *  !HH  s0A$C:'C.C:.C73C:: D DDoriginal_lyricsuuidsc  t||\}}t|}t|} d\dtdtfdfd} dtffd } | |} | |} | | }| | | }| | |}g}|j t j t||||d |f |j t j t||||d |f |D]}|j|D]}|jy) N)r)g333333?g@txtrctjddtjdz}t j d|j |jjS)Nru‘’“”"…—–NFKD) r maketransstring punctuation unicodedata normalize translaterr)rtbls r%_normz!create_remix_pairs.._normsPmmBF$6$69Q$QR$$VS1;;C@FFHNNPPr-c&|Dcgc]}|d }}|Dcgc]}|d }}t||d}g}dx}}|jD]\} } } } } | dk7rtt| | t| | D]|\}}||||}}|d|d}}|d|d}}||ks||kr/||z t ||z dz }|cxkrksPnS|j ||f|j ||f||}}~|s t d |Scc}wcc}w) NrQF)abautojunkgequalstart_send_sru)No anchors aligned – check transcripts.)r get_opcodesziprangemaxrr) old_alignment new_alignmentwv_wordss_wordssmanchorsprev_vsprev_sstagi1i2j1j2visjvwswvs0ve0ss0se0ratio RATIO_MAX RATIO_MINs r% get_anchorsz'create_remix_pairs..get_anchorssQ&341V944&341V944 w'E B  '#%>>#3 0 CRRg~eBmU2r]; 0B&r*M",=Bi="W+Si="W+S'>SG^sc#)T&::22NNC:.NNC:.'*CWG 0 0JK K/54s D  Drc j|dDcgc]}|dD]}|d|dizc}}Scc}}w)NrrrQr)rsectionrrs r%make_cleaned_lyricsz/create_remix_pairs..make_cleaned_lyricssX""23 ( VU4<01 1 1   s!/rtargetargsr) rrrdictr threadingThreadr startr)r rLrrrrremix1remix2 lyrics_remix1 lyrics_remix2r?rBclean_original_lyricsclean_remix_lyrics1clean_remix_lyrics2deduped1deduped2rthreadr=r>rs @@@r%create_remix_pairsrTs."'7NFFv&Mv&M#IyQ3Q3Q4 D 0@-m<-m<02EFH02EFHG NN!ha((S   NN!ha((S    r-z aws-bucketzjason-backend-api-key)secretsT)docs num_pairsc |rtj|nd}td|tj}td|j t ||j }t||j }tt|}t|}g} g} t|D]{} |s4tdD cgc]} ttj!} } n|| dz| dzdz} | | z } | jt!j"t$||||| |f}| D]}|j'| D]}|j)| Scc} w)Nzuuids: z Temporary directory created at: rrrC)r3loadsrtempfileTemporaryDirectoryr rurrarrr$rrrrrGrHrTrIr)r8rWrrrrr>rLr  all_videosri_ this_uuidsrSs r%songify_a_videor`sR"'DJJu DE GE7 **,H ,X]]O <= hmm ,C #x}} -E4;'G )OJG 9   5:1X>#djjl+>J>q1uA{3Jj    )%wXz5Q      !?s$E)N)rr)NrK)z:https://www.tiktok.com/@miryummi/video/7532607674991512862rrN)=r3rrlrrrZrGrUrrdifflibrrtypingrrdotenvnumpyrrMmoviepy.editorrryt_dlprmodalImage debian_slimpip_install_from_pyproject apt_installr Appapp S3_BUCKETrOr load_dotenvrr&r,rarurrrrrrndarrayintrrlistr rFrTfunctionSecret from_namefastapi_endpointr`rr-r%rws    # 7  KK 01[ eii*%8  1E#ESECE*TaTacaH Q Qc Qc Q'c'C'C'& $$y$y$yNM6#M6UM6M6`"**cbjjFGEPB BB&)B14BGKB6FF F F F 9 F  FR |, 67   L & && &  & &r-