o 沪gP@sUddlmZddlmZddlmZmZmZmZm Z m Z m Z m Z m Z ddlmZddlmZddlmZmZddlmZmZddlmZmZmZmZmZmZdd lm Z dd l!m"Z"dd l#m$Z%dd l&m'Z'dd l(m)Z)ddl*m+Z+ddl,m-Z-m.Z.ddl/m0Z0erddl1m2Z2ddl3m4Z4ddl*m5Z5m6Z6m7Z7ddl8m9Z9m:Z:m;Z;e dZd<dZ?de>d<dZ@de>d<dZAde>d<dZBde>d <e d!ZCd"e>d#<Gd$d%d%e e0ZDGd&d'd'e e0ZEdMd,d-ZFe%jGjHd.d/d0d/fdNdFdGZIdOdIdJZJGdKdLdLZKd/S)P) annotations)Sequence) TYPE_CHECKINGAnyCallableFinalGenericLiteralTypeVarcastoverload) TypeAlias)current_form_id)(convert_to_sequence_and_check_comparableget_default_indices)check_widget_policiesmaybe_raise_label_warnings)KeyLabelVisibilitycompute_and_register_element_id get_label_visibility_proto_valuesave_for_app_testingto_key)MultiSelectSerdeStreamlitAPIException) ButtonGroup)gather_metrics)get_script_run_ctx)register_widget)is_emojivalidate_material_icon)T)OptionSequence)DeltaGenerator) WidgetArgsWidgetCallback WidgetKwargs)RegisterWidgetResultWidgetDeserializerWidgetSerializerV)z:material/thumb_up:z:material/thumb_down:r _THUMB_ICONS)z:material/sentiment_sad:z!:material/sentiment_dissatisfied:z:material/sentiment_neutral:z:material/sentiment_satisfied:z#:material/sentiment_very_satisfied: _FACES_ICONS _NUMBER_STARSz:material/star: _STAR_ICONz:material/star_filled:_SELECTED_STAR_ICONsinglemultir SelectionModec@s4eZdZdZ ddd d ZdddZddddZdS)SingleSelectSerdeaUses the MultiSelectSerde under-the-hood, but accepts a single index value and deserializes to a single index value. This is because button_group can be single and multi select, but we use the same proto for both and, thus, map single values to a list of values and a receiving value wrapped in a list to a single value. When a default_value is provided is provided, the option corresponding to the index is serialized/deserialized. Noption_indices Sequence[T] default_valuelist[int] | NonereturnNonecCst||dur|ng|_dSN)rmultiselect_serde)selfr7r9r@c/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/streamlit/elements/widgets/button_group.py__init__gs zSingleSelectSerde.__init__valueT | None list[int]cCs|dur|gng}|j|Sr=)r> serialize)r?rC_valuer@r@rArFqs zSingleSelectSerde.serializeui_value widget_idstrcCs&|j||}t|dkrdS|dS)Nr)r> deserializelen)r?rIrJ deserializedr@r@rArLus zSingleSelectSerde.deserializer=)r7r8r9r:r;r<)rCrDr;rErH)rIr:rJrKr;rD__name__ __module__ __qualname____doc__rBrFrLr@r@r@rAr6\s  r6c@s2eZdZdZddd Zdd dZ ddddZdS)SingleOrMultiSelectSerdeaA serde that can handle both single and multi select options. It uses the same proto to wire the data, so that we can send and receive single values via a list. We have different serdes for both cases though so that when setting / getting the value via session_state, it is mapped correctly. So for single select, the value will be a single value and for multi select, it will be a list of values. optionsr8default_valuesrEtypeLiteral['single', 'multi']cCs<||_||_||_|dkrt||d|_dSt|||_dS)Nr3r9)rVrWrXr6rserde)r?rVrWrXr@r@rArBs z!SingleOrMultiSelectSerde.__init__rCT | list[T] | Noner;cCs|jtt|Sr=)r[rFr r)r?rCr@r@rArFsz"SingleOrMultiSelectSerde.serializerHrIr:rJrKlist[T] | T | NonecCs|j||Sr=)r[rL)r?rIrJr@r@rArLsz$SingleOrMultiSelectSerde.deserializeN)rVr8rWrErXrY)rCr\r;rErO)rIr:rJrKr;r]rPr@r@r@rArU~s  rUfeedback_option#Literal['thumbs', 'faces', 'stars']r;/tuple[list[ButtonGroupProto.Option], list[int]]cCsg}g}|dkrttttt}ddtD}||fS|dkr4tttt}ddtD}||fS|dkrHttt}tjt t dgt}||fS)NthumbscSg|]}tj|dqS) content_iconButtonGroupProtoOption.0iconr@r@rA z&get_mapped_options..facescSrbrcrerhr@r@rArkrlstars)rdselected_content_icon) listreversedrangerMr,r-r/rfrgr0r1)r^rVoptions_indicesr@r@rAget_mapped_optionss(   rtpillsNvisiblerJrKformatted_options!Sequence[ButtonGroupProto.Option]rWrEdisabledboolr click_mode$ButtonGroupProto.ClickMode.ValueTypeselection_visualization1ButtonGroupProto.SelectionVisualization.ValueTypestyle3Literal['borderless', 'pills', 'segmented_control']label str | Nonelabel_visibilityrhelprfc Cst} || _|| jdd<|| _|| _|| _tj|| _ |dur3|| _ t | | j _ | dur3| | _|D]} | j| q5|| _| Sr=)rfiddefaultform_idryr{StyleValueupperrrrrrCrrVappendr}) rJrwrWryrr{r}rrrrprotoformatted_optionr@r@rA _build_protos$rselection_modecCs|dvr td|ddS)zHCheck if the selection_mode value is valid or raise exception otherwise.r2zYThe selection_mode argument must be one of ['single', 'multi']. The argument passed was ''.Nr)rr@r@rA#_maybe_raise_selection_mode_warningsrc@seZdZe dWdddddddXddZe dWdddddddYddZed dZddddddd[ddZeddddddddddd d\d-d.Zedddddddddd/ d]d3d.Zed4ddddddddddd d^d7d.Zeddddddddddd d_d9d:Zedddddddddd/ d`d;d:Zed<ddddddddddd dad=d:Zed>dddddd>> import streamlit as st >>> >>> sentiment_mapping = ["one", "two", "three", "four", "five"] >>> selected = st.feedback("stars") >>> if selected is not None: >>> st.markdown(f"You selected {sentiment_mapping[selected]} star(s).") .. output :: https://doc-feedback-stars.streamlit.app/ height: 200px Display a feedback widget with thumbs, and show the selected sentiment: >>> import streamlit as st >>> >>> sentiment_mapping = [":material/thumb_down:", ":material/thumb_up:"] >>> selected = st.feedback("thumbs") >>> if selected is not None: >>> st.markdown(f"You selected: {sentiment_mapping[selected]}") .. output :: https://doc-feedback-thumbs.streamlit.app/ height: 200px )rarmrnzjThe options argument to st.feedback must be one of ['thumbs', 'faces', 'stars']. The argument passed was 'rrnNr3 borderless) rrrry deserializer serializerrrrr}r) rrtr6intrfSelectionVisualization ONLY_SELECTEDALL_UP_TO_SELECTED _button_grouprLrFrC) r?rVrryrrrtransformed_optionsrsr[r} sentimentr@r@rAr s6\  r3rv) rr format_funcrrrrrryrrrKOptionSequence[V]rLiteral['single']rV | NonerCallable[[Any], str] | Nonerrrrc Crr=r@ r?rrVrrrrrrrrryrr@r@rAruzButtonGroupMixin.pills) rrrrrrrryrLiteral['multi']Sequence[V] | V | Nonelist[V]c Crr=r@rr@r@rArurrurYlist[V] | V | Nonec C$|j|||||||d|| | | | d S)a`Display a pills widget. A pills widget is similar to a ``st.selectbox`` or ``st.multiselect`` where the ``options`` are displayed as pill-buttons instead of a drop-down list. Parameters ---------- label: str A short label explaining to the user what this widget is for. The label can optionally contain GitHub-flavored Markdown of the following types: Bold, Italics, Strikethroughs, Inline Code, Links, and Images. Images display like icons, with a max height equal to the font height. Unsupported Markdown elements are unwrapped so only their children (text contents) render. Display unsupported elements as literal characters by backslash-escaping them. E.g., ``"1\. Not an ordered list"``. See the ``body`` parameter of |st.markdown|_ for additional, supported Markdown directives. For accessibility reasons, you should never set an empty label, but you can hide it with ``label_visibility`` if needed. In the future, we may disallow empty labels by raising an exception. .. |st.markdown| replace:: ``st.markdown`` .. _st.markdown: https://docs.streamlit.io/develop/api-reference/text/st.markdown options: Iterable of V Labels for the select options in an ``Iterable``. This can be a ``list``, ``set``, or anything supported by ``st.dataframe``. If ``options`` is dataframe-like, the first column will be used. Each label will be cast to ``str`` internally by default and can optionally contain GitHub-flavored Markdown, including the Markdown directives described in the ``body`` parameter of ``st.markdown``. selection_mode: "single" or "multi" The selection mode for the widget. If this is ``"single"`` (default), only one option can be selected. If this is ``"multi"``, multiple options can be selected. default: Iterable of V, V, or None The value of the widget when it first renders. If the ``selection_mode`` is ``multi``, this can be a list of values, a single value, or ``None``. If the ``selection_mode`` is ``"single"``, this can be a single value or ``None``. format_func: function Function to modify the display of the options. It receives the raw option as an argument and should output the label to be shown for that option. This has no impact on the return value of the command. The output can optionally contain GitHub-flavored Markdown, including the Markdown directives described in the ``body`` parameter of ``st.markdown``. 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. help: str or None A tooltip that gets displayed next to the widget label. Streamlit only displays the tooltip when ``label_visibility="visible"``. If this is ``None`` (default), no tooltip is displayed. The tooltip can optionally contain GitHub-flavored Markdown, including the Markdown directives described in the ``body`` parameter of ``st.markdown``. on_change: callable An optional callback invoked when this widget's value changes. args: tuple An optional tuple of args to pass to the callback. kwargs: dict An optional dict of kwargs to pass to the callback. disabled: bool An optional boolean that disables the widget if set to ``True``. The default is ``False``. label_visibility: "visible", "hidden", or "collapsed" The visibility of the label. The default is ``"visible"``. If this is ``"hidden"``, Streamlit displays an empty spacer instead of the label, which can help keep the widget alligned with other widgets. If this is ``"collapsed"``, Streamlit displays no label or spacer. Returns ------- list of V, V, or None If the ``selection_mode`` is ``multi``, this is a list of selected options or an empty list. If the ``selection_mode`` is ``"single"``, this is a selected option or ``None``. Examples -------- **Example 1: Multi-select pills** Display a multi-select pills widget, and show the selection: >>> import streamlit as st >>> >>> options = ["North", "East", "South", "West"] >>> selection = st.pills("Directions", options, selection_mode="multi") >>> st.markdown(f"Your selected options: {selection}.") .. output:: https://doc-pills-multi.streamlit.app/ height: 200px **Example 2: Single-select pills with icons** Display a single-select pills widget with icons: >>> import streamlit as st >>> >>> option_map = { ... 0: ":material/add:", ... 1: ":material/zoom_in:", ... 2: ":material/zoom_out:", ... 3: ":material/zoom_out_map:", ... } >>> selection = st.pills( ... "Tool", ... options=option_map.keys(), ... format_func=lambda option: option_map[option], ... selection_mode="single", ... ) >>> st.write( ... "Your selected option: " ... f"{None if selection is None else option_map[selection]}" ... ) .. output:: https://doc-pills-single.streamlit.app/ height: 200px ru rrrrrrrrrrryr_internal_button_grouprr@r@rArus !str | int | Nonec Crr=r@rr@r@rAsegmented_controlWrz"ButtonGroupMixin.segmented_controlc Crr=r@rr@r@rArhrrc Cr)aDisplay a segmented control widget. A segmented control widget is a linear set of segments where each of the passed ``options`` functions like a toggle button. Parameters ---------- label : str A short label explaining to the user what this widget is for. The label can optionally contain GitHub-flavored Markdown of the following types: Bold, Italics, Strikethroughs, Inline Code, Links, and Images. Images display like icons, with a max height equal to the font height. Unsupported Markdown elements are unwrapped so only their children (text contents) render. Display unsupported elements as literal characters by backslash-escaping them. E.g., ``"1\. Not an ordered list"``. See the ``body`` parameter of |st.markdown|_ for additional, supported Markdown directives. For accessibility reasons, you should never set an empty label, but you can hide it with ``label_visibility`` if needed. In the future, we may disallow empty labels by raising an exception. .. |st.markdown| replace:: ``st.markdown`` .. _st.markdown: https://docs.streamlit.io/develop/api-reference/text/st.markdown options: Iterable of V Labels for the select options in an ``Iterable``. This can be a ``list``, ``set``, or anything supported by ``st.dataframe``. If ``options`` is dataframe-like, the first column will be used. Each label will be cast to ``str`` internally by default and can optionally contain GitHub-flavored Markdown, including the Markdown directives described in the ``body`` parameter of ``st.markdown``. selection_mode: "single" or "multi" The selection mode for the widget. If this is ``"single"`` (default), only one option can be selected. If this is ``"multi"``, multiple options can be selected. default: Iterable of V, V, or None The value of the widget when it first renders. If the ``selection_mode`` is ``multi``, this can be a list of values, a single value, or ``None``. If the ``selection_mode`` is ``"single"``, this can be a single value or ``None``. format_func: function Function to modify the display of the options. It receives the raw option as an argument and should output the label to be shown for that option. This has no impact on the return value of the command. The output can optionally contain GitHub-flavored Markdown, including the Markdown directives described in the ``body`` parameter of ``st.markdown``. 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. help: str or None A tooltip that gets displayed next to the widget label. Streamlit only displays the tooltip when ``label_visibility="visible"``. If this is ``None`` (default), no tooltip is displayed. The tooltip can optionally contain GitHub-flavored Markdown, including the Markdown directives described in the ``body`` parameter of ``st.markdown``. on_change: callable An optional callback invoked when this widget's value changes. args: tuple An optional tuple of args to pass to the callback. kwargs: dict An optional dict of kwargs to pass to the callback. disabled: bool An optional boolean that disables the widget if set to ``True``. The default is ``False``. label_visibility: "visible", "hidden", or "collapsed" The visibility of the label. The default is ``"visible"``. If this is ``"hidden"``, Streamlit displays an empty spacer instead of the label, which can help keep the widget alligned with other widgets. If this is ``"collapsed"``, Streamlit displays no label or spacer. Returns ------- list of V, V, or None If the ``selection_mode`` is ``multi``, this is a list of selected options or an empty list. If the ``selection_mode`` is ``"single"``, this is a selected option or ``None``. Examples -------- **Example 1: Multi-select segmented control** Display a multi-select segmented control widget, and show the selection: >>> import streamlit as st >>> >>> options = ["North", "East", "South", "West"] >>> selection = st.segmented_control( ... "Directions", options, selection_mode="multi" ... ) >>> st.markdown(f"Your selected options: {selection}.") .. output:: https://doc-segmented-control-multi.streamlit.app/ height: 200px **Example 2: Single-select segmented control with icons** Display a single-select segmented control widget with icons: >>> import streamlit as st >>> >>> option_map = { ... 0: ":material/add:", ... 1: ":material/zoom_in:", ... 2: ":material/zoom_out:", ... 3: ":material/zoom_out_map:", ... } >>> selection = st.segmented_control( ... "Tool", ... options=option_map.keys(), ... format_func=lambda option: option_map[option], ... selection_mode="single", ... ) >>> st.write( ... "Your selected option: " ... f"{None if selection is None else option_map[selection]}" ... ) .. output:: https://doc-segmented-control-single.streamlit.app/ height: 200px rrrrr@r@rArzs #r) rrrryrrrrrrrrr%Literal['pills', 'segmented_control']c szt| | d fdd }t|}t||}tt|||}|j||||||| ||j|j|| | | | d}|dkr:|jS|jS) Noptionr+r;ButtonGroupProto.Optioncsr|nt|}|d}d}t|dkrD|d}z|dr(t|}nt|r.|}|r9d|dd}Wn tyCYnwt j ||dS)zoIf option starts with a material icon or an emoji, we extract it to send it parsed to the frontend. Nrz :material)contentrd) rKsplitrMstrip startswithr!r joinrrfrg)r transformedtransformed_partsrj maybe_iconrr@rA_transformed_format_func@s(      zIButtonGroupMixin._internal_button_group.._transformed_format_func)rrryrrrrrrrrrrrr4)rr+r;r) rrrrUr+rrFrLrC)r?rVrrrryrrrrrrrrrindexable_optionsrWr[resr@rrAr,s6  z'ButtonGroupMixin._internal_button_group) rrrryrrrrrr}rrrr Sequence[Any]r:r5r-Callable[[V], ButtonGroupProto.Option] | NonerWidgetDeserializer[T]rWidgetSerializer[T]r}r~RegisterWidgetResult[T]c sft||dkr tjntj}|tjkr&|dur&t|tr&t|dkr&td|dvr2td|dt|}|}|durDt|dkrDd}t |j || |dd }t }t |j }dur]n fd d t D}t|||||||d }t|||pzg|||| ||||d }t|j| | | || |dd}|jr| |j|jdd<d|_|rt|||j |||S)Nr3rzYThe default argument to `st.pills` must be a single value when `selection_mode='single'`.)rrurziThe style argument must be one of ['borderless', 'pills', 'segmented_control']. The argument passed was 'rrrZ button_groupcsg|] \}}|qSr@r@)riindex_rrr@rArks z2ButtonGroupMixin._button_group..)user_keyrrVrr{r)r{r}rrrrint_array_value)on_change_handlerrrrrctx value_typeT)rrf SINGLE_SELECT MULTI_SELECT isinstancerrMrrrdgrr enumeraterrrr value_changedrC set_valuer_enqueue)r?rrrrryrrrrrrrr}rrrparsed_selection_mode_default widget_namerrrw element_idr widget_stater@rrArxs       zButtonGroupMixin._button_groupr$cCs td|S)zGet our DeltaGenerator.r$)r )r?r@r@rArs zButtonGroupMixin.dg).)rVrrrryrzrrrrrrr;r)rVrrrryrzrrrrrrr;r)ra)rVr_rrryrzrrrrrrr;r)rrKrVrrrrrrrrrrrrrrrrrryrzrrr;r)rrKrVrrrrrrrrrrrrrrrrrryrzrrr;r)rrKrVrrrYrrrrrrrrrrrrrrryrzrrr;r)rrKrVrrrrrrrrrrrrrrrrrryrzrrr;r)rrKrVrrrrrrrrrrrrrrrrrryrzrrr;r)rrKrVrrrYrrrrrrrrrrrrrrryrzrrr;r)rVrrrrrrrYryrzrrrrrrrrrrrrrrrrr;r)"rrrrrr:rr5ryrzrrrrrrrrrrrrrrr}r~rrrrrrr;r)r;r$)rQrRrSr rrrurrrfrrrpropertyrr@r@r@rArs  z02Our)r^r_r;r`)rJrKrwrxrWrEryrzrrKr{r|r}r~rrrrrrrrr;rf)rr5)L __future__rcollections.abcrtypingrrrrrr r r r typing_extensionsr !streamlit.elements.lib.form_utilsr-streamlit.elements.lib.options_selector_utilsrrstreamlit.elements.lib.policiesrrstreamlit.elements.lib.utilsrrrrrr&streamlit.elements.widgets.multiselectrstreamlit.errorsrstreamlit.proto.ButtonGroup_pb2rrfstreamlit.runtime.metrics_utilr7streamlit.runtime.scriptrunner_utils.script_run_contextrstreamlit.runtime.staterstreamlit.string_utilr r!streamlit.type_utilr"streamlit.dataframe_utilr#streamlit.delta_generatorr$r%r&r'streamlit.runtime.state.commonr(r)r*r+r,__annotations__r-r/r0r1r5r6rUrtrrrrrr@r@r@rAsN ,                " "% '