PCNUSMSE/transcript_service
0
1o
2 E��h}^ � @ s� d Z ddlZddlZddlmZ ddlmZmZmZm Z m3Z4 ddl Z ddl
ZddlmZ ddlmZmZmZ ddlmZ dd lmZ G d5d � d �Ze� Zd efd
d�Zd e j fdd�Z dS )uo Gradio用户界面模块6 7提供基于Gradio的Web界面，支持文件上传、进度显示和结果展示。8� N)�Path)�Dict�List�Optional�Tuple�Any� )�9get_config)�get_task_manager�10TaskStatus� TaskPriority)�get_task_logger)�get_file_validatorc @ s e Zd ZdZdd� Zdejfdd�Zdefdd�Z de11fd d12�Z defd d �Z d
e
dede
e dedededee dB dededededeee13ef fdd�Zdeee14f fdd �Zdedee15e
eee16f fd d �Zdeee17ee18e
e19ef fd d �Zdefd!d"�Zd#d$� Zd%d&� ZdS )'�GradioInterfaceu Gradio界面管理器c C sF t � | _t� | _t� | _tdd�| _d| _| � � | _20| j� | j � dS )u 初始化Gradio界面ztranscript_service.gradio)� logger_nameN)
r �configr21 � task_managerr �file_validatorr
�logger�current_task_id�_create_interface� interface�add_status_callback�_on_task_status_change��self� r �I/Users/chengpeng/MyProject/transcript_service/src/api/gradio_interface.py�__init__ s 22zGradioInterface.__init__�returnc C sV | j �� }tjdtj�� dd����}t�d� t�� ��T tjdd��� tj ddt23|d �d24d �}t�� � tj d d
dgdd�}W d � n1 sKw Y tj ddd���}tj
dg d�ddgd�}t�� � tjddd�}tjd dd�}tjd dd�} W d � n1 s�w Y t�� � tjd ddd d d!d"�}25tjd#d$d%d&�} W d � n1 s�w Y tj d'dd��" tjd(d)d*d&�} tjd+d)d,d&�}
tjd-d)d.d/d0d1�}W d � n1 s�w Y W d � n1 s�w Y t�� � tjd2d3d4d5�}tjd6d7d8�}tjd9d7d8�}W d � n 1 �sw Y W d � n 1 �s*w Y tjd.d��+ tjd:d;dd<gd=�}tjd>d?d@dAddBgdC�}tjg dD�g dE�dFddG�}W d � n 1 �sdw Y W d � n 1 �stw Y tj dHdd���}t�� �� t�dI�� tjdJ| �� d�}tjdK|d�}W d � n 1 �s�w Y t�dL�� tjdMi d�}W d � n 1 �s�w Y t�dN�� tjdOi d�}W d � n 1 �s�w Y t�dP�� tjdQdRd@dddS�}tj dTddU�}W d � n 1 �sw Y W d � n 1 �sw Y W d � n 1 �s(w Y t�� � tjdVd7dWd5�} | j| j|||||||gdX� W d � n 1 �sTw Y |j| j|||||| |26| | |
|g |||gdY� |j| j||gdX� |j| j|||||||gdX� |j | j|||||||gdX� W d � |S 1 �s�w Y |S )Zu 创建Gradio界面u 音频转文字服务a0 27 .main-container { max-width: 1000px; margin: 0 auto; }28 .upload-area { border: 2px dashed #ccc; border-radius: 10px; padding: 20px; text-align: center; }29 .result-area { margin-top: 20px; }30 .status-simple { font-size: 16px; font-weight: bold; }31 )�title�theme�cssu # 🎵 音频转文字服务� )�scaleu, 📁 选择音频文件（支持多文件）�multiple�32extensions�x )�label�33file_countZ34file_types�heightu 优先级)u 普通�NORMAL)u 高优先级�HIGHr+ )r( �choices�valueNu ⚙️ 转录参数设置F)�openu 识别语言))u 中文�zh)u 英文�en)u 日语�ja)u 粤语Zyue)u 韩语�ko)u 德语�de)u 法语�fr)u 俄语�rur0 r1 u 过滤语气词T)r( r. u 时间戳校准u 说话人分离u 说话人数量（可选）�d � uA 留空则自动判断，如需指定请输入2-100之间的数值)r( r. �minimum�maximum�step�infou 音轨索引�0u0 多音轨文件的音轨索引，用逗号分隔)r( r. r< u 高级选项u 热词ID v2� u v2模型的热词IDu 热词ID v1u v1模型的热词IDu 敏感词过滤配置r u JSON格式配置u 敏感词过滤的JSON配置)r( r. �lines� placeholderr< u 🚀 开始转录�primaryZlg)�variant�sizeu35 ❌ 取消Z secondary)rB u 🗑️ 清空u 📊 当前状态� 等待上传文件...z
status-simple)r( r. � interactive� elem_classesu 📝 转录结果u! 转录结果将在这里显示...� � z result-area)r( r@ r? � max_lines�show_copy_buttonrF )u 文件名u 时长u 文本长度u 置信度)�strrK �numberrL u 📈 处理统计)�headers�datatyper( �visibleu 📋 详细信息u 系统信息� 服务状态u 支持格式u 任务信息u 当前任务u 完整结果u36 JSON结果u 处理日志u 详细日志� )r( r? rI rE rJ u 下载日志文件)r( rO u 🔄 刷新状态�sm)�fn�outputs)rS �inputsrT ) r �get_supported_formats�gr�Blocks�themesZSoft�Markdown�Row�Column�File�listZRadioZ AccordionZ
CheckboxGroupZCheckbox�NumberZTextboxZButtonZ DataframeZTabsZTab�JSON�_get_system_info�click�_update_interface�_process_files�_cancel_current_task�_clear_interface�load) r Zsupported_formatsr Z file_uploadZpriority_selectZparams_sectionZlanguage_select�disfluency_removal�timestamp_alignment�diarization_enabled�
speaker_countZchannel_select�
vocabulary_id� phrase_id�special_word_filterZ start_btnZ37cancel_btnZ clear_btn� status_text� result_textZstats_dfZdetail_section� system_info� format_info� task_info� result_json�log_textZ log_downloadZ refresh_btnr r r r & sv 38� 39 40�41� � �42 43��� �44�� ���� �� �45? �� �W��46� �� �t47 �� � 48� � � � �� �� �� �49(� ��� ��� �50� M �� Mz!GradioInterface._create_interfacec C s dS )u 获取自定义CSS样式a 51 .gradio-container {52 max-width: 1200px !important;53 }54 .gr-button-primary {55 background: linear-gradient(45deg, #FF6B6B, #4ECDC4) !important;56 border: none !important;57 }58 .gr-button-primary:hover {59 transform: translateY(-2px) !important;60 box-shadow: 0 4px 12px rgba(0,0,0,0.15) !important;61 }62 .progress-bar {63 background: linear-gradient(90deg, #FF6B6B, #4ECDC4) !important;64 }65 r r r r r �_get_custom_css� s z GradioInterface._get_custom_cssc C sF | j �� }d|d |d |d |d |d |d |d |d d66�S ) u 获取系统信息u 运行中� total_tasks�pending�67validating� uploading� transcribing� completed�failed�68queue_size)rP u 当前任务数u 待处理u 处理中u 已完成u 失败u 队列大小)r �get_statistics)r �statsr r r ra s 69�z GradioInterface._get_system_infoc C s ddl m } |�� �d�S )u 获取当前时间戳r )�datetimez%Y-%m-%d %H:%M:%S)r� �now�strftime)r r� r r r �_get_timestamp s z GradioInterface._get_timestamp�files�priority� languagesrh ri rj rk N�70channel_idrl rm rn c � st ��z�|s di dfW S g } | � d| �� � d�� | � d| �� � dt|�� d�� dd� |D �}
| � d| �� � d �� t|
�D ]M\}}z |�� j}| � d| �� � d71|d � d |j� d
|� d� � W q> ty� } z | � d| �� � d72|d � d |j� dt|�� d� � W Y d}~q>d}~ww z dd� |� d�D �}W n t73y� dg}Y nw d}|dur�t |t t
f�r�|dkr�|dkr�t |�}n| � d| �� � d|� d�� d}| �� r�zt�| �}W n tjy� } z| � d| �� � d�� W Y d}~nd}~ww |dk�rtjntj}|t|
�||||||| �� �r| �� nd|74�� �r!|75�� nd|�r*t�|�ndd� d �}| � d| �� � d |j� �� | � d| �� � d |�rNd �|�nd � �� | jj|
||d!�I dH | _| j�| j�}| � d| �� � d"| j� �� d#| j� �|�r�|� � ni d$�| �d%t|
�� d&� fW S t�y� } z d't|�� �}| j � |� |i d(|� d$�fW Y d}~S d}~ww ))u� 处理上传的文件76 77 Args:78 files: 上传的文件列表79 languages: 选择的语言80 priority: 任务优先级81 channel_id: 音轨索引82 disfluency_removal: 是否过滤语气词83 timestamp_alignment: 是否启用时间戳校准84 diarization_enabled: 是否启用说话人分离85 speaker_count: 说话人数量参考值86 vocabulary_id: 热词ID v287 phrase_id: 热词ID v188 special_word_filter: 敏感词过滤配置89 90 Returns:91 (状态信息, 任务信息, 日志信息)92 u 请先上传音频文件u 错误: 未选择任何文件�[u ] 开始处理文件上传请求u ] 接收到 �93 个文件c S s g | ]}t |j��qS r )r �name)�.0�fr r r �94<listcomp>H s z2GradioInterface._process_files.<locals>.<listcomp>u ] 转换文件路径完成u ] 文件 r8 z: u95 (大小: u 字节)� (无法获取文件信息: �)Nc S s g | ] }|� � rt|� � ��qS r )�strip�int)r� �xr r r r� U s �,r r r7 u" ] 警告: 说话人数量无效（u ），将使用自动判断uC ] 警告: 敏感词过滤配置格式错误，将使用默认设置r, ) �language_hintsr� �disfluency_removal_enabled�timestamp_alignment_enabledrj rk rl rm rn )r� r) �paraformer_paramsu ] 创建任务，优先级: u ] 选择语言: z, u 自动识别)�96file_pathsr� �metadatau! ] 任务创建成功，任务ID: u 任务已创建: �97u 98开始处理 u 个文件...99u 创建任务失败: u 错误: ) �appendr� �len� enumerate�stat�st_sizer� � ExceptionrK �split�100ValueError�101isinstancer� �floatr� �json�loads�JSONDecodeErrorr r, r+ �dumpsr. �joinr � create_taskr �get_task�to_dictr � exception)r r� r� r� rh ri rj rk r� rl rm rn Z log_messagesr� �i� file_path� file_size�eZ channel_listZvalidated_speaker_countZspecial_filterZ
task_priorityr� �task� error_msgr r r rd s� � 1022>�� 103� 104 "���� *� 105� ��z GradioInterface._process_filesc C s@ | j sdi fS t�| j�| j ��}|r d| j � d�i fS di fS )u 取消当前任务u 没有正在执行的任务u 任务 u106 已取消u 取消任务失败)r �asyncior� r � cancel_task)r �successr r r re � s �z$GradioInterface._cancel_current_taskc C s d| _ ddi g ddi fS )u 清空界面Nr> u' 界面已清空，等待上传文件...)r r r r r rf � s �z GradioInterface._clear_interfacec C s� d}i }d}i }g }d}| j r�| j�| j �}|r�|�� }d|jj� d|jj� �}| �|�}|jt j107kr�| j � d|j
j� �� |j
jr�|j
j}|�dg �}| j � d|� �� d�d d108� |D ��}g }|D ]2} d | vr�|�| �d d��d
�d | �dd�d�d�t| �dd��| �dd�t| �dd�d�g� qcn"| j � d� n|jt jkr�|j
r�|j
jr�|d| �� � d|j
j� �7 }| �� }109||||||110|fS )u 更新界面状态rD r> r� z] u& 任务已完成，检查转录结果: �transcriptionsu 转录结果: z111 112c S s@ g | ] }|� d �rd|� dd��d�d � d|� d d�� ��qS )�textu 文件: �file_urlr> �/�����r� )�getr� )r� �tr r r r� � s113 114�(�z5GradioInterface._update_interface.<locals>.<listcomp>�errorr� r� r� �durationr z.1f�sr� �language�unknown�115confidencer# u$ 任务已完成但没有转录结果z116[u ] 任务失败: )r r r� r� �statusr. �progress�message�_collect_task_logsr � COMPLETEDr �debug�result�transcription_resultsr� r� r� r� r� �round�FAILED�
error_messager� ra ) r ro rs rp rt Z117stats_dataru r� r� r� rq r r r rc � sN 118 119�120���121 z!GradioInterface._update_interfacec
C s� |sdS g }|� d| �� � d|j� �� |� d| �� � d|jj� �� |� d| �� � d|j� �� |jrC|� d| �� � d|jj� �� t|d�r�|j r�|� d| �� � d�� t122|j �D ]C\}}z|� � j }|� d |d123 � d |j
� d |� d
�� W q[ ty� } z|� d |d124 � d |j
� dt|�� d�� W Y d}~q[d}~ww |jtjkr�|jr�|� d| �� � d|j� �� t|jd�r�|jjr�|jj�dg �}|� d| �� � dt|�� d�� d�|�S )u� 收集任务的详细日志125 126 Args:127 task: 任务对象128 129 Returns:130 格式化的日志字符串131 u 无任务信息r� u ] 任务ID: u ] 任务状态: u ] 任务创建时间: u ] 进度信息: r� u ] 文件列表:z r8 z. z (z bytes)r� r� Nu ] 任务完成时间: r� r� u ] 转录结果: r� r� )r� r� �idr� r. �132created_atr� r� �hasattrr� r� r� r� r� r� rK r r� r� � completed_atr� r� r� r� )r r� � log_linesr� r� r� r� r� r r r r� � s0 133(4�� 134z"GradioInterface._collect_task_logsc C s" | j �d|j� d|jj� �� dS )u 任务状态变化回调u 任务状态变化: z -> N)r r� r� r� r. )r r� r r r r s "z&GradioInterface._on_task_status_changec K sb d| j jjd| j jjd| j jj d�}|�|� | j�d|d � d|d � �� | jjd i |��S )135u 启动Gradio界面z0.0.0.0T)� server_name� server_port�sharer� Z136show_error�quietu 启动Gradio界面: http://r� �:r� Nr ) r �app�portr� �updater r< r �launch)r �kwargsZdefault_kwargsr r r r� s 137�138 zGradioInterface.launch)�__name__�139__module__� __qualname__�__doc__r rW rX r rK rv r ra r� r �boolr r� r rd re rf rc r� r r� r r r r r sN T
������140� �141� � �
142�w
8, r r c C s t S )uJ 获取Gradio界面实例143 144 Returns:145 Gradio界面实例146 )�gradio_interfacer r r r �get_gradio_interface/ s r� c C s t jS )uD 创建演示界面147 148 Returns:149 Gradio界面对象150 )r� r r r r r �create_demo_interface8 s r� ) r� r� r� �pathlibr �typingr r r r r �gradiorW Zpandas�pdZ core.configr Zcore.task_managerr151 r r Z utils.loggerr
Zservices.file_validatorr r r� r� rX r� r r r r �<module> s&