a pDfh@sdZddlmZddlmZddlmZddlmZddl m Z m Z m Z m Z mZddlZdd lmZdd lmZmZdd lmZdd lmZdd lmZddlmZddlmZmZddl m!Z!m"Z"ddl#m$Z$ddl%m&Z&m'Z'eGdddZ(Gddde"Z)dS)z The interface module provides an even higher-level API for interacting with a list of `ChatMessage` objects compared to the `ChatFeed` through a frontend input UI. ) annotations) dataclass)partial)BytesIO)AnyCallableClassVarDictListN)CDN_DIST)RowTabs) ImageBase)Viewable)Widget)Button) FileInput TextInput) CallbackStateChatFeed) ChatAreaInput) ChatMessage_FileInputMessagec@sNeZdZUdZded<ded<ded<ded<ded <d ed <d Zd ed<d S)_ChatButtonDataa- A dataclass to hold the metadata and data related to the chat buttons. Parameters ---------- index : int The index of the button. name : str The name of the button. icon : str The icon to display. objects : List The objects to display. buttons : List The buttons to display. callback : Callable The callback to execute when the button is clicked. js_on_click : dict | str | None The JavaScript `code` and `args` to execute when the button is clicked. intindexstrnameiconr objectsbuttonsrcallbackNzdict | str | None js_on_click)__name__ __module__ __qualname____doc____annotations__r$r*r*a/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/panel/chat/interface.pyrs rcseZdZUdZejddZejee e e fddZ ej dddZej dd dZej dd dZej dd dZej dd dZej dd dZej dddZejdddZejeefdddZejiddZejidddZejeddZejeefddZ ejiddZ!ejiddZ"e#dgZ$de%d<fdd Z&d!d"d#d$Z'ej(d%dd&d'd(Z)ej(d)d*dd&d+d,Z*dkd.d.d/d0d1d2Z+dld3d4d5d6d7d8Z,dmd3d4d9d6d:d;Z-dd?Z.d@d9dAdBdCZ/dDdEZ0dnd3d4d5d6dFdGZ1dod3d4d5d6dHdIZ2dpd3d4d5d6dJdKZ3e4dLd=dMdNZ5e4d>> async def repeat_contents(contents, user, instance): >>> yield contents >>> chat_interface = ChatInterface( callback=repeat_contents, widgets=[TextInput(), FileInput()] ) z The widget types to automatically send when the user presses enter or clicks away from the widget. If not provided, defaults to `[TextInput]`.)docz The avatar to use for the user. Can be a single character text, an emoji, or anything supported by `pn.pane.Image`. If not set, uses the first character of the name.)class_r-Tzl Whether to reset the widget's value after sending a message; has no effect for `TextInput`.)defaultr-z) Whether to show the send button.z Whether to show the stop button temporarily replacing the send button during callback; has no effect if `callback` is not async.z* Whether to show the rerun button.z) Whether to show the undo button.z* Whether to show the clear button.Nz) Whether to show the button name.ZUserz( Name of the ChatInterface user.FzZ Widgets to use for the input. If not provided, defaults to `[TextInput]`.)r. allow_refsr-a Allows addition of functionality or customization of buttons by supplying a mapping from the button name to a dictionary containing the `icon`, `callback`, `post_callback`, and/or `js_on_click` keys. If the button names correspond to default buttons (send, rerun, undo, clear), the default icon can be updated and if a `callback` key value pair is provided, the specified callback functionality runs before the existing one. For button names that don't match existing ones, new buttons are created and must include a `callback`, `post_callback`, and/or `js_on_click` key. The provided callbacks should have a signature that accepts two positional arguments: instance (the ChatInterface instance) and event (the button click event). The `js_on_click` key should be a str or dict. If str, provide the JavaScript code; else if dict, it must have a `code` key, containing the JavaScript code to execute when the button is clicked, and optionally an `args` key, containing dictionary of arguments to pass to the JavaScript code. z The input widgets.)r/r0r-z The input message row that wraps the input layout (Tabs / Row) to easily swap between Tabs and Rows, depending on number of widgets.z: The input layout that contains the input widgets.z2 Metadata and data related to the buttons.z The rendered buttons.zcss/chat_interface.csszClassVar[List[str]] _stylesheetscs|d}|dur$tddg|d<nt|ts8|g|d<|dd}tj|i|tdg|jd|_ | | |dur||_ |j jj|j j|j gdgddS) NwidgetszSend a message) placeholderactivezchat-interface-input-container) css_classes stylesheetszchat-interface)r!r5)getr isinstancelistpopsuper__init__r r1_input_container_update_input_width _init_widgetsr4Z_cardparamupdater!)selfr!paramsr2r4 __class__r*r+r<s&    zChatInterface.__init__r)objcCs6dD],}t||t|||j|fi||iqdS)zg Link the disabled and loading attributes of the chat box to the given object. )disabledZloadingN)setattrgetattrlink)rBrFattrr*r*r+_link_disabled_loadingsz$ChatInterface._link_disabled_loadingwidth)watchcCs$|jdur |jdup|jdk|_dS)z) Update the input width. Ni)show_button_namerMrBr*r*r+r>s z!ChatInterface._update_input_widthr2button_propertiesc Csd|jdd|jdd|jdd|jdd|jdd}t|jdk|_i||j}t| D]\}\}}| }| d }| d }| d }| |pi} | r| d } |d us|d ur|j |||d| n| }nf|d ur|d ur|j |d|}nD|d ur|d ur|}n*|d ur@|d ur@|s@t d|d| dpT| d} t||| gg||d|j|<qb|j} t|jtr|jg} i|_g} | D]N}|jp|jj}t|tr|}|j ||ur||j|<| |q|j}|d ur&d|vsd|vrd}nd|vr&d }t|dg|jdd}|j D] \}}t|t|jpnt|ttfv}|r|| vrt |jdj!|}|j"#|d|j"j$ddgdt|tr|j%|ddi|_&|j'D]8}|j}z"|d kr|j"d!|nd"}Wnt(y"d}Yn0|j"j)*}t+|j*,|j-d#|j.d|j*,d$d%d&d'd(|d)}|d krx|/||j!rt |j!|}|0||j1r|j1}t|t2rd*|vrt d+|d,|j1| d-i|d*d.nt|t3r|j1|d/||j&|<|j4|qt5|gt6|j&'Rdd0g|jd1d2}|||fqDt|jd3krl|d}|g|j7_8||_9d S)4zk Initialize the input widgets. Returns ------- The input widgets. send)r _default_callbackz player-stoprepeatz arrow-backZtrash)rRstopZrerunundoclearrr# post_callbackr$rSNr#rXr)rXz%A 'callback' key is required for the z buttonr )rrr r!r"r#r$ZbothZ scale_heightZ stretch_widthheightzchat-interface-input-tabsT) sizing_moder5r6Zdynamicvaluezchat-interface-input-widget)r[r5Zdisabled_enter)rGrUZshow_FZ-2)rrrcenter)rr r[ max_widthZ max_heightmarginalignvisiblecodez!A 'code' key is required for the z button's 'js_on_click' keyargs)rhrg)rgzchat-interface-input-rowstart)r[r5r6rer): _click_send _click_stop _click_rerun _click_undo _click_clearlenrQ _allow_revert enumerateitemslowerr7_wrap_callbacks ValueErrorr _button_datar2r8r_widgetsrrEr%typeappendr[rr1tupleauto_send_typesrrrr#r@rNrArJ_buttonsvaluesKeyErrorrOrxrwheretitler rLZon_clickr$dictrr"r r9r=r! _input_layout)rBZdefault_button_propertiesrQrrZ propertiesr#rXr$Zdefault_propertiesdefault_callbackr r2Z new_widgetsZwidgetkeyr[Z input_layoutZ auto_send button_dataactionrfZ show_exprbuttonZ message_rowr*r*r+r?s               "                zChatInterface._init_widgetsr]zCallable | NonerrYcsddfdd }|S)zR Wrap the callback and post callback around the default callback. rrcsddfdd }|S)Nzparam.parameterized.Event)eventcszdkr|jjsdSdur@zd|_||Wd|_nd|_0||durvzd|_||Wd|_nd|_0dS)NrRTF) active_widgetr\rG)rBr)r#rrrXr*r+wrapperps   z@ChatInterface._wrap_callbacks..decorate..wrapperr*)rrr#rrXrr+decorateosz/ChatInterface._wrap_callbacks..decorater*)rBr#rXrrr*rr+rtfs zChatInterface._wrap_callbacksz param.parameterized.Event | Nonez'ChatInterface' | NoneNone)rinstancereturncCs|jr dS|j}|j}|s*t|dr*|j}|rt|trJt||j|j d}t |t us\|j rddi}t|drvd|d<z:t ||j |Wdn1s0YWqtyYq0ndS||j||j|jdddS)z= Send the input when the user presses Enter. N value_input)contents mime_type file_namer\r]T)r\useravatarrespond)rGrr\hasattrrr8rrrfilenamerxr reset_on_sendr@Zdiscard_eventsrAru_reset_button_datarRrr)rBrrrr\Zupdatesr*r*r+rjs2    . zChatInterface._click_sendboolcCs|S)zL Cancel the callback when the user presses the Stop button. )rU)rBrrr*r*r+rkszChatInterface._click_stopr)rcCs<|jddd}t|dD]\}}|j|jkr|SqdS)z9 Get the index of the last user message. Nrr)r!rqr)rBmessagesrmessager*r*r+_get_last_user_entry_indexs   z(ChatInterface._get_last_user_entry_indexr)rr4cCsZ|jD]N}|r"|jr"dddd}n&d|jr4|jnd|jr@dndd}|j|qdS) zk Toggle the button's icon and name to indicate whether the action can be reverted. warningZRevertr^)Z button_typerrMr/r]r_N)r"r!rOrrr@rA)rBrr4rZ button_updater*r*r+_toggle_reverts   zChatInterface._toggle_revertcCs*|jD]}|j||dq dS)zY Clears all the objects in the button data to prevent reverting. FN)rvr}r!rWr)rBrr*r*r+rs z ChatInterface._reset_button_datacCs0|}||}|sdS|j|ddddS)z| Upon clicking the rerun button, rerun the last user message, which can trigger the callback again. NrT)r\r)rrVrR)rBrrcountrr*r*r+rls  zChatInterface._click_reruncCsb|jd}|j}|sL||}|||_|jrD||dq^g|_n|||dS)z Upon clicking the undo button, undo (remove) messages up to the last user message. If the button is clicked again without performing any other actions, revert the undo. rVTN)rvr!rrrVrprextend)rBrrZ undo_dataZ undo_objectsrr*r*r+rms   zChatInterface._click_undocCs^|jd}|j}|sB|||_|jr:||dqZg|_n||dd<|dS)z Upon clicking the clear button, clear the chat log. If the button is clicked again without performing any other actions, revert the clear. rWTN)rvr!rrWrprcopy)rBrrZ clear_dataZ clear_objectsr*r*r+rn s  zChatInterface._click_clearrcCs*t|jtr|j|jjdS|jjdS)zj The currently active widget. Returns ------- The active widget. r)r8rrr4r!rPr*r*r+r s zChatInterface.active_widgetcCst|jtr|jjSdS)z The currently active input widget tab index; -1 if there is only one widget available which is not in a tab. Returns ------- The active input widget tab index. rr8rrr4rPr*r*r+r4-s zChatInterface.active)rrcCst|jtr||j_dS)z Set the active input widget tab index. Arguments --------- index : int The active index to set. Nr)rBrr*r*r+r4<s  assistantzList[ChatMessage]z!Dict[str, str | List[str]] | Nonez str | NonerzList[Dict[str, Any]])r role_names default_rolecustom_serializerrcs,|dur|jg|jgd}t||||S)a Exports the chat log for use with transformers. Arguments --------- messages : list(ChatMessage) A list of ChatMessage objects to serialize. role_names : dict(str, str | list(str)) | None A dictionary mapping the role to the ChatMessage's user name. Defaults to `{"user": [self.user], "assistant": [self.callback_user]}` if not set. The keys and values are case insensitive as the strings will all be lowercased. The values can be a string or a list of strings, e.g. `{"user": "user", "assistant": ["executor", "langchain"]}`. default_role : str The default role to use if the user name is not found in role_names. If this is set to None, raises a ValueError if the user name is not found. custom_serializer : callable A custom function to format the ChatMessage's object. The function must accept one positional argument, the ChatMessage object, and return a string. If not provided, uses the serialize method on ChatMessage. Returns ------- A list of dictionaries with a role and content keys. N)rr)rZ callback_userr;_serialize_for_transformers)rBrrrrrDr*r+rIs z)ChatInterface._serialize_for_transformers_callback_statecstjtjf}|jr&|j|vs&|jdurltj|(d|j d_ d|j d_ Wdq1s`0YnDtj|(d|j d_ d|j d_ Wdn1s0YdS)NTrRFrU) rZRUNNINGZ GENERATING show_stoprZ_callback_futurer@Z parameterizedZbatch_call_watchersr|rf)rBZ busy_statesr*r*r+_update_input_disabledps  , z$ChatInterface._update_input_disabledcs"tIdH|IdHdS)zF Events to always execute after the callback is done. N)r;_cleanup_responserrPrDr*r+r|szChatInterface._cleanup_responsezChatMessage | dict | Anyzstr | bytes | BytesIO | NonezChatMessage | None)r\rrrrcs:t|ts&|dur|j}|dur&|j}tj||||dS)a Sends a value and creates a new message in the chat log. If `respond` is `True`, additionally executes the callback, if provided. Arguments --------- value : ChatMessage | dict | Any The message contents to send. user : str | None The user to send as; overrides the message message's user if provided. Will default to the user parameter. avatar : str | bytes | BytesIO | None The avatar to use; overrides the message message's avatar if provided. Will default to the avatar parameter. respond : bool Whether to execute the callback. Returns ------- The message that was created. N)rrr)r8rrrr;rR)rBr\rrrrDr*r+rRs  zChatInterface.sendzstr | dict | ChatMessage)r\rrrreplacercs4t|ts|p|j}|p|j}tj|||||dS)aR Streams a token and updates the provided message, if provided. Otherwise creates a new message in the chat log, so be sure the returned message is passed back into the method, e.g. `message = chat.stream(token, message=message)`. This method is primarily for outputs that are not generators-- notably LangChain. For most cases, use the send method instead. Arguments --------- value : str | dict | ChatMessage The new token value to stream. user : str | None The user to stream as; overrides the message's user if provided. Will default to the user parameter. avatar : str | bytes | BytesIO | None The avatar to use; overrides the message's avatar if provided. Will default to the avatar parameter. message : ChatMessage | None The message to update. replace : bool Whether to replace the existing text when streaming a string or dict. Returns ------- The message that was updated. )rrrr)r8rrrr;stream)rBr\rrrrrDr*r+rs$   zChatInterface.stream)NNr])NN)NN)NN)NN)NN)NrN)NNT)NNNF)>r%r&r'r(r@r r{Z ClassSelectorrrbytesrrBooleanrZ show_sendrZ show_rerunZ show_undoZ show_clearrOStringrrr9r2r rQrwr r=rrrvr|r r1r)r<rLdependsr>r?rtrjrkrrrrlrmrnpropertyrr4setterrrrrRr __classcell__r*r*rDr+r,?s      #,    '  'r,)*r( __future__rZ dataclassesr functoolsriortypingrrrr r r@Z io.resourcesr Zlayoutr rZ pane.imagerZviewablerZ widgets.baserZwidgets.buttonrZ widgets.inputrrfeedrrinputrrrrrr,r*r*r*r+s&