&eb0ddlmZddlmZddlmZddlmZmZm Z m Z ddl m Z ddl mZddlmZmZmZddlmZmZdd lmZdd lmZdd lmZdd lmZ dd l!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(m)Z)m*Z*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1m2Z2erddl3m4Z4Gdde5eZ6d$dZ7dZ8eGdd Z9Gd!d"Z:d#S)%) annotations) dataclass)Enum) TYPE_CHECKINGOptionalTuplecast)Literal)runtime) AtomicImageWidthBehaviour image_to_url)check_callback_rulescheck_session_state_rules)StreamlitAPIException)Block) ChatInput)StringTriggerValue) RootContainer)gather_metrics)get_script_run_ctx) WidgetArgsWidgetCallback WidgetKwargsregister_widget)compute_widget_id)is_emoji)Keyto_key)DeltaGeneratorceZdZdZdZdZdZdS) PresetNamesuser assistantaihumanN)__name__ __module__ __qualname__USER ASSISTANTAIHUMAN?lib/python3.11/site-packages/streamlit/elements/widgets/chat.pyr"r".s" DI B EEEr/r"avatarstr | AtomicImage | None delta_pathstrreturn7Tuple[BlockProto.ChatMessage.AvatarType.ValueType, str]c tjj}| |jdfSt |t r8|dt Dvr%|j|t jt jfvrdndfSt |t rt|r |j |fS |j t|tjddd| fS#t$r}t!d |d}~wwxYw) aDetects the avatar type and prepares the avatar data for the frontend. Parameters ---------- avatar : The avatar that was provided by the user. delta_path : str The delta path is used as media ID when a local image is served via the media file manager. Returns ------- Tuple[AvatarType, str] The detected avatar type and the prepared avatar data. Nch|] }|j Sr.value.0items r0 z(_process_avatar_input..Ks/S/S/St /S/S/Sr/r$r#FRGBauto)widthclampchannels output_formatimage_idz5Failed to load the provided avatar value as an image.) BlockProto ChatMessage AvatarTypeICON isinstancer4r"r,r+rEMOJIIMAGErr ORIGINAL Exceptionr)r1r3rIexs r0_process_avatar_inputrQ5s!$'2J ~"" FC V/S/S{/S/S/S%S%S O+.+*?@@@ K   FC Xf%5%5'' #\$-$# &&&    'G  s%B>> CCCzm`st.chat_input()` can't be used inside an `st.expander`, `st.form`, `st.tabs`, `st.columns`, or `st.sidebar`.c"eZdZ d ddZdd Zd S)ChatInputSerder8ui_value!Optional[StringTriggerValueProto] widget_idr4r5 str | NonecB||dsdS|jS)Ndata)HasFieldrY)selfrTrVs r0 deserializezChatInputSerde.deserializejs*  8#4#4V#<#< 4}r/vStringTriggerValueProtoc"t|S)N)rY)r^)r[r]s r0 serializezChatInputSerde.serializers&A....r/N)r8)rTrUrVr4r5rW)r]rWr5r^)r'r(r)r\r`r.r/r0rSrShsHMO//////r/rSc eZdZedddd d Zed  d!ddd ddddd"dZed#dZdS)$ ChatMixin chat_messageN)r1name1Literal['user', 'assistant', 'ai', 'human'] | strr17Literal['user', 'assistant'] | str | AtomicImage | Noner5'DeltaGenerator'c|td|H|dtDvst|r|}t ||j\}}tj}||_ ||_ ||_ t}d|_ |j ||j|S)uInsert a chat message container. To add elements to the returned container, you can use ``with`` notation (preferred) or just call methods directly on the returned object. See the examples below. Parameters ---------- name : "user", "assistant", "ai", "human", or str The name of the message author. Can be "human"/"user" or "ai"/"assistant" to enable preset styling and avatars. Currently, the name is not shown in the UI but is only set as an accessibility label. For accessibility reasons, you should not use an empty string. avatar : str, numpy.ndarray, or BytesIO The avatar shown next to the message. Can be one of: * A single emoji, e.g. "🧑‍💻", "🤖", "🦖". Shortcodes are not supported. * An image using one of the formats allowed for ``st.image``: path of a local image file; URL to fetch the image from; an SVG image; array of shape (w,h) or (w,h,1) for a monochrome image, (w,h,3) for a color image, or (w,h,4) for an RGBA image. If None (default), uses default icons if ``name`` is "user", "assistant", "ai", "human" or the first letter of the ``name`` value. Returns ------- Container A single container that can hold multiple elements. Examples -------- You can use ``with`` notation to insert any element into an expander >>> import streamlit as st >>> import numpy as np >>> >>> with st.chat_message("user"): ... st.write("Hello 👋") ... st.line_chart(np.random.randn(30, 3)) .. output :: https://doc-chat-message-user.streamlit.app/ height: 450px Or you can just call methods directly in the returned objects: >>> import streamlit as st >>> import numpy as np >>> >>> message = st.chat_message("assistant") >>> message.write("Hello human") >>> message.bar_chart(np.random.randn(30, 3)) .. output :: https://doc-chat-message-user1.streamlit.app/ height: 450px NzWThe author name is required for a chat message, please set it via the parameter `name`.ch|] }|j Sr.r:r<s r0r?z)ChatMixin.chat_message..s@@@DTZ@@@r/T) block_proto)rlowerr"rrQdg_get_delta_path_strrGrHrdr1 avatar_type allow_emptyrcCopyFrom_block)r[rdr1rnconverted_avatarmessage_container_protorjs r0rczChatMixin.chat_messagewsL <'i  > JJLL@@K@@@ @ @HTNN @ZZ\\F(= DG//11) ) % %#-"8":":'+$)9&.9+ ll "&  ))*ABBBw~~+~666r/ chat_input Your messageF)key max_charsdisabled on_submitargskwargs placeholderr4rv Key | Nonerw int | NonerxboolryWidgetCallback | NonerzWidgetArgs | Noner{WidgetKwargs | NonerWc Xd}t|}t|j|t||dt } t d||||| r| jnd} tjrbtt|jj j dks|jj j tjkrt!t"t%} | | _t)|| _||| _|| _t$jj| _t7} t9d| ||||| j| j|  } || _| j r| j!| j!| _!d | _"|j#d| | j s| j!ndS) a?Display a chat input widget. .. warning:: Chat input can only be used once per app page and inside the main area of the app. It cannot be used in the sidebar, columns, expanders, forms or tabs. We plan to support this in the future. Parameters ---------- placeholder : str A placeholder text shown when the chat input is empty. Defaults to "Your message". For accessibility reasons, you should not use an empty string. key : str or int An optional string or integer to use as the unique key for the widget. If this is omitted, a key will be generated for the widget based on its content. Multiple widgets of the same type may not share the same key. max_chars : int or None The maximum number of characters that can be entered. If None (default), there will be no maximum. disabled : bool Whether the chat input should be disabled. Defaults to False. on_submit : callable An optional callback invoked when the chat input's value is submitted. args : tuple An optional tuple of args to pass to the callback. kwargs : dict An optional dict of kwargs to pass to the callback. Returns ------- str or None The current (non-empty) value of the text input widget on the last run of the app, None otherwise. Examples -------- >>> import streamlit as st >>> >>> prompt = st.chat_input("Say something") >>> if prompt: ... st.write(f"User has sent the following prompt: {prompt}") .. output :: https://doc-chat-input.streamlit.app/ height: 350px r8F) default_valuervwrites_allowedrtN)user_keyrvr|rwpager)ron_change_handlerrzr{ deserializer serializerctxT)$rrrlrrrpage_script_hashr existslenlist _active_dg_parent_block_types_root_containerrSIDEBARr DISALLOWED_CONTAINERS_ERROR_TEXTChatInputProtoidr4r|rwdefaultPositionBOTTOMpositionrSrr\r`rx value_changedr; set_value_enqueue)r[r|rvrwrxryrzr{rrrchat_input_protoserde widget_states r0rtzChatMixin.chat_inputsFSkkTWi000!SQVWWWW ""  #),6%%$     >   ND+?@@AAAEE7%59NNN,,LMMM)++ '*;'7'7$  )2  &#* $2$;$B!  &  '*    %-!  % .,*<*H%1%7  ")-  & '7888)5)CM|!!Mr/c"td|S)zGet our DeltaGenerator.r )r )r[s r0rlz ChatMixin.dgUs$d+++r/)rdrer1rfr5rg)ru)r|r4rvr}rwr~rxrryrrzrr{rr5rW)r5rg)r'r(r)rrcrtpropertyrlr.r/r0rbrbvs^N## KO [7[7[7[7[7$#[7z^L!!*}N $+/"&&*}N}N}N}N}N"!}N~,,,X,,,r/rbN)r1r2r3r4r5r6); __future__r dataclassesrenumrtypingrrrr typing_extensionsr streamlitr streamlit.elements.imager r rstreamlit.elements.utilsrrstreamlit.errorsrstreamlit.proto.Block_pb2rrGstreamlit.proto.ChatInput_pb2rrstreamlit.proto.Common_pb2rr^!streamlit.proto.RootContainer_pb2rstreamlit.runtime.metrics_utilrstreamlit.runtime.scriptrunnerrstreamlit.runtime.staterrrrstreamlit.runtime.state.commonrstreamlit.string_utilrstreamlit.type_utilrrstreamlit.delta_generatorr r4r"rQrrSrbr.r/r0rs#"""""!!!!!!777777777777%%%%%%NNNNNNNNNNTTTTTTTT222222999999EEEEEETTTTTT;;;;;;999999====== =<<<<<******++++++++9888888#t----`$S   / / / / / / /  /b,b,b,b,b,b,b,b,b,b,r/