sourav-das/stem-separator
3
1�
2 ��i� � �� � % S SK r S SKJr S SKJr S SKJr \ " S S5 5 r0 r\ \3\4 \ S' Sq \ R S- \ S' S \ R 4S4 jrS \5S \S- 4S jrS \6S
\\7 S\8S \4S jrS rg)� N)� dataclass)�StemSeparatorService)� file_managerc �r � \ rS rSr% Sr\\S' Sr\\S' Sr \\S' Sr9\ \\4 S- \S ' Sr \S- \S10' S r
g) � JobProgress� �queued�stateg �progresszWaiting in queue...�messageN�stems�error� )�__name__�11__module__� __qualname__�__firstlineno__r12 �str�__annotations__r �floatr r
�dictr �__static_attributes__r � �backend\task_queue.pyr r sE � ��E�3���H�e��(�G�S�(�#'�E�4��S��>�D� �'� �E�3��:� r r �jobs�_queue�returnc �D � [ c [ R " SS9q [ $ )N� )�maxsize)r �asyncio�Queuer r r � get_queuer# s � �
�~�� � �q�)�� �Mr �job_idc �, � [ R U 5 $ )N)r �get)r$ s r �get_job_progressr' s � � �8�8�F� �r r
�
output_formatc � � # � [ 5 nUR 5 ( a g[ 5 [ U ' UR XU45 I Sh v�N g N7f)z9Enqueue a separation job. Returns False if queue is full.FNT)r# �fullr r �put)r$ r
r( �qs r � enqueue_jobr- ! sD � � �� �A��v�v�x�x�� �=�D��L�13 �%�%��
�.�14/�/�/� � 0�s �AA�15A� Ac � � ^ # � [ 5 n [ 5 n UR 5 I Sh v�N u p#n [ R U5 m T c [ 5 m T [ U' S[16 S[ 4U 4S jjn[ R " U5 nUc( ST l ST l17 ST l UR 5 M� [ [ R " U5 5 n[ R " 5 nUR! SU R" [ U5 UUUU5 I Sh v�N n ST l S T l S18T l U T l UR 5 GM1 GN N6! [( aR n19[ R U[ 5 5 m ST l [ U205 T l21 S U22 3T l T [ U' Sn23A24NoSn25A26ff = f! UR 5 f = f7f) z:Single worker that processes separation jobs sequentially.Nr27 �pctc �b >� U Tl UTl SSSSS.nUR X S35 Tl g )Nz Loading BS-RoFormer model...zSeparating stems...zFinalizing output files...�Separation complete!)�
loading_model�28separating�29finalizing�donez...)r30 r r& r )r31 r/ �messagesr s �r �update_progress�$worker_loop.<locals>.update_progress8 s> �� �!&���$'��!�%C�"7�">� 2� �� $,�<�<����}�#E�� r r zInput file not foundzError: input file not foundr5 g �?r1 zError: )r r# r& r r r r r �get_input_filer32 r r � task_done�get_output_dirr! �get_event_loop�run_in_executor�separater r
� Exception) � separatorr, r$ r
r( r7 �33input_file�34output_dir�loop�result�er s @r � worker_looprF + s� �� � �$�&�I�� �A�35�-.�U�U�W�}�$��}�2 ��x�x��'�H�� �&�=�� '��V� �
F�s�
F��
F� &�4�4�V�<�J��!�!(���!7���#@�� ��8
�K�K�M�5 �\� 8� 8�� @�A�J� �)�)�+�D� �/�/���"�"��J���� � � � �F� $�H�N� #�H� � 5�H� � #�H�N�
�K�K�M�i �'4�:�� � $��x�x�� �
�6�H� $�H�N� ��V�H�N�!(���}�H� �#�D��L�� $��
�K�K�M�se �)G�E36�G�A-E �#G�5A!E �E
� E �7G�
E �37F+�AF&�!F. �&F+�+F. �.G � G)r! � dataclassesr �backend.separatorr �backendr r r r r r r r" r# r' �list�boolr- rF r r r �<module>rL s� �� � !� 2� � � � � � � "��d�3� � � � !� #���
�
�� � #��7�=�=� � �S� �[�4�%7� ��c� �$�s�)� �C� �D� �9r