�
] �g� � �� � d dl mZmZmZmZmZmZ erd dlmZ ddl m
Z
ddl m Z ddl
mZ erd dlmZ dZ G d� d � � Z G d
� d � � Zd ee d
efd�Zd ee ded
dfd�Zdeded
dfd�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
�f � e Zd ZdZdededdfd�Zdee dee d edeeef fd
�Z dS ) �JupyterRenderablez)A shim to write html to Jupyter notebook.�html�text�returnNc �" � || _ || _ d S )N)r r )�selfr r s �\/home/asafur/pinokio/api/open-webui.git/app/env/lib/python3.11/site-packages/rich/jupyter.py�__init__zJupyterRenderable.__init__ s � ��� ��� � � � �include�exclude�kwargsc � ��� | j | j d�}�r �fd�|� � � D � � }�r �fd�|� � � D � � }|S )N�z
text/plainz text/htmlc �$ �� i | ] \ }}|�v � ||��
S � r ��.0�k�vr s �r �
<dictcomp>z7JupyterRenderable._repr_mimebundle_.<locals>.<dictcomp> �$ �� �D�D�D�V�a��q�G�|�|�A�q�|�|�|r c �$ �� i | ] \ }}|�v� ||��
S r r �r r! r" r s �r r# z7JupyterRenderable._repr_mimebundle_.<locals>.<dictcomp> �) �� �H�H�H�V�a��q��7G�7G�A�q�7G�7G�7Gr )r r �items)r r r r �datas `` r �_repr_mimebundle_z#JupyterRenderable._repr_mimebundle_ sp ��� � #�i�d�i�@�@�� � E�D�D�D�D�t�z�z�|�|�D�D�D�D� � I�H�H�H�H�t�z�z�|�|�H�H�H�D�� r )
�__name__�
__module__� __qualname__�__doc__�strr r r r r* r r r r r s� � � � � � �3�3��S� �� �� � � � �� ��}��/7��}��HK��
�c�3�h� �� � � � � r r c �Z � e Zd ZdZdZdddee dee dedeeef f
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 �P ��� t � � }t |� | |j � � � � }t |� � }|� |� � }||d�}�r �fd�|� � � D � � }�r �fd�|� � � D � � }|S )Nr c �$ �� i | ] \ }}|�v � ||��
S r r r s �r r# z2JupyterMixin._repr_mimebundle_.<locals>.<dictcomp>5 r$ r c �$ �� i | ] \ }}|�v� ||��
S r r r& s �r r# z2JupyterMixin._repr_mimebundle_.<locals>.<dictcomp>7 r' r )r �list�render�options�_render_segments�_render_bufferr( ) r r r r �console�segmentsr r r) s `` r r* z JupyterMixin._repr_mimebundle_) s� ��� � �-�-������t�W�_�=�=�>�>�� ��)�)���%�%�h�/�/�� "��6�6�� � E�D�D�D�D�t�z�z�|�|�D�D�D�D� � I�H�H�H�H�t�z�z�|�|�H�H�H�D�� r N)
r+ r, r- r. � __slots__r r/ r r r* r r r r1 r1 $ sq � � � � � �J�J��I��!���#� �� �#� �� � �
�c�3�h� � � � � � � r r1 r; r c � � dt dt fd�}g }|j }t }t j | � � D ]X\ }}}|r� ||� � }|r7|� |� � }|r d|� d|� d�n|}|j rd|j � d|� d �} ||� � �Yd
� |� � } t � | � � � }
|
S ) Nr r c �~ � | � dd� � � dd� � � dd� � S )z Escape html.�&z&�<z<�>z>)�replace)r s r �escapez _render_segments.<locals>.escape< s6 � ��|�|�C��)�)�1�1�#�v�>�>�F�F�s�F�S�S�Sr z
<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) r; rC � fragments�append_fragment�themer �style�control�rulerE r s r r8 r8 ; s � �T�S� T�S� T� T� T� T� �I� �&�O� "�E� '� 0�� :� :� � � ��e�W� � � ��v�d�|�|�� � L��'�'��.�.�D�<@�J�8�4�8�8�4�8�8�8�8�d�D��z�
L�K�5�:�K�K��K�K�K����� � � � �
�7�7�9� � �D� � %� %�4� %� 0� 0�D� �Kr r Nc � � t | � � }t ||� � } ddlm} ||� � dS # t $ r Y dS w xY w)zRender segments to Jupyter.r )�displayN)r8 r �IPython.displayrT �ModuleNotFoundError)r; r r �jupyter_renderable�ipython_displays r rT rT T sq � � �H� %� %�D�*�4��6�6��
�>�>�>�>�>�>���*�+�+�+�+�+�� �
�
�
�
���
���s �4 �
A�A�argsr c �8 � t � � } |j | i |��S )zProxy for Console print.)r �print)rY r r: s r r[ r[ b s" � ��m�m�G� �7�=�$� )�&� )� )�)r )�typingr r r r r r � rich.consoler rD r �segmentr �terminal_themer
rK r r1 r/ r8 rT r[ r r r �<module>r` s �� E� E� E� E� E� E� E� E� E� E� E� E� E� E� E� E�� /�.�.�.�.�.�.� � � � � � � � � � � � � 2� 2� 2� 2� 2� 2�� /�.�.�.�.�.�.�� �
� � � � � � � �$� � � � � � � �.�x�� 0� �S� � � � �2
�h�w�'�
�s�
�t�
�
�
�
� *�� *�� *�� *� *� *� *� *� *r