g6OUddlmZddlZddlmZddlmZmZmZm Z m Z m Z m Z ddl mZddlmZddlmZddlmZdd lmZdd lmZmZmZdd lmZdd lmZdd l m!Z!ddl"m#Z#ddl$m%Z%m&Z&erddl'm(Z(m)Z)ddl*m+Z+ddl,m-Z-ddddddiZ.de/d<e dZ0de/d<ddhZ1de/d< d+dZ2Gd d!e d"#Z3Gd$d%e d"#Z4eGd&d'Z5Gd(d)Z6d,d*Z7y)-) annotationsN) dataclass) TYPE_CHECKINGAnyFinalLiteral TypedDictcastoverload) TypeAlias)config)AttributeDictionary)current_form_id)check_widget_policies)Keycompute_and_register_element_idto_key)StreamlitAPIException)DeckGlJsonChart)gather_metrics)get_script_run_ctx)WidgetCallbackregister_widget)IterableMapping)Deck)DeltaGeneratorinitialViewState)latitude longitudepitchzoomzFinal[Mapping[str, Any]] EMPTY_MAP) single-object multi-objectr SelectionModer%r&zFinal[set[SelectionMode]]_SELECTION_MODESct|tr|h}ntd|dd|jtstd|dt|j ddhr tdg}|D]`}|dk(r*|j tjj2|dk(s8|j tjjbt|S)z2Parse and check the user provided selection modes.zInvalid selection mode: z. z=Selection mode must be a single value, but got a set instead.z. Valid options are: r%r&zPOnly one of `single-object` or `multi-object` can be selected as selection mode.) isinstancestrrissubsetr( issupersetappend PydeckProtor' SINGLE_OBJECT MULTI_OBJECTset)selection_modeselection_mode_setparsed_selection_modess Z/var/www/openai/venv/lib/python3.12/site-packages/streamlit/elements/deck_gl_json_chart.pyparse_selection_moder7As.#&,- $&~&6b 9 K   & &'7 8#&~&67""2!3 5  $$o~%FG# ^   , _ , " ) )+*C*C*Q*Q R ~ - " ) )+*C*C*P*P Q - % &&c&eZdZUdZded<ded<y)PydeckSelectionStatea4 The schema for the PyDeck chart selection state. The selection state is stored in a dictionary-like object that supports both key and attribute notation. Selection states cannot be programmatically changed or set through Session State. You must define ``id`` in ``pydeck.Layer`` to ensure statefulness when using selections with ``st.pydeck_chart``. Attributes ---------- indices : dict[str, list[int]] A dictionary of selected objects by layer. Each key in the dictionary is a layer id, and each value is a list of object indices within that layer. objects : dict[str, list[dict[str, Any]]] A dictionary of object attributes by layer. Each key in the dictionary is a layer id, and each value is a list of metadata dictionaries for the selected objects in that layer. Examples -------- The following example has multi-object selection enabled. The chart displays US state capitals by population (2023 US Census estimate). You can access this `data `_ from GitHub. >>> import streamlit as st >>> import pydeck >>> import pandas as pd >>> >>> capitals = pd.read_csv( ... "capitals.csv", ... header=0, ... names=[ ... "Capital", ... "State", ... "Abbreviation", ... "Latitude", ... "Longitude", ... "Population", ... ], ... ) >>> capitals["size"] = capitals.Population / 10 >>> >>> point_layer = pydeck.Layer( ... "ScatterplotLayer", ... data=capitals, ... id="capital-cities", ... get_position=["Longitude", "Latitude"], ... get_color="[255, 75, 75]", ... pickable=True, ... auto_highlight=True, ... get_radius="size", ... ) >>> >>> view_state = pydeck.ViewState( ... latitude=40, longitude=-117, controller=True, zoom=2.4, pitch=30 ... ) >>> >>> chart = pydeck.Deck( ... point_layer, ... initial_view_state=view_state, ... tooltip={"text": "{Capital}, {Abbreviation}\nPopulation: {Population}"}, ... ) >>> >>> event = st.pydeck_chart(chart, on_select="rerun", selection_mode="multi-object") >>> >>> event.selection .. output :: https://doc-pydeck-event-state-selections.streamlit.app/ height: 700px This is an example of the selection state when selecting a single object from a layer with id, ``"captial-cities"``: >>> { >>> "indices":{ >>> "capital-cities":[ >>> 2 >>> ] >>> }, >>> "objects":{ >>> "capital-cities":[ >>> { >>> "Abbreviation":" AZ" >>> "Capital":"Phoenix" >>> "Latitude":33.448457 >>> "Longitude":-112.073844 >>> "Population":1650070 >>> "State":" Arizona" >>> "size":165007.0 >>> } >>> ] >>> } >>> } zdict[str, list[int]]indiceszdict[str, list[dict[str, Any]]]objectsN__name__ __module__ __qualname____doc____annotations__r8r6r:r:fsdL"! ,,r8r:F)totalceZdZUdZded<y) PydeckStateaQ The schema for the PyDeck event state. The event state is stored in a dictionary-like object that supports both key and attribute notation. Event states cannot be programmatically changed or set through Session State. Only selection events are supported at this time. Attributes ---------- selection : dict The state of the ``on_select`` event. This attribute returns a dictionary-like object that supports both key and attribute notation. The attributes are described by the ``PydeckSelectionState`` dictionary schema. r: selectionNr=rCr8r6rFrFs&$#r8rFc"eZdZdZdddZddZy)PydeckSelectionSerdezUPydeckSelectionSerde is used to serialize and deserialize the Pydeck selection state.cdiidi}||ntj|}d|vr|}ttt |S)NrG)r;r<)jsonloadsr rFr)selfui_value widget_idempty_selection_stateselection_states r6 deserializez PydeckSelectionSerde.deserializesU . &.%5 !4::h;O  o -3OK!4_!EFFr8c8tj|tS)N)default)rKdumpsr+)rMrQs r6 serializezPydeckSelectionSerde.serializeszz/377r8N))rNz str | NonerOr+returnrF)rQrFrXr+)r>r?r@rArRrVrCr8r6rIrIs_G(8r8rIc eZdZe d ddddd ddZe d ddddddd ddZed  d ddddd dd dd Zedd Zy) PydeckMixinNT)use_container_widthwidthheightkeycyNrCrM pydeck_objr[r\r]r3 on_selectr^s r6 pydeck_chartzPydeckMixin.pydeck_chartsr8r%rerun)r[r\r]r3rcr^cyr`rCras r6rdzPydeckMixin.pydeck_chartsr8rdignorec  t}t} |tjt} n|j } | |_||_|r||_|r||_t|} | rtj| |_ tjd} | r| |_ t|}|dk7} |dvrt|st!d|d| r|j"j%t'|t|}t)|j*||rt-t.|nddd|t1|j*|_t5d || ||| |j2 |_t9}t;|j6| |j<t|r|nd|j>d }|j*jAd |t-tB|jDS|j*jAd |S) aDraw a chart using the PyDeck library. This supports 3D maps, point clouds, and more! More info about PyDeck at https://deckgl.readthedocs.io/en/latest/. These docs are also quite useful: - DeckGL docs: https://github.com/uber/deck.gl/tree/master/docs - DeckGL JSON docs: https://github.com/uber/deck.gl/tree/master/modules/json When using this command, Mapbox provides the map tiles to render map content. Note that Mapbox is a third-party product and Streamlit accepts no responsibility or liability of any kind for Mapbox or for any content or information made available by Mapbox. Mapbox requires users to register and provide a token before users can request map tiles. Currently, Streamlit provides this token for you, but this could change at any time. We strongly recommend all users create and use their own personal Mapbox token to avoid any disruptions to their experience. You can do this with the ``mapbox.token`` config option. The use of Mapbox is governed by Mapbox's Terms of Use. To get a token for yourself, create an account at https://mapbox.com. For more info on how to set config options, see https://docs.streamlit.io/develop/api-reference/configuration/config.toml. Parameters ---------- pydeck_obj : pydeck.Deck or None Object specifying the PyDeck chart to draw. use_container_width : bool Whether to override the figure's native width with the width of the parent container. If ``use_container_width`` is ``True`` (default), Streamlit sets the width of the figure to match the width of the parent container. If ``use_container_width`` is ``False``, Streamlit sets the width of the chart to fit its contents according to the plotting library, up to the width of the parent container. width : int or None Desired width of the chart expressed in pixels. If ``width`` is ``None`` (default), Streamlit sets the width of the chart to fit its contents according to the plotting library, up to the width of the parent container. If ``width`` is greater than the width of the parent container, Streamlit sets the chart width to match the width of the parent container. To use ``width``, you must set ``use_container_width=False``. height : int or None Desired height of the chart expressed in pixels. If ``height`` is ``None`` (default), Streamlit sets the height of the chart to fit its contents according to the plotting library. on_select : "ignore" or "rerun" or callable How the figure should respond to user selection events. This controls whether or not the chart behaves like an input widget. ``on_select`` can be one of the following: - ``"ignore"`` (default): Streamlit will not react to any selection events in the chart. The figure will not behave like an input widget. - ``"rerun"``: Streamlit will rerun the app when the user selects data in the chart. In this case, ``st.pydeck_chart`` will return the selection data as a dictionary. - A ``callable``: Streamlit will rerun the app and execute the callable as a callback function before the rest of the app. In this case, ``st.pydeck_chart`` will return the selection data as a dictionary. If ``on_select`` is not ``"ignore"``, all layers must have a declared ``id`` to keep the chart stateful across reruns. selection_mode : "single-object" or "multi-object" The selection mode of the chart. This can be one of the following: - ``"single-object"`` (default): Only one object can be selected at a time. - ``"multi-object"``: Multiple objects can be selected at a time. key : str An optional string to use for giving this element a stable identity. If ``key`` is ``None`` (default), this element's identity will be determined based on the values of the other parameters. Additionally, if selections are activated and ``key`` is provided, Streamlit will register the key in Session State to store the selection state. The selection state is read-only. Returns ------- element or dict If ``on_select`` is ``"ignore"`` (default), this command returns an internal placeholder for the chart element. Otherwise, this method returns a dictionary-like object that supports both key and attribute notation. The attributes are described by the ``PydeckState`` dictionary schema. Example ------- Here's a chart using a HexagonLayer and a ScatterplotLayer. It uses either the light or dark map style, based on which Streamlit theme is currently active: >>> import streamlit as st >>> import pandas as pd >>> import numpy as np >>> import pydeck as pdk >>> >>> chart_data = pd.DataFrame( ... np.random.randn(1000, 2) / [50, 50] + [37.76, -122.4], ... columns=["lat", "lon"], ... ) >>> >>> st.pydeck_chart( ... pdk.Deck( ... map_style=None, ... initial_view_state=pdk.ViewState( ... latitude=37.76, ... longitude=-122.4, ... zoom=11, ... pitch=50, ... ), ... layers=[ ... pdk.Layer( ... "HexagonLayer", ... data=chart_data, ... get_position="[lon, lat]", ... radius=200, ... elevation_scale=4, ... elevation_range=[0, 1000], ... pickable=True, ... extruded=True, ... ), ... pdk.Layer( ... "ScatterplotLayer", ... data=chart_data, ... get_position="[lon, lat]", ... get_color="[200, 30, 0, 160]", ... get_radius=200, ... ), ... ], ... ) ... ) .. output:: https://doc-pydeck-chart.streamlit.app/ height: 530px .. note:: To make the PyDeck chart's style consistent with Streamlit's theme, you can set ``map_style=None`` in the ``pydeck.Deck`` object. Nz mapbox.tokenrg)rgrezYou have passed zH to `on_select`. But only 'ignore', 'rerun', or a callable is supported.F) on_change default_valuewrites_allowedenable_check_callback_rulesdeck_gl_json_chart)user_keyis_selection_activatedr3r[specform_id string_value)ctx deserializeron_change_handler serializer value_type)#r/rrKrUr$to_jsonr[r\r]_get_pydeck_tooltiptooltipr get_option mapbox_tokenrcallablerr3extendr7rdgr rrrqridrIrrRrV_enqueuerFvalue)rMrbr[r\r]r3rcr^ pydeck_protorsrprzr|ro is_callbackserde widget_states r6rdzPydeckMixin.pydeck_chart!s@#}  "  ::i(D%%'D +> ( !&L  "(L %j1 #'::g#6L ((8 (4L %Sk!*h!6 / /8K'"9+-uv  "  ' ' . ./CN/S T#9-K !=H$~y9d"$,7  $3477#;L =$'=-$7$,,LO)*E*"../7 /B) ??) L GG  1< @ \%7%78 8ww 4lCCr8ctd|S)zGet our DeltaGenerator.r)r )rMs r6rzPydeckMixin.dg s$d++r8r`)rb Deck | Noner[boolr\ int | Noner]rr3zLiteral['single-object']rczLiteral['ignore']r^ Key | NonerXr)rbrr[rr\rr]rr3r'rcz!Literal['rerun'] | WidgetCallbackr^rrXrF)rbrr[rr\rr]rr3r'rcz+Literal['rerun', 'ignore'] | WidgetCallbackr^rrXzDeltaGenerator | PydeckState)rXr)r>r?r@r rdrpropertyrrCr8r6rZrZs #' %) !   "       %     #' %) !(77>  "     & 5     N##'iD%) !(7AIiDiD" iD  iD  iD&iD?iDiD &iD$iDV,,r8rZc|yt|dd}|&t|jtr |jSt|dd}|-t|trt tt t f|Sy)N deck_widget_tooltip)getattrr*rzdictr r+)rb desk_widgetrzs r6ryryss*mT:K:k.A.A4#H"""j*d3Gz'48DcNG,, r8)r3z'SelectionMode | Iterable[SelectionMode]rXz(set[PydeckProto.SelectionMode.ValueType])rbrrXzdict[str, str] | None)8 __future__rrK dataclassesrtypingrrrrr r r typing_extensionsr streamlitr "streamlit.elements.lib.event_utilsr!streamlit.elements.lib.form_utilsrstreamlit.elements.lib.policiesrstreamlit.elements.lib.utilsrrrstreamlit.errorsr#streamlit.proto.DeckGlJsonChart_pb2rr/streamlit.runtime.metrics_utilr7streamlit.runtime.scriptrunner_utils.script_run_contextrstreamlit.runtime.staterrcollections.abcrrpydeckrstreamlit.delta_generatorrr$rBr'r(r7r:rFrIrZryrCr8r6rs# !(B=AUU2N9V 18 QQAN' ###BC yC/+ "';"'-"'Jh-9Eh-V$)5$. 88 86L,L,^r8