U ARd0@sddlmZddlmZddlmZmZmZmZm Z m Z m Z m Z m Z mZddlmZddlmZmZmZddlmZddlmZddlmZdd lmZmZdd lm Z m!Z!m"Z"m#Z#dd l$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-erdd l.m/Z/ee ed e e e0dddZ1ee ee e eefe e0dddZ1e ee e eed fe e e0dddZ1e e eed fe0dddZ2e0e0dddZ3eGdddee(Z4GdddZ5d S)) dataclass)dedent) TYPE_CHECKINGAnyCallableGenericListOptionalSequenceUnioncastoverload)current_form_id)check_callback_rulescheck_session_state_rules get_label_visibility_proto_value)StreamlitAPIException) MultiSelect)gather_metrics)ScriptRunContextget_script_run_ctx) WidgetArgsWidgetCallback WidgetKwargsregister_widget) KeyLabelVisibilityOptionSequenceTensure_indexable is_iterableis_typemaybe_raise_label_warningsto_key)DeltaGeneratorN)optdefault_valuesreturncCsdSNr%r&r)r)B/tmp/pip-unpacked-wheel-b9et7o5g/streamlit/elements/multiselect.py_check_and_convert_to_indices>sr,cCsdSr(r)r*r)r)r+r,Escs|dkrdkrdSt|tsbt|ds2t|drFtttt|}n|rR|krZ|g}nt|}|D]}|krftdqffdd|DS)zIPerform validation checks and return indices based on the default values.Nz numpy.ndarrayzpandas.core.series.Seriesz5Every Multiselect default value must exist in optionscsg|]}|qSr))index).0valuer%r)r+ fsz1_check_and_convert_to_indices..) isinstancelistr!r r rr)r%r&r/r)r0r+r,Ls"   )defaultr'cCs*|dkr dSt|sdStttt|S)Nr)r lenr r r)r4r)r)r+_get_default_countis r7)current_selectionsmax_selectionsc CsD|dkr dnd}|dkrdnd}d|d|d|d|d|d S) Nr5optionoptionsz Multiselect has  z) selected but `max_selections` is set to z. This happened because you either gave too many options to `default` or you manipulated the widget's state through `st.session_state`. Note that the latter can happen before the line indicated in the traceback. Please select at most z. r))r8r9Zcurr_selections_nounZmax_selections_nounr)r)r+_get_over_max_options_messageqsr=c@s^eZdZUeeed<eeed<eeeedddZd e eee eeddd Z d S) MultiSelectSerder; default_value)r/r'cCs t|j|Sr()r,r;)selfr/r)r)r+ serializeszMultiSelectSerde.serialize)ui_value widget_idr'cs$|dk r |nj}fdd|DS)Ncsg|]}j|qSr))r;)r.ir@r)r+r1sz0MultiSelectSerde.deserialize..)r?)r@rCrDZ current_valuer)rFr+ deserializeszMultiSelectSerde.deserializeN)rB) __name__ __module__ __qualname__r r__annotations__rintrAr strrGr)r)r)r+r>}s    r>c@seZdZeddedddddfddddeeeeee egefee eeee ee ee eeeeeed ddZdedddddfddddd eeeeeeedfe egefee eeee ee ee eeeeeeeed d d Zed dddZdS)MultiSelectMixin multiselectNFZvisible)disabledlabel_visibilityr9) labelr;r4 format_funckeyhelp on_changeargskwargsrPrQr9r'c Cs*t} |j||||||||| | | | | d S)aDisplay a multiselect widget. The multiselect widget starts as empty. Parameters ---------- label : str A short label explaining to the user what this select widget is for. The label can optionally contain Markdown and supports the following elements: Bold, Italics, Strikethroughs, Inline Code, Emojis, and Links. This also supports: * Emoji shortcodes, such as ``:+1:`` and ``:sunglasses:``. For a list of all supported codes, see https://share.streamlit.io/streamlit/emoji-shortcodes. * LaTeX expressions, by wrapping them in "$" or "$$" (the "$$" must be on their own lines). Supported LaTeX functions are listed at https://katex.org/docs/supported.html. * Colored text, using the syntax ``:color[text to be colored]``, where ``color`` needs to be replaced with any of the following supported colors: blue, green, orange, red, violet. Unsupported 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``. For accessibility reasons, you should never set an empty label (label="") but hide it with label_visibility if needed. In the future, we may disallow empty labels by raising an exception. options : Sequence[V], numpy.ndarray, pandas.Series, pandas.DataFrame, or pandas.Index Labels for the select options. This will be cast to str internally by default. For pandas.DataFrame, the first column is selected. default: [V], V, or None List of default values. Can also be a single value. format_func : function Function to modify the display of selectbox 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 multiselect. 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 An optional tooltip that gets displayed next to the multiselect. on_change : callable An optional callback invoked when this multiselect'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, which disables the multiselect widget if set to True. The default is False. This argument can only be supplied by keyword. label_visibility : "visible" or "hidden" or "collapsed" The visibility of the label. If "hidden", the label doesn't show but there is still empty space for it above the widget (equivalent to label=""). If "collapsed", both the label and the space are removed. Default is "visible". This argument can only be supplied by keyword. max_selections : int The max selections that can be selected at a time. This argument can only be supplied by keyword. Returns ------- list A list with the selected options Example ------- >>> import streamlit as st >>> >>> options = st.multiselect( ... 'What are your favorite colors', ... ['Green', 'Yellow', 'Red', 'Blue'], ... ['Yellow', 'Red']) >>> >>> st.write('You selected:', options) .. output:: https://doc-multiselect.streamlitapp.com/ height: 420px ) rRr;r4rSrTrUrVrWrXrPrQctxr9)r _multiselect)r@rRr;r4rSrTrUrVrWrXrPrQr9rYr)r)r+rOs izMultiSelectMixin.multiselect)rPrQrYr9)rRr;r4rSrTrUrVrWrXrPrQrYr9r'c  s@t|}t|j|t||dt|}t|| t||}t}||_|dkrTgn|}||j dd<fdd|D|j dd<t |j|_ | pd|_ |dk rt||_t||}td||||| |j|j| d }t|j}| r|| krtt|| | |_t| |j_|jr,||j|jdd<d|_|jd||jS)N)r?rTcsg|]}t|qSr))rM)r.r:rSr)r+r1(sz1MultiSelectMixin._multiselect..rrO)Zuser_keyZon_change_handlerrWrXZ deserializer serializerrYT)r#rdgrrr"r,MultiSelectProtorRr4r;rZform_idr9rrUr>rrGrAr7r/rr=rPrrQ value_changed set_valueZ_enqueue)r@rRr;r4rSrTrUrVrWrXrPrQrYr9r%indicesZmultiselect_protor?ZserdeZ widget_stateZ default_countr)r[r+rZ sR          zMultiSelectMixin._multiselectr$)r'cCs td|S)zGet our DeltaGenerator.r$)r rFr)r)r+r]NszMultiSelectMixin.dg)rHrIrJrrMrrr rrrrrrboolrrLrrOr r rrZpropertyr]r)r)r)r+rNsr  }   CrN)6Z dataclassesrtextwraprtypingrrrrrr r r r r Zstreamlit.elements.formrZstreamlit.elements.utilsrrrZstreamlit.errorsrZstreamlit.proto.MultiSelect_pb2rr^Zstreamlit.runtime.metrics_utilrZstreamlit.runtime.scriptrunnerrrZstreamlit.runtime.staterrrrZstreamlit.type_utilrrrrrr r!r"r#Zstreamlit.delta_generatorr$rLr,r7r=r>rNr)r)r)r+s>  0    ,