o
    ßÙ·iO	  ã                
   @   sö   d dl mZmZ d dlmZ d dlmZ G dd„ deƒZG dd„ deƒZG dd	„ d	eƒZ	G d
d„ deƒZ
G dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZee	e
eeeeeeef	 ZdS )é    )ÚLiteralÚUnion)ÚServerMessage)Ú	BaseModelc                   @   s&   e Zd ZU eed< dZedB ed< dS )ÚBaseMessageÚmsgNÚevent_id)Ú__name__Ú
__module__Ú__qualname__r   Ú__annotations__r   Ústr© r   r   úJ/home/ubuntu/.local/lib/python3.10/site-packages/gradio/server_messages.pyr      s   
 r   c                   @   sf   e Zd ZU dZeeB dB ed< dZeeB dB ed< dZe	dB ed< dZ
edB ed< dZe	dB ed< dS )ÚProgressUnitNÚindexÚlengthÚunitÚprogressÚdesc)r	   r
   r   r   ÚintÚfloatr   r   r   r   r   r   r   r   r   r   r      s   
 r   c                   @   s2   e Zd ZU ejZeej ed< g Ze	e
 ed< dS )ÚProgressMessager   Úprogress_dataN)r	   r
   r   r   r   r   r   r   r   Úlistr   r   r   r   r   r      ó   
 r   c                   @   sZ   e Zd ZU ejZeej ed< eed< ed ed< dZ	e
dB ed< dZeed	< eed
< dS )Ú
LogMessager   Úlog)ÚinfoÚwarningÚsuccessÚlevelé
   NÚdurationTÚvisibleÚtitle)r	   r
   r   r   r   r   r   r   r   r#   r   r$   Úboolr   r   r   r   r      s   
 r   c                   @   sJ   e Zd ZU ejZeej ed< dZe	dB ed< e	ed< dZ
edB ed< dS )ÚEstimationMessager   NÚrankÚ
queue_sizeÚrank_eta)r	   r
   r   r   Ú
estimationr   r   r   r(   r   r*   r   r   r   r   r   r'   "   s
   
 r'   c                   @   s2   e Zd ZU ejZeej ed< dZe	dB ed< dS )ÚProcessStartsMessager   NÚeta)
r	   r
   r   r   Úprocess_startsr   r   r   r-   r   r   r   r   r   r,   )   r   r,   c                   @   sB   e Zd ZU ejZeej ed< eed< e	ed< dZ
edB ed< dS )ÚProcessCompletedMessager   Úoutputr    Nr%   )r	   r
   r   r   Úprocess_completedr   r   r   Údictr&   r%   r   r   r   r   r   r/   .   s
   
 r/   c                   @   sH   e Zd ZU ejZeejejf ed< e	ed< e
ed< dZedB ed< dS )ÚProcessGeneratingMessager   r0   r    NÚ
time_limit)r	   r
   r   r   Úprocess_generatingr   r   Úprocess_streamingr   r2   r&   r4   r   r   r   r   r   r3   5   s   
 ÿr3   c                   @   ó"   e Zd ZU ejZeej ed< dS )ÚHeartbeatMessager   N)r	   r
   r   r   Ú	heartbeatr   r   r   r   r   r   r   r8   >   ó   
 r8   c                   @   r7   )ÚCloseStreamMessager   N)r	   r
   r   r   Úclose_streamr   r   r   r   r   r   r   r;   B   r:   r;   c                   @   sF   e Zd ZU ejZeej ed< eed< dZ	e
ed< dZed ed< dS )ÚUnexpectedErrorMessager   ÚmessageFÚsession_not_foundr    N)r	   r
   r   r   Úunexpected_errorr   r   r   r   r?   r&   r    r   r   r   r   r=   F   s
   
 r=   N)Útypingr   r   Úgradio_client.utilsr   Úpydanticr   r   r   r   r   r'   r,   r/   r3   r8   r;   r=   ÚEventMessager   r   r   r   Ú<module>   s4    		øÿ