NhanNguyen1309/audio-separation-api
0
1o
2 ��j� � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl3m Z d dl m
Z
d dlmZ dd d4�dd d d
d�d�Zd&dd�Zd'dd�Zd(d d �Zd)d!d"�Zd*d$d%�ZdS )+� )� annotationsN)�Path)�settings)�ApiError)�registry)� JobRecord�5vocals.wavzinstrumental.wav)r z
no_vocals.wav� drums.wav�bass.wav� other.wav)r r r6 r )�2-stemz4-stem�jobr �return� list[str]c C s� t jdddtjdt| j�g}tjdkr |�dttj�g� |�dttj�dttj �g� | j7d kr6|� d8� |� t| j �� |S ) Nz-mZdemucsz-nz-or z-jz --overlapz--shiftsr z--two-stems=vocals)
�sys�9executabler � demucs_model�str�10output_dir� demucs_jobs�extend�demucs_overlap�
demucs_shifts� stem_mode�append�11input_path)r
�command� r �QD:\workspace\nextjs\music_toolkit_projects\audio-separation-api\app\separation.py�_demucs_command s �12 13 1415r � source_namer r c C sF t | j }| j�|�D ]}|j|v r|�� r|j S q tdddd��)N�separation_failedz7Demucs finished but expected stem files were not found.�� ��code�message� status_code)�EXPECTED_STEMSr r �rglob�name�is_file�parentr )r
r �expected� candidater r r �_find_demucs_output4 s 1617��r. �demucs_stem_dir�list[tuple[Path, str]]c C sR g }t | j �� D ] \}}|| }|�� s tdd|� �dd��|�||f� q |S )Nr! z Missing expected output stem: r" r# )r' r �items�existsr r )r
r/ �18stem_pathsr �19clean_name� source_pathr r r �_collect_expected_stemsA s �r6 r3 �Nonec C sV t j| jdt jd��}|D ] \}}|j||d� q
W d � d S 1 s$w Y d S )N�w)� compression)�arcname)�zipfile�ZipFile�zip_path�20ZIP_STORED�write)r
r3 �archiveZ stem_pathr4 r r r � _create_zipQ s21 �"�rA �job_idc 22 C s� t �| �}t�� }z�td| � d�dd� t j| dddd� td| � d d23�t|��� �dd� t j| d d d
� t�� }tj t|�ddt24j dd�}|j dkrw|j
�� pW|j�� pWd}td| � d|j � d|dd � � �dd� td|dd � dd��td| � dt�� | d��dd� t j| dd d
� t�� }ttt|j ��}t||�}t||�} t|| � td| � d t�� | d�d t�� | d�d |j� �dd� t j| d d!d d� W d S tjy� td| � d"�dd� t j| d#d!d#d$d%� Y d S t�y }25 z td| � d&|26j� �dd� t j| d#d!d#|27jd%� W Y d }28~29d S d }30~31w t�yI }32 z td| � d&|33� �dd� t j| d#d!d#t |34�d%� W Y d }35~36d S d }37~38ww )'Nz[separation] job=z stage=preparing_audioT)�flush�39processing� �preparing_audio)�status�progress�stagez stage=separating command=� � Z40separating)rH rI F)�capture_output�text�timeout�checkr zDemucs separation failed.z stage=failed returncode=z error=i ���r! i���r" r# z" stage=packaging separate_seconds=z.2f�U Z packagingz# stage=completed packaging_seconds=z total_seconds=z zip=� completed�d z stage=failed error=timeout�failedzAudio separation timed out.)rG rH rI �errorz stage=failed error=) r �get_job�time� perf_counter�print�41update_job�joinr �42subprocess�runr �processing_timeout_seconds�43returncode�stderr�strip�stdoutr �next�iterr' r r. r6 rA r= �TimeoutExpiredr% � Exceptionr ) rB r
Z44started_atZseparate_started_at�resultr% Zpackaging_started_atZexpected_first_outputr/ r3 �excr r r �run_separation_jobW sb 45"�46("4748490� �$�&��rh )r
r r r )r
r r r r r )r
r r/ r r r0 )r
r r3 r0 r r7 )rB r r r7 )�50__future__r r[ r rV r; �pathlibr �51app.configr �52app.errorsr �app.jobsr �53app.modelsr r' r r. r6 rA rh r r r r �<module> s0 ���545556
57