Vh%ddlZddlZddlmZddlZddlmZddlZddl Z ddl m Z m Z ddlmZmZmZddlmZddlmZmZmZd Zd Zd d Zed kr edSdS)N)Path)fatal) apply_model BagOfModels) AudioFile convert_audio save_audio)HTDemucs)get_model_from_argsadd_model_flagsModelLoadingErrorc"i}d} t|d||}n*#t$rd|d<Yntj$rd|d<YnwxYw|_ t jt|\}}t||||}n'#t$r}|j d|d<Yd}~nd}~wwxYw|Vtd|d| D]\}}td |d |tjd |S) Nr)streams sampleratechannelszFFmpeg is not installed.ffmpegzFFmpeg could not read the file. torchaudiozCould not load file z+. Maybe it is not a supported file format? zWhen trying to load using z, got the following error: r)rreadFileNotFoundError subprocessCalledProcessErrortaloadstrr RuntimeErrorargsprintitemssysexit) trackaudio_channelsrerrorswavsrerrbackenderrors U/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/demucs/separate.py load_trackr+s F C=##!#$%% 6665x  (===<x= { Egc%jj))GC R^DDCC / / /#&8A;F<  /  { :U::: ; ; ;$llnn \ \NGU ZwZZSXZZ [ [ [ [  Js,%,AAA$B B4B//B4ctjdd}|ddtgdt ||dd d |d d ttdd|ddd|ddt jrdndd|ddtd|ddtd | }|d!d"d#d$d%&|d'td()|d*d+d,d-.| }|d/d d01|d2d d31|d4d5d5d6gd78| }|d9d d:1|d;d d<1|d=d>td?|d@tdAdBtdAdCD|dEdFdGtdH|S)INzdemucs.separatez)Separate the sources for the given tracks) descriptiontracks+zPath to tracks)nargstypedefaulthelpz-vz --verbose store_true)actionz-oz--out separatedzVFolder where to put extracted tracks. A subfolder with the model name will be created.)r1r2r3z --filenamez{track}/{stem}.{ext}zSet the name of output file. Use "{track}", "{trackext}", "{stem}", "{ext}" to use variables of track name without extension, track extension, stem name and default output file extension. Default is "{track}/{stem}.{ext}".)r2r3z-dz--devicecudacpuz4Device to use, default is cuda if available else cpuz--shiftsrzNumber of random shifts for equivariant stabilization.Increase separation time but improves quality for Demucs. 10 was used in the original paper.)r2r1r3z --overlapg?zOverlap between the splits.z --no-split store_falsesplitTzDDoesn't split audio in chunks. This can use large amounts of memory.)r5destr2r3z --segmentzISet split size of each chunk. This can help save memory of graphic card. )r1r3z --two-stemsstemSTEMz/Only separate audio into {STEM} and no_{STEM}. )r;metavarr3z--int24zSave wav output as 24 bits wav.)r5r3z --float32z'Save wav output as float32 (2x bigger).z --clip-moderescaleclampziStrategy for avoiding clipping: rescaling entire signal if necessary (rescale) or hard clipping (clamp).)r2choicesr3z--flacz Convert the output wavs to flac.z--mp3zConvert the output wavs to mp3.z --mp3-bitratei@zBitrate of converted mp3.z --mp3-presetzOEncoder preset of MP3, 2 for highest quality, 7 for fastest speed. Default is 2)rAr1r2r3z-jz--jobsrziNumber of jobs. This can increase memory usage but will be much faster when multiple cores are available.) argparseArgumentParser add_argumentrr thr7 is_availableintfloatadd_mutually_exclusive_grouprange)parser split_groupgroup format_groups r* get_parserrQ5s  $%61\^^^F $IYZZZF k,??? ! $[ 1 1? @@@   6=>>> "*,'*>*>*@*@ KeSUUU  ! 1222   $":<<<5577K\$1")%)#E FFF [s#KLLL  #VNPPP  / / 1 1E y=??? {<EGGG  y9gBVQRRR6688Lh|#EGGGgl#DFFF  # 8::: a #q6777 h ! QRRR Mc 4 t}||} t|}n1#t$r$}t |jdYd}~nd}~wwxYwt d}t|trt |j }nt|tr|j }|j |j |krt d|t|tr%tdt|jd|||jO|j|jvrAt d|jd|j|j|jz }|d d td ||jD]}|s td |d t6j7td|t;||j|j}| d} || z}|| !z}tE||d|j#|j$|j%|j&d |j'|j  d} | | !z} | | z } |j(rd} n |j)rd} nd} |j|j*|j+|j,|j-|j.rdndd} |jt_| |jD]\} }||j0|j1ddd|j1ddd|| z }|j2d d tg| ti|fi| tk| } ||j0|j1ddd|j1ddd|j| z }|j2d d tg| 6|j7|jti|fi| tqj9| d}| D]}||z }||j0|j1ddd|j1dddd|jz| z }|j2d d tg|ti|fi| dS)NrinfzbCannot use a Transformer model with a longer segment than it was trained for. Maximum segment is: zSelected model is a bag of z8 models. You will see that many progress bars per track.zMerror: stem "{stem}" is not in selected model. STEM must be one of {sources}.z, )r<sourcesT)parentsexist_okz#Separated tracks will be stored in zFile zp does not exist. If the path contains spaces, please try again after surrounding the entire path with quotes "".)filezSeparating track )deviceshiftsr:overlapprogress num_workerssegmentmp3flacr%)rbitratepresetclipas_floatbits_per_sample.r)r"trackextr<extno_):rQ parse_argsr rrrrJ isinstancer r^rmax_allowed_segmentrlenmodelsr8evalr<rUformatjoinoutnamemkdirresolver.existsr stderrr+r#rmeanstdrrYrZr:r[jobsr_r` mp3_bitrate mp3_preset clip_modefloat32int24zipfilenamersplitparentr rlistpopindexrG zeros_like)optsrMrmodelr)rorur"r%refrUrkkwargssourcervr< other_stemis r*mainrs \\F   T " "D#D))  ejm ,,%""8#EM22 E; ' '8#7 |DL3F$F$F T>QTT U U U%%%A @C ,=,=@@@ A A A IIKKK JJLLL y%-!?!? [ b bY %-(@(@ c B B C C C (TY CIIdTI*** ? ? ?@@@7878||~~  WWWWZ ! ! ! !  )%))*** 4e6FGGhhqkk sxxzz swwyyeSYt{4;$(J t*.)T\KKKKLN 37799388:: 8 CC Y CCC*'oN %)Z7rrR    9  #GU] ; ; 8 8 T]11 8I8I#q8Q8QRS8T;@:;L;LSRS;T;TUW;X7;2FFF !!$!>>>63t99777777  87mmG--EJ4E4Ec14M4Ma4P7rs& ++++++++7777777777OOOOOOOOOO>HHHVW8W8W8W8t zDFFFFFrR