NhanNguyen1309/audio-separation-api
0
1o
2 x[ j%9 � @ s" d dl mZ d dlZd dlmZ d dlmZ d dlmZm Z m3Z4m Z m Z m
Z
d dlmZ d dlmZ d dlmZ d d lmZ d d5lmZ d d lmZ d d lmZ d d
l m Z m Z d dl m Z d dl!m"Z"m#Z# d dl$m%Z% d dl&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1 d dl2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZA d dlBmCZC d dlDmEZEmFZFmGZGmHZHmIZI h d�ZJdZKd�dd �ZLed�d d ��ZMe d!d"eMd#�ZNeN�Oe"e#� eNjPee jQd$g d%�d&gd'� eNjRd(e7d)�d�d+d,��ZSd�d0d1�ZTd�d5d6�ZUd�d:d;�ZVd�d?d@�ZWd�dDdE�ZXd�dHdI�ZYd�dMdN�ZZd�dRdS�Z[d�dWdX�Z\d�d[d\�Z]d�dbdc�Z^eNj_dde8d)�e6de�e de�fd�dldm��Z`eNj_dne4d)�e7de�fd�dpdq��ZaeNjRdre5d)�d�dtdu��ZbeN�cdr�d�dwdx��ZdeNjRdye9d)�d�dzd{��ZeeN�fd|�d�d�d���ZgeN�Rd��d�d�d���ZheN�cdy�d�d�d���ZidS )�� )� annotationsN)�asynccontextmanager)�Path)�BackgroundTasks�FastAPI�File�Form�8UploadFile� WebSocket)�CORSMiddleware)� FileResponse)�WebSocketDisconnect)�chord_registry)�ensure_chord_duration)�check_chordino_available)�run_chord_detection_job)�cleanup_expired_chord_jobs�cleanup_expired_jobs)�settings)�ApiError�api_error_handler)�registry) �BpmCandidateRecord�ChordJobRecord�ChordSegmentRecord� JobRecord�KeyCandidateRecord�KeyDetectionResultRecord�KeyScoreComponentsRecord�SectionKeyResultRecord�SnapConfigRecord�VisualGridRecord�VisualGridCellRecord)�BpmCandidateResponse�ChordJobCreateResponse�ChordJobStatusResponse�ChordSegmentResponse�HealthResponse�JobCreateResponse�JobStatusResponse�TimeSignatureResponse�KeyCandidateResponse�KeyDetectionResultResponse�KeyScoreComponentsResponse�SectionKeyResultResponse�SnapConfigResponse�VisualGridResponse�VisualGridCellResponse)�run_separation_job)�ensure_file_size�ensure_supported_extension�ensure_supported_mime_type�ensure_supported_stem_mode�ensure_duration> � completed�failedZexpiredg >@�jobr �returnr) c C sF | j dkr d| j� d�nd }t| j| j | j| j| j| j| j| j|d� S )Nr8 z9/api/jobs/z /download) �id�status� stem_mode�progress�stage�10created_at�11updated_at�error� download_url) r= r< r) r> r? r@ rA rB rC )r: rD � rE �KD:\workspace\nextjs\music_toolkit_projects\audio-separation-api\app\main.py�build_job_status_response@ s �rG �_r c C s8 �t jjddd� t jjddd� t� t� d V d S )NT)�parents�exist_ok)r Z jobs_root�mkdirZchord_jobs_rootr r )rH rE rE rF �lifespanO s �12rL z AI Audio Stem Separation APIz0.1.0)�title�versionrL F)�GET�POST�DELETE�*)Z
allow_originsZallow_credentialsZ
allow_methodsZ
allow_headersz/health)�response_modelr' c C s t � } tdtj| j| j| jd�S )N�ok)r= � demucs_model�chord_detection_available� chord_engine�13chord_host)r r' r rU rV rW rX )�chordino_statusrE rE rF �healthg s �rZ �segmentr r& c C s t | j| j| j| jd�S )N)�startSec�endSec�chord�14confidence)r& � start_sec�end_secr^ r_ )r[ rE rE rF � build_chord_segment_responses s �rb � candidater r# c C s t | j| j| jd�S )N)�bpm�score�source)r# rd re rf �rc rE rE rF � build_bpm_candidate_response| s rh �cellr" r1 c C s$ t | j| j| j| j| j| j| jd�S )N)�indexZvisualBarIndexZ cellInBarr\ r] Z15isBarStartZ beatIndex)r1 rj Zvisual_bar_indexZ cell_in_barr` ra Z is_bar_startZ16beat_index)ri rE rE rF � build_visual_grid_cell_response� s �rk �gridr! r0 c C s t | jdd� | jD �d�S )Nc S � g | ]}t |��qS rE )rk )�.0ri rE rE rF �17<listcomp>� � z.build_visual_grid_response.<locals>.<listcomp>)ZvisualCellsPerBar�cells)r0 Zvisual_cells_per_barrq )rl rE rE rF �build_visual_grid_response� s �rr �18componentsr r- c 19 C sN t t| jd�t| jd�t| jd�t| jd�t| jd�t| jd�t| jd�d�S )N� )Z chromaScoreZ
chordFitScoreZtonicDurationScoreZfirstLastScoreZ cadenceScoreZ
bassRootScoreZsectionVoteScore) r- �roundZ chroma_scoreZchord_fit_scoreZtonic_duration_scoreZfirst_last_scoreZ
cadence_scoreZbass_root_scoreZsection_vote_score)rs rE rE rF �#build_key_score_components_response� s 20212223242526�rv r r+ c C s t | j| j| j| jt| j�d�S )N)�key�tonic�scalere rs )r+ rw rx ry re rv rs rg rE rE rF � build_key_candidate_response� s �rz �sectionr r. c C s t | j| j| j| j| jd�S )N)r\ r] rw ry r_ )r. r` ra rw ry r_ )r{ rE rE rF �build_section_key_response� s �r| �resultr r, c 27 C sF t | j| j| j| j| jdd� | jD �| jr dd� | jD �nd | jd�S )Nc S rm rE )rz �rn rc rE rE rF ro � rp z0build_key_detection_response.<locals>.<listcomp>c S rm rE )r| )rn r{ rE rE rF ro � rp )rw rx ry r_ �method�28candidatesZ sectionKeys�warnings) r, rw rx ry r_ r r� Z section_keysr� )r} rE rE rF � build_key_detection_response� s �r� �configr r/ c C s t | j| j| j| j| j| jd�S )N)ZchordSnapToleranceSecZdownbeatSnapToleranceSecZforwardDownbeatToleranceSecZ backwardDownbeatToleranceSecZvisualCellSnapToleranceSecZdownbeatSnapEnabled)r/ Zchord_snap_tolerance_secZdownbeat_snap_tolerance_secZ forward_downbeat_tolerance_secZ backward_downbeat_tolerance_secZ visual_cell_snap_tolerance_secZdownbeat_snap_enabled)r� rE rE rF �build_snap_config_response� s �r� r r% c C s� | j }td!i d| j�d| j�d| j�d| j�d| j�d| j�d| jr*dd � | jD �nd �d29|r2|j nd �d |r:|j30nd �d |rGd
d � |j D �nd �d|rO|j nd �d|rW|j
nd �d|r_|jnd �d|rnt|jj|jjd�nd �d|rv|jnd �d|r�t|j�nd �d|r�|jnd �d|r�|jnd �d|r�|jr�dd � |jD �nd �d|r�|jnd �d|r�|jnd �d|r�|jnd �d |r�|j r�t |j �nd �d |r�|j r�t |j �nd �d |r�|j n d �d | j!�d | j"��S �d | j!�d | j"��S )"Nr< r= r? r@ rA rB Zchordsc S rm rE �rb �rn r[ rE rE rF ro � rp z3build_chord_job_status_response.<locals>.<listcomp>rd Z
bpmConfidenceZrawBpmCandidatesc S rm rE )rh r~ rE rE rF ro � rp �beats� downbeatsZbarStartOffsetSecZ
timeSignature)� numerator� denominatorZ beatsPerBarZ31visualGridZ chordSourceZchordAudioSourceZkeyAnalysisChordsc S rm rE r� r� rE rE rF ro � rp ZkeyAudioSourceZkeyDetectionSourceZ32beatSourceZ33snapConfigZ originalKeyr� Z errorCoderC rE )#�analysisr% r< r= r? r@ rA rB r} rd Zbpm_confidenceZraw_bpm_candidatesr� r� Zbar_start_offset_secr* Ztime_signaturer� r� Z
beats_per_barrr Z visual_gridZ chord_sourceZchord_audio_sourceZkey_analysis_chordsZkey_audio_sourceZkey_detection_sourceZ beat_sourceZ snap_configr� Z original_keyr� r� Z34error_coderC )r: r� rE rE rF � build_chord_job_status_response� s� �������� �� ��35�� �� ��
����� ���� �� �� �� �� �� ���� �� � � � � � �r� �uploadr � destinationr �intc � sn �d}|� d��%} | �d�I d H }|sn|t|�7 }t|� |�|� q36W d � |S 1 s0w Y |S )Nr �wbTi )�open�read�lenr3 �write)r� r� �total�output�chunkrE rE rF �save_upload_with_limit� s � 37�38�39�40r� z /api/jobs.�background_tasksr �filer> �strr( c 41 � s| �|j s tdddd��t|j � t|j� t|�}tj||j d�}td|j � d|� d|j � �d d42� zvz.tj43|j d d d
d� t ||j �I d H t
|j � tj44|j dddd� td|j � �d d45� W n> tyy td|j � �d d46� t�|j � � ty� } ztd|j � d|� �d d47� t�|j � tdddd�|�d }~ww W |�� I d H n|�� I d H w | �t|j � t|j d�S )N�unsupported_file_type�A file name is required.� ��code�message� status_code)r> �input_filenamez[jobs] create job=z stem_mode=z48 filename=T)�flush�49processing�50 �51validating�r= r? r@ �queued� z[jobs] queued job=z[jobs] rejected job=z[jobs] invalid_audio job=z error=Z
invalid_audioz*The uploaded audio could not be processed.��job_id)�filenamer r4 r5 � content_typer6 r �52create_job�printr< �53update_jobr� �54input_pathr7 �55delete_job� Exception�close�add_taskr2 r( )r� r� r> Zsupported_moder: �excrE rE rF r� s8 �5657"58 ��� 59 r� z/api/chords/jobsr$ c 60 � s0 �|j s tdddd��t|j � t|j� t� }|js%t|jp dddd��tj |j d�}zVz%tj61|j d d62d d � t ||j
�I d H t|j
�}tj63|j d
dd
|d� W n' ty` t�|j � � tyy } zt�|j � tdddd�|�d }~ww W |�� I d H n|�� I d H w | �t|j � t|j d�S )Nr� r� r� r� Zchordino_unavailablez8Chordino chord detection is unavailable on this backend.i� )r� r� r� r� r� r� r� )r= r? r@ Z duration_secZchord_detection_failedz;The uploaded audio could not be queued for chord detection.)Z chord_job_id)r� r r4 r5 r� r rV rC r r� r� r< r� r� r r� r� r� r� r r$ )r� r� rY r: �durationr� rE rE rF �create_chord_job$ s< �6465�66 ��� r� z/api/chords/jobs/{job_id}r� c C � t �| �}t|�S �N)r �get_jobr� �r� r: rE rE rF �
get_chord_jobJ � 67r� �dict[str, bool]c C � t �| � ddiS �N�deletedT)r r� r� rE rE rF �delete_chord_jobP r� r� z/api/jobs/{job_id}c C r� r� )r r� rG r� rE rE rF r� V r� r� z/ws/jobs/{job_id}� websocketr68 �Nonec 69 � s70 �zt �| �}W n ty& } z|jd|jd�I d H W Y d }~d S d }~ww |�� I d H z/ |�t|�jdd��I d H |j t71v rO|jdd�I d H W d S t � t j
|j|jt�I d H }q0 ty{ } z|jd|jd�I d H W Y d }~d S d }~w ty� Y d S w )Ni� )r� �reasonT�json)�modei� )r� )r r� r r� r� �accept� send_jsonrG �72model_dumpr= �TERMINAL_JOB_STATUSES�asyncio� to_threadZwait_for_updater< rB � WEBSOCKET_WAIT_TIMEOUT_SECONDSr
)r� r� r: r� rE rE rF �73stream_job\ s8 ��� 7475��
$� �r� z/api/jobs/{job_id}/downloadr c C sR t �| �}|jdkrtdddd��|j�� s tdddd��t|jd |j� d76�d �S ) Nr8 Z
job_not_readyz"The job is not ready for download.i� r� Zdownload_missingz(The download file is missing or expired.i� zapplication/zipz.zip)�path�77media_typer� )r r� r= r �zip_path�existsr r< r� rE rE rF � download_jobz s 78798081�r� c C r� r� )r r� r� rE rE rF r� � r� r� )r: r r; r) )rH r )r; r' )r[ r r; r& )rc r r; r# )ri r" r; r1 )rl r! r; r0 )rs r r; r- )rc r r; r+ )r{ r r; r. )r} r r; r, )r� r r; r/ )r: r r; r% )r� r r� r r; r� )r� r r� r r> r� r; r( )r� r r� r r; r$ )r� r� r; r% )r� r� r; r� )r� r� r; r) )r� r� r� r82 r; r� )r� r� r; r )j�83__future__r r� �84contextlibr �pathlibr �fastapir r r r r r85 Zfastapi.middleware.corsr �fastapi.responsesr �starlette.websocketsr
Zapp.chord_jobsr Zapp.chord_validationr Z app.chordinor Z86app.chordsr Z app.cleanupr r Z87app.configr Z88app.errorsr r Zapp.jobsr Z89app.modelsr r r r r r r r r r! r" Z app.schemasr# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 Zapp.separationr2 Zapp.validationr3 r4 r5 r6 r7 r� r� rG rL �app�add_exception_handler�add_middlewareZ cors_origins�getrZ rb rh rk rr rv rz r| r� r� r� r� �postr� r� r� �deleter� r� r� r� r� r� rE rE rE rF �<module> s� 4D
90 �
91 92 9394 9596 97 98 99 100 101
102 103% � #� %