o
�J�h� � @ s� d dl mZmZmZmZmZmZ erd dlmZ ddl m
Z
ddl m Z ddl
mZ er2d dlmZ dZG dd � d �ZG d
d � d �Zd ee d
efdd�Zd ee ded
dfdd�Zdeded
dfdd�ZdS )� )�
TYPE_CHECKING�Any�Dict�Iterable�List�Sequence)�ConsoleRenderable� )� get_console)�Segment)�DEFAULT_TERMINAL_THEMEz�<pre style="white-space:pre;overflow-x:auto;line-height:normal;font-family:Menlo,'DejaVu Sans Mono',consolas,'Courier New',monospace">{code}</pre>
c
@ sP e Zd ZdZdededdfdd�Zdee d ee d
edeeef fd d �Z dS )
�JupyterRenderablez)A shim to write html to Jupyter notebook.�html�text�returnNc C s || _ || _d S )N)r r )�selfr r � r �JC:\pinokio\api\whisper-webui.git\app\env\lib\site-packages\rich\jupyter.py�__init__ s
zJupyterRenderable.__init__�include�exclude�kwargsc sF | j | jd�}�r�fdd�|�� D �}� r!� fdd�|�� D �}|S )N�z
text/plainz text/htmlc � i | ] \}}|� v r||�qS r r ��.0�k�v�r r r �
<dictcomp> � z7JupyterRenderable._repr_mimebundle_.<locals>.<dictcomp>c � i | ] \}}|� vr||�qS r r r �r r r r r )r r �items)r r r r �datar �r r r �_repr_mimebundle_ s z#JupyterRenderable._repr_mimebundle_)
�__name__�
__module__� __qualname__�__doc__�strr r r r r& r r r r r
s ���
�r
c @ sB e Zd ZdZdZdddee dee dedeeef f
d d
�Z d S ) � JupyterMixinz@Add to an Rich renderable to make it render in Jupyter notebook.r r r r r r r c sl t � }t|�| |j��}t|�}|�|�}||d�}�r'�fdd�|�� D �}� r4� fdd�|�� D �}|S )Nr c r r r r r r r r 5 r z2JupyterMixin._repr_mimebundle_.<locals>.<dictcomp>c r! r r r r" r r r 7 r )r
�list�render�options�_render_segments�_render_bufferr# ) r r r r �console�segmentsr r r$ r r% r r& ) s
z JupyterMixin._repr_mimebundle_N)
r' r( r) r* � __slots__r r+ r r r& r r r r r, $ s ����
�r, r3 r c C s� dt dt fdd�}g }|j}t}t�| �D ]1\}}}|r q||�}|rB|�|�}|r3d|� d|� d�n|}|jrBd|j� d |� d
�}||� qd �|�} tj | d �}
|
S )
Nr r c S s | � dd�� dd�� dd�S )z Escape html.�&z&�<z<�>z>)�replace)r r r r �escape< s z _render_segments.<locals>.escapez
<span style="z">z</span>z <a href="z" target="_blank">z</a>� )�code)
r+ �appendr r �simplify�get_html_style�link�join�JUPYTER_HTML_FORMAT�format) r3 r9 � fragmentsZappend_fragment�themer �style�control�ruler; r r r r r0 ; s"
r0 r Nc C sB t | �}t||�}z
ddlm} ||� W dS ty Y dS w )zRender segments to Jupyter.r )�displayN)r0 r
ZIPython.displayrH �ModuleNotFoundError)r3 r r Zjupyter_renderableZipython_displayr r r rH T s
�rH �argsr c O s t � }|j| i |��S )zProxy for Console print.)r
�print)rJ r r2 r r r rK b s rK )�typingr r r r r r � rich.consoler r: r
�segmentr �terminal_themer rA r
r, r+ r0 rH rK r r r r �<module> s