o 沪g8@s`ddlmZddlmZddlmZddlmZmZm Z ddl m Z ddl m Z ddlmZddlmZmZmZdd lmZdd lmZmZmZmZdd lmZdd lmZdd l m!Z"ddl#m$Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-m.Z.m/Z/m0Z0ddl1m2Z2m3Z3erddl4m5Z5Gddde6eZ7d$ddZ8eGdd d Z9Gd!d"d"Z:d#S)%) annotations) dataclass)Enum) TYPE_CHECKINGLiteralcast)runtime)get_dg_singleton_instance) is_in_form) AtomicImage WidthBehavior image_to_url)check_widget_policies)Keycompute_and_register_element_idsave_for_app_testingto_key)StreamlitAPIException)Block) ChatInput)StringTriggerValue) RootContainer)gather_metrics)get_script_run_ctx) WidgetArgsWidgetCallback WidgetKwargsregister_widget)is_emojivalidate_material_icon)DeltaGeneratorc@seZdZdZdZdZdZdS) PresetNamesuser assistantaihumanN)__name__ __module__ __qualname__USER ASSISTANTAIHUMANr-r-[/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/streamlit/elements/widgets/chat.pyr!7s r!avatarstr | AtomicImage | None delta_pathstrreturn7tuple[BlockProto.ChatMessage.AvatarType.ValueType, str]c Cstjj}|dur |jdfSt|tr+|ddtDvr+|j|tjtjfvr(dfSdfSt|tr9t |r9|j |fSt|trJ| drJ|jt |fSz|j t|tjdd d |d fWStyj}ztd |d}~ww) 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. NcSh|]}|jqSr-value.0itemr-r-r. Tz(_process_avatar_input..r#r"z :materialFRGBauto)widthclampchannels output_formatimage_idz5Failed to load the provided avatar value as an image.) BlockProto ChatMessage AvatarTypeICON isinstancer2r!r+r*rEMOJI startswithrIMAGEr r ORIGINAL Exceptionr)r/r1rGexr-r-r._process_avatar_input>s>   rPc@s$eZdZ dddd Zdd d ZdS)ChatInputSerder5ui_valueStringTriggerValueProto | None widget_idr2r3 str | NonecCs|dus |ds dS|jS)Ndata)HasFieldrV)selfrRrTr-r-r. deserializeuszChatInputSerde.deserializevStringTriggerValueProtocCs t|dS)N)rV)r[)rXrZr-r-r. serialize}s zChatInputSerde.serializeN)r5)rRrSrTr2r3rU)rZrUr3r[)r&r'r(rYr\r-r-r-r.rQss rQc @sXeZdZedddd#d d Zed  d$dddddddd%dd Zed&d!d"ZdS)' ChatMixin chat_messageN)r/name1Literal['user', 'assistant', 'ai', 'human'] | strr/7Literal['user', 'assistant'] | str | AtomicImage | Noner3r cCs|durtd|dur|ddtDvst|r|}t||j\}}t}||_ ||_ ||_ t}d|_ |j ||jj|dS)u Insert 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 : Anything supported by st.image (except list), str, or None The avatar shown next to the message. If ``avatar`` is ``None`` (default), the icon will be determined from ``name`` as follows: - If ``name`` is ``"user"`` or ``"human"``, the message will have a default user icon. - If ``name`` is ``"ai"`` or ``"assistant"``, the message will have a default bot icon. - For all other values of ``name``, the message will show the first letter of the name. In addition to the types supported by |st.image|_ (except list), the following strings are valid: - A single-character emoji. For example, you can set ``avatar="🧑‍💻"`` or ``avatar="🦖"``. Emoji short codes are not supported. - An icon from the Material Symbols library (rounded style) in the format ``":material/icon_name:"`` where "icon_name" is the name of the icon in snake case. For example, ``icon=":material/thumb_up:"`` will display the Thumb Up icon. Find additional icons in the `Material Symbols `_ font library. .. |st.image| replace:: ``st.image`` .. _st.image: https://docs.streamlit.io/develop/api-reference/media/st.image 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`.cSr6r-r7r9r-r-r.r<r=z)ChatMixin.chat_message..T) block_proto)rlowerr!rrPdg_get_delta_path_strrErFr_r/ avatar_type allow_emptyr^CopyFrom_block)rXr_r/rfconverted_avatarmessage_container_protorbr-r-r.r^s$Z  zChatMixin.chat_message chat_input Your messageF)key max_charsdisabled on_submitargskwargs placeholderr2rn Key | Nonero int | NonerpboolrqWidgetCallback | NonerrWidgetArgs | NonersWidgetKwargs | NonerUc Cs:d}t|}t|j|||ddt} td|d||d} tr)t|jr)tdt |jj j } |jj j t jkr=| s=d} nd } t} | | _t|| _|durQ|| _|| _t}t| j||||j|j| d d }|| _|jrx|jdurx|j| _d | _| rt| | |j| dkrtj d| n|j d| |js|jSdS) ac Display a chat input widget. 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. No two widgets may have 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. Otherwise, ``None``. Examples -------- When ``st.chat_input`` is used in the main body of an app, it will be pinned to the bottom of the page. >>> 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 The chat input can also be used inline by nesting it inside any layout container (container, columns, tabs, sidebar, etc) or fragment. Create chat interfaces embedded next to other content or have multiple chatbots! >>> import streamlit as st >>> >>> with st.sidebar: >>> messages = st.container(height=300) >>> if prompt := st.chat_input("Say something"): >>> messages.chat_message("user").write(prompt) >>> messages.chat_message("assistant").write(f"Echo: {prompt}") .. output :: https://doc-chat-input-inline.streamlit.app/ height: 350px r5F) default_valuewrites_allowedrlN)user_keyform_idrtroz1`st.chat_input()` can't be used in a `st.form()`.bottominlinestring_trigger_value)on_change_handlerrrrs deserializer serializerctx value_typeT)!rrrdrrrexistsr rset _active_dg_ancestor_block_types_root_containerrMAINChatInputProtoidr2rtrodefaultrQrrYr\rp value_changedr8 set_valuerr bottom_dg_enqueue)rXrtrnrorprqrrrsrr element_idancestor_block_typespositionchat_input_protoserde widget_stater-r-r.rlsrQ   zChatMixin.chat_inputcCs td|S)zGet our DeltaGenerator.r )r)rXr-r-r.rds z ChatMixin.dg)r_r`r/rar3r )rm)rtr2rnrurorvrprwrqrxrrryrsrzr3rU)r3r )r&r'r(rr^rlpropertyrdr-r-r-r.r]s"q$r]N)r/r0r1r2r3r4); __future__r dataclassesrenumrtypingrrr streamlitr$streamlit.delta_generator_singletonsr !streamlit.elements.lib.form_utilsr "streamlit.elements.lib.image_utilsr r r streamlit.elements.lib.policiesrstreamlit.elements.lib.utilsrrrrstreamlit.errorsrstreamlit.proto.Block_pb2rrEstreamlit.proto.ChatInput_pb2rrstreamlit.proto.Common_pb2rr[!streamlit.proto.RootContainer_pb2rstreamlit.runtime.metrics_utilr7streamlit.runtime.scriptrunner_utils.script_run_contextrstreamlit.runtime.staterrrrstreamlit.string_utilrrstreamlit.delta_generatorr r2r!rPrQr]r-r-r-r.s4                5