streamlit-nightly 1.38.1.dev20240902__py2.py3-none-any.whl → 1.38.1.dev20240903__py2.py3-none-any.whl

streamlit/components/v1/custom_component.py

@@ -22,6 +22,7 @@ from streamlit.dataframe_util import is_dataframe_like
 from streamlit.delta_generator_singletons import get_dg_singleton_instance
 from streamlit.elements.form_utils import current_form_id
 from streamlit.elements.lib.policies import check_cache_replay_rules
+from streamlit.elements.lib.utils import compute_and_register_element_id
 from streamlit.errors import StreamlitAPIException
 from streamlit.proto.Components_pb2 import ArrowTable as ArrowTableProto
 from streamlit.proto.Components_pb2 import SpecialArg
@@ -29,7 +30,6 @@ from streamlit.proto.Element_pb2 import Element
 from streamlit.runtime.metrics_util import gather_metrics
 from streamlit.runtime.scriptrunner_utils.script_run_context import get_script_run_ctx
 from streamlit.runtime.state import NoValue, register_widget
-from streamlit.runtime.state.common import compute_element_id
 from streamlit.type_util import is_bytes_like, to_bytes
 
 if TYPE_CHECKING:
@@ -172,7 +172,7 @@ And if you're using Streamlit Cloud, add "pyarrow" to your requirements.txt."""
 
             if key is None:
                 marshall_element_args()
-                computed_id = compute_element_id(
+                computed_id = compute_and_register_element_id(
                     "component_instance",
                     user_key=key,
                     name=self.name,
@@ -184,7 +184,7 @@ And if you're using Streamlit Cloud, add "pyarrow" to your requirements.txt."""
                     page=ctx.active_script_hash if ctx else None,
                 )
             else:
-                computed_id = compute_element_id(
+                computed_id = compute_and_register_element_id(
                     "component_instance",
                     user_key=key,
                     name=self.name,
@@ -202,8 +202,6 @@ And if you're using Streamlit Cloud, add "pyarrow" to your requirements.txt."""
             component_state = register_widget(
                 element_type="component_instance",
                 element_proto=element.component_instance,
-                user_key=key,
-                widget_func_name=self.name,
                 deserializer=deserialize_component,
                 serializer=lambda x: x,
                 ctx=ctx,

streamlit/delta_generator.py

@@ -417,6 +417,7 @@ class DeltaGenerator(
         delta_type: str,
         element_proto: Message,
         add_rows_metadata: AddRowsMetadata | None = None,
+        user_key: str | None = None,
     ) -> DeltaGenerator:
         """Create NewElement delta, fill it, and enqueue it.
 
@@ -426,6 +427,10 @@ class DeltaGenerator(
             The name of the streamlit method being called
         element_proto : proto
             The actual proto in the NewElement type e.g. Alert/Button/Slider
+        add_rows_metadata : AddRowsMetadata or None
+            Metadata for the add_rows method
+        user_key : str or None
+            A custom key for the element provided by the user.
 
         Returns
         -------

streamlit/elements/arrow.py

@@ -43,7 +43,7 @@ from streamlit.elements.lib.column_config_utils import (
 from streamlit.elements.lib.event_utils import AttributeDictionary
 from streamlit.elements.lib.pandas_styler_utils import marshall_styler
 from streamlit.elements.lib.policies import check_widget_policies
-from streamlit.elements.lib.utils import Key, to_key
+from streamlit.elements.lib.utils import Key, compute_and_register_element_id, to_key
 from streamlit.errors import StreamlitAPIException
 from streamlit.proto.Arrow_pb2 import Arrow as ArrowProto
 from streamlit.proto.ForwardMsg_pb2 import ForwardMsg
@@ -53,7 +53,6 @@ from streamlit.runtime.scriptrunner_utils.script_run_context import (
     get_script_run_ctx,
 )
 from streamlit.runtime.state import WidgetCallback, register_widget
-from streamlit.runtime.state.common import compute_element_id
 
 if TYPE_CHECKING:
     from numpy import typing as npt
@@ -566,7 +565,7 @@ class ArrowMixin:
             proto.form_id = current_form_id(self.dg)
 
             ctx = get_script_run_ctx()
-            proto.id = compute_element_id(
+            proto.id = compute_and_register_element_id(
                 "dataframe",
                 user_key=key,
                 data=proto.data,
@@ -586,7 +585,6 @@ class ArrowMixin:
             widget_state = register_widget(
                 "dataframe",
                 proto,
-                user_key=key,
                 on_change_handler=on_select if callable(on_select) else None,
                 deserializer=serde.deserialize,
                 serializer=serde.serialize,

streamlit/elements/layouts.py

@@ -19,6 +19,7 @@ from typing import TYPE_CHECKING, Literal, Sequence, Union, cast
 from typing_extensions import TypeAlias
 
 from streamlit.delta_generator_singletons import get_dg_singleton_instance
+from streamlit.elements.lib.utils import Key, compute_and_register_element_id, to_key
 from streamlit.errors import StreamlitAPIException
 from streamlit.proto.Block_pb2 import Block as BlockProto
 from streamlit.runtime.metrics_util import gather_metrics
@@ -35,7 +36,11 @@ SpecType: TypeAlias = Union[int, Sequence[Union[int, float]]]
 class LayoutsMixin:
     @gather_metrics("container")
     def container(
-        self, *, height: int | None = None, border: bool | None = None
+        self,
+        *,
+        height: int | None = None,
+        border: bool | None = None,
+        key: Key | None = None,
     ) -> DeltaGenerator:
         """Insert a multi-element container.
 
@@ -49,6 +54,7 @@ class LayoutsMixin:
 
         Parameters
         ----------
+
         height : int or None
             Desired height of the container expressed in pixels. If ``None`` (default)
             the container grows to fit its content. If a fixed height, scrolling is
@@ -66,6 +72,12 @@ class LayoutsMixin:
             border is shown if the container is set to a fixed height and not
             shown otherwise.
 
+        key : str or None
+            An optional string to give this container a stable identity.
+
+            Additionally, if ``key`` is provided, it will be used as CSS
+            class name prefixed with ``st-key-``.
+
 
         Examples
         --------
@@ -129,6 +141,7 @@ class LayoutsMixin:
             height: 400px
 
         """
+        key = to_key(key)
         block_proto = BlockProto()
         block_proto.allow_empty = False
         block_proto.vertical.border = border or False
@@ -143,6 +156,11 @@ class LayoutsMixin:
                 # containers.
                 block_proto.vertical.border = True
 
+        if key:
+            block_proto.id = compute_and_register_element_id(
+                "container", user_key=key, height=height, border=border
+            )
+
         return self.dg._block(block_proto)
 
     @gather_metrics("columns")

streamlit/elements/lib/utils.py

@@ -14,6 +14,8 @@
 
 from __future__ import annotations
 
+import hashlib
+from datetime import date, datetime, time, timedelta
 from enum import Enum, EnumMeta
 from typing import (
     TYPE_CHECKING,
@@ -27,15 +29,27 @@ from typing import (
     overload,
 )
 
+from google.protobuf.message import Message
 from typing_extensions import TypeAlias
 
 from streamlit import config, dataframe_util, errors, logger
+from streamlit.errors import StreamlitDuplicateElementId, StreamlitDuplicateElementKey
 from streamlit.proto.LabelVisibilityMessage_pb2 import LabelVisibilityMessage
-from streamlit.runtime.state.common import RegisterWidgetResult
+from streamlit.runtime.scriptrunner_utils.script_run_context import (
+    get_script_run_ctx,
+)
+from streamlit.runtime.state.common import (
+    GENERATED_ELEMENT_ID_PREFIX,
+    RegisterWidgetResult,
+    user_key_from_element_id,
+)
+from streamlit.util import HASHLIB_KWARGS
 
 if TYPE_CHECKING:
-    from streamlit.type_util import T
+    from builtins import ellipsis
 
+    from streamlit.runtime.state.widgets import NoValue
+    from streamlit.type_util import T
 
 _LOGGER: Final = logger.get_logger(__name__)
 
@@ -43,6 +57,19 @@ Key: TypeAlias = Union[str, int]
 
 LabelVisibility: TypeAlias = Literal["visible", "hidden", "collapsed"]
 
+PROTO_SCALAR_VALUE = Union[float, int, bool, str, bytes]
+SAFE_VALUES = Union[
+    date,
+    time,
+    datetime,
+    timedelta,
+    None,
+    "NoValue",
+    "ellipsis",
+    Message,
+    PROTO_SCALAR_VALUE,
+]
+
 
 def get_label_visibility_proto_value(
     label_visibility_string: LabelVisibility,
@@ -226,3 +253,94 @@ def _extract_common_class_from_iter(iterable: Iterable[Any]) -> Any:
     if all(type(item) is first_class for item in inner_iter):
         return first_class
     return None
+
+
+def _register_element_id(element_type: str, element_id: str) -> None:
+    """Register the element ID and key for the given element.
+
+    If the element ID or key is not unique, an error is raised.
+
+    Parameters
+    ----------
+
+    element_type : str
+        The type of the element to register.
+
+    element_id : str
+        The ID of the element to register.
+
+    Raises
+    ------
+
+    StreamlitDuplicateElementKey
+        If the element key is not unique.
+
+    StreamlitDuplicateElementID
+        If the element ID is not unique.
+
+    """
+    ctx = get_script_run_ctx()
+    if ctx is None or not element_id:
+        return
+
+    if user_key := user_key_from_element_id(element_id):
+        if user_key not in ctx.widget_user_keys_this_run:
+            ctx.widget_user_keys_this_run.add(user_key)
+        else:
+            raise StreamlitDuplicateElementKey(user_key)
+
+    if element_id not in ctx.widget_ids_this_run:
+        ctx.widget_ids_this_run.add(element_id)
+    else:
+        raise StreamlitDuplicateElementId(element_type)
+
+
+def _compute_element_id(
+    element_type: str,
+    user_key: str | None = None,
+    **kwargs: SAFE_VALUES | Iterable[SAFE_VALUES],
+) -> str:
+    """Compute the ID for the given element.
+
+    This ID is stable: a given set of inputs to this function will always produce
+    the same ID output. Only stable, deterministic values should be used to compute
+    element IDs. Using nondeterministic values as inputs can cause the resulting
+    element ID to change between runs.
+
+    The element ID includes the user_key so elements with identical arguments can
+    use it to be distinct. The element ID includes an easily identified prefix, and the
+    user_key as a suffix, to make it easy to identify it and know if a key maps to it.
+    """
+    h = hashlib.new("md5", **HASHLIB_KWARGS)
+    h.update(element_type.encode("utf-8"))
+    # This will iterate in a consistent order when the provided arguments have
+    # consistent order; dicts are always in insertion order.
+    for k, v in kwargs.items():
+        h.update(str(k).encode("utf-8"))
+        h.update(str(v).encode("utf-8"))
+    return f"{GENERATED_ELEMENT_ID_PREFIX}-{h.hexdigest()}-{user_key}"
+
+
+def compute_and_register_element_id(
+    element_type: str,
+    user_key: str | None = None,
+    **kwargs: SAFE_VALUES | Iterable[SAFE_VALUES],
+) -> str:
+    """Compute and register the ID for the given element.
+
+    This ID is stable: a given set of inputs to this function will always produce
+    the same ID output. Only stable, deterministic values should be used to compute
+    element IDs. Using nondeterministic values as inputs can cause the resulting
+    element ID to change between runs.
+
+    The element ID includes the user_key so elements with identical arguments can
+    use it to be distinct. The element ID includes an easily identified prefix, and the
+    user_key as a suffix, to make it easy to identify it and know if a key maps to it.
+
+    The element ID gets registered to make sure that only one ID and user-specified
+    key exists at the same time. If there are duplicated IDs or keys, an error
+    is raised.
+    """
+    element_id = _compute_element_id(element_type, user_key, **kwargs)
+    _register_element_id(element_type, element_id)
+    return element_id

streamlit/elements/media.py

@@ -24,13 +24,13 @@ from typing_extensions import TypeAlias
 
 from streamlit import runtime, type_util, url_util
 from streamlit.elements.lib.subtitle_utils import process_subtitle_data
+from streamlit.elements.lib.utils import compute_and_register_element_id
 from streamlit.errors import StreamlitAPIException
 from streamlit.proto.Audio_pb2 import Audio as AudioProto
 from streamlit.proto.Video_pb2 import Video as VideoProto
 from streamlit.runtime import caching
 from streamlit.runtime.metrics_util import gather_metrics
 from streamlit.runtime.scriptrunner_utils.script_run_context import get_script_run_ctx
-from streamlit.runtime.state.common import compute_element_id
 from streamlit.time_util import time_to_seconds
 from streamlit.type_util import NumpyShape
 
@@ -567,7 +567,7 @@ def marshall_video(
     if autoplay:
         ctx = get_script_run_ctx()
         proto.autoplay = autoplay
-        proto.id = compute_element_id(
+        proto.id = compute_and_register_element_id(
             "video",
             url=proto.url,
             mimetype=mimetype,
@@ -741,7 +741,7 @@ def marshall_audio(
     if autoplay:
         ctx = get_script_run_ctx()
         proto.autoplay = autoplay
-        proto.id = compute_element_id(
+        proto.id = compute_and_register_element_id(
             "audio",
             url=proto.url,
             mimetype=mimetype,

streamlit/elements/plotly_chart.py

@@ -42,13 +42,12 @@ from streamlit.elements.lib.policies import check_widget_policies
 from streamlit.elements.lib.streamlit_plotly_theme import (
     configure_streamlit_plotly_theme,
 )
-from streamlit.elements.lib.utils import Key, to_key
+from streamlit.elements.lib.utils import Key, compute_and_register_element_id, to_key
 from streamlit.errors import StreamlitAPIException
 from streamlit.proto.PlotlyChart_pb2 import PlotlyChart as PlotlyChartProto
 from streamlit.runtime.metrics_util import gather_metrics
 from streamlit.runtime.scriptrunner_utils.script_run_context import get_script_run_ctx
 from streamlit.runtime.state import WidgetCallback, register_widget
-from streamlit.runtime.state.common import compute_element_id
 
 if TYPE_CHECKING:
     import matplotlib
@@ -212,7 +211,9 @@ class PlotlyState(TypedDict, total=False):
 
 @dataclass
 class PlotlyChartSelectionSerde:
-    """PlotlyChartSelectionSerde is used to serialize and deserialize the Plotly Chart selection state."""
+    """PlotlyChartSelectionSerde is used to serialize and deserialize the Plotly Chart
+    selection state.
+    """
 
     def deserialize(self, ui_value: str | None, widget_id: str = "") -> PlotlyState:
         empty_selection_state: PlotlyState = {
@@ -442,18 +443,22 @@ class PlotlyMixin:
 
         if "sharing" in kwargs:
             show_deprecation_warning(
-                "The `sharing` parameter has been deprecated and will be removed in a future release. "
-                "Plotly charts will always be rendered using Streamlit's offline mode."
+                "The `sharing` parameter has been deprecated and will be removed "
+                "in a future release. Plotly charts will always be rendered using "
+                "Streamlit's offline mode."
             )
 
         if theme not in ["streamlit", None]:
             raise StreamlitAPIException(
-                f'You set theme="{theme}" while Streamlit charts only support theme=”streamlit” or theme=None to fallback to the default library theme.'
+                f'You set theme="{theme}" while Streamlit charts only support '
+                "theme=”streamlit” or theme=None to fallback to the default "
+                "library theme."
             )
 
         if on_select not in ["ignore", "rerun"] and not callable(on_select):
             raise StreamlitAPIException(
-                f"You have passed {on_select} to `on_select`. But only 'ignore', 'rerun', or a callable is supported."
+                f"You have passed {on_select} to `on_select`. But only 'ignore', "
+                "'rerun', or a callable is supported."
             )
 
         key = to_key(key)
@@ -498,7 +503,7 @@ class PlotlyMixin:
         # We are computing the widget id for all plotly uses
         # to also allow non-widget Plotly charts to keep their state
         # when the frontend component gets unmounted and remounted.
-        plotly_chart_proto.id = compute_element_id(
+        plotly_chart_proto.id = compute_and_register_element_id(
             "plotly_chart",
             user_key=key,
             key=key,
@@ -523,7 +528,6 @@ class PlotlyMixin:
             widget_state = register_widget(
                 "plotly_chart",
                 plotly_chart_proto,
-                user_key=key,
                 on_change_handler=on_select if callable(on_select) else None,
                 deserializer=serde.deserialize,
                 serializer=serde.serialize,

streamlit/elements/vega_charts.py

@@ -47,7 +47,7 @@ from streamlit.elements.lib.built_in_chart_utils import (
 )
 from streamlit.elements.lib.event_utils import AttributeDictionary
 from streamlit.elements.lib.policies import check_widget_policies
-from streamlit.elements.lib.utils import Key, to_key
+from streamlit.elements.lib.utils import Key, compute_and_register_element_id, to_key
 from streamlit.errors import StreamlitAPIException
 from streamlit.proto.ArrowVegaLiteChart_pb2 import (
     ArrowVegaLiteChart as ArrowVegaLiteChartProto,
@@ -55,7 +55,6 @@ from streamlit.proto.ArrowVegaLiteChart_pb2 import (
 from streamlit.runtime.metrics_util import gather_metrics
 from streamlit.runtime.scriptrunner_utils.script_run_context import get_script_run_ctx
 from streamlit.runtime.state import WidgetCallback, register_widget
-from streamlit.runtime.state.common import compute_element_id
 from streamlit.util import HASHLIB_KWARGS
 
 if TYPE_CHECKING:
@@ -1887,7 +1886,7 @@ class VegaChartsMixin:
             vega_lite_proto.form_id = current_form_id(self.dg)
 
             ctx = get_script_run_ctx()
-            vega_lite_proto.id = compute_element_id(
+            vega_lite_proto.id = compute_and_register_element_id(
                 "arrow_vega_lite_chart",
                 user_key=key,
                 key=key,
@@ -1910,7 +1909,6 @@ class VegaChartsMixin:
             widget_state = register_widget(
                 "vega_lite_chart",
                 vega_lite_proto,
-                user_key=key,
                 on_change_handler=on_select if callable(on_select) else None,
                 deserializer=serde.deserialize,
                 serializer=serde.serialize,

streamlit/elements/widgets/button.py

@@ -33,7 +33,7 @@ from typing_extensions import TypeAlias
 from streamlit import runtime
 from streamlit.elements.form_utils import current_form_id, is_in_form
 from streamlit.elements.lib.policies import check_widget_policies
-from streamlit.elements.lib.utils import Key, to_key
+from streamlit.elements.lib.utils import Key, compute_and_register_element_id, to_key
 from streamlit.errors import StreamlitAPIException
 from streamlit.file_util import get_main_script_directory, normalize_path_join
 from streamlit.navigation.page import StreamlitPage
@@ -49,7 +49,9 @@ from streamlit.runtime.state import (
     WidgetKwargs,
     register_widget,
 )
-from streamlit.runtime.state.common import compute_element_id, save_for_app_testing
+from streamlit.runtime.state.common import (
+    save_for_app_testing,
+)
 from streamlit.string_util import validate_icon_or_emoji
 from streamlit.url_util import is_url
 
@@ -662,7 +664,7 @@ class ButtonMixin:
             writes_allowed=False,
         )
 
-        element_id = compute_element_id(
+        element_id = compute_and_register_element_id(
             "download_button",
             user_key=key,
             label=label,
@@ -703,7 +705,6 @@ class ButtonMixin:
         button_state = register_widget(
             "download_button",
             download_button_proto,
-            user_key=key,
             on_change_handler=on_click,
             args=args,
             kwargs=kwargs,
@@ -843,7 +844,7 @@ class ButtonMixin:
             enable_check_callback_rules=not is_form_submitter,
         )
 
-        element_id = compute_element_id(
+        element_id = compute_and_register_element_id(
             "button",
             user_key=key,
             label=label,
@@ -892,7 +893,6 @@ class ButtonMixin:
         button_state = register_widget(
             "button",
             button_proto,
-            user_key=key,
             on_change_handler=on_click,
             args=args,
             kwargs=kwargs,

streamlit/elements/widgets/button_group.py

@@ -34,6 +34,7 @@ from streamlit.elements.lib.options_selector_utils import (
 from streamlit.elements.lib.policies import check_widget_policies
 from streamlit.elements.lib.utils import (
     Key,
+    compute_and_register_element_id,
     maybe_coerce_enum_sequence,
     to_key,
 )
@@ -47,7 +48,6 @@ from streamlit.runtime.state.common import (
     RegisterWidgetResult,
     WidgetDeserializer,
     WidgetSerializer,
-    compute_element_id,
     save_for_app_testing,
 )
 
@@ -396,7 +396,7 @@ class ButtonGroupMixin:
             if format_func is None
             else [format_func(option) for option in indexable_options]
         )
-        element_id = compute_element_id(
+        element_id = compute_and_register_element_id(
             widget_name,
             user_key=key,
             key=key,
@@ -420,7 +420,6 @@ class ButtonGroupMixin:
         widget_state = register_widget(
             widget_name,
             proto,
-            # user_key=key,
             on_change_handler=on_change,
             args=args,
             kwargs=kwargs,

streamlit/elements/widgets/camera_input.py

@@ -28,6 +28,7 @@ from streamlit.elements.lib.policies import (
 from streamlit.elements.lib.utils import (
     Key,
     LabelVisibility,
+    compute_and_register_element_id,
     get_label_visibility_proto_value,
     to_key,
 )
@@ -43,7 +44,6 @@ from streamlit.runtime.state import (
     WidgetKwargs,
     register_widget,
 )
-from streamlit.runtime.state.common import compute_element_id
 from streamlit.runtime.uploaded_file_manager import DeletedFile, UploadedFile
 
 if TYPE_CHECKING:
@@ -203,7 +203,7 @@ class CameraInputMixin:
         )
         maybe_raise_label_warnings(label, label_visibility)
 
-        element_id = compute_element_id(
+        element_id = compute_and_register_element_id(
             "camera_input",
             user_key=key,
             label=label,
@@ -230,7 +230,6 @@ class CameraInputMixin:
         camera_input_state = register_widget(
             "camera_input",
             camera_input_proto,
-            user_key=key,
             on_change_handler=on_change,
             args=args,
             kwargs=kwargs,

streamlit/elements/widgets/chat.py

@@ -23,7 +23,7 @@ from streamlit.delta_generator_singletons import get_dg_singleton_instance
 from streamlit.elements.form_utils import is_in_form
 from streamlit.elements.image import AtomicImage, WidthBehaviour, image_to_url
 from streamlit.elements.lib.policies import check_widget_policies
-from streamlit.elements.lib.utils import Key, to_key
+from streamlit.elements.lib.utils import Key, compute_and_register_element_id, to_key
 from streamlit.errors import StreamlitAPIException
 from streamlit.proto.Block_pb2 import Block as BlockProto
 from streamlit.proto.ChatInput_pb2 import ChatInput as ChatInputProto
@@ -37,7 +37,9 @@ from streamlit.runtime.state import (
     WidgetKwargs,
     register_widget,
 )
-from streamlit.runtime.state.common import compute_element_id, save_for_app_testing
+from streamlit.runtime.state.common import (
+    save_for_app_testing,
+)
 from streamlit.string_util import is_emoji, validate_material_icon
 
 if TYPE_CHECKING:
@@ -323,7 +325,7 @@ class ChatMixin:
         )
 
         ctx = get_script_run_ctx()
-        element_id = compute_element_id(
+        element_id = compute_and_register_element_id(
             "chat_input",
             user_key=key,
             key=key,
@@ -368,7 +370,6 @@ class ChatMixin:
         widget_state = register_widget(
             "chat_input",
             chat_input_proto,
-            user_key=key,
             on_change_handler=on_submit,
             args=args,
             kwargs=kwargs,

streamlit/elements/widgets/checkbox.py

@@ -26,6 +26,7 @@ from streamlit.elements.lib.policies import (
 from streamlit.elements.lib.utils import (
     Key,
     LabelVisibility,
+    compute_and_register_element_id,
     get_label_visibility_proto_value,
     to_key,
 )
@@ -38,7 +39,6 @@ from streamlit.runtime.state import (
     WidgetKwargs,
     register_widget,
 )
-from streamlit.runtime.state.common import compute_element_id
 
 if TYPE_CHECKING:
     from streamlit.delta_generator import DeltaGenerator
@@ -291,7 +291,7 @@ class CheckboxMixin:
         )
         maybe_raise_label_warnings(label, label_visibility)
 
-        element_id = compute_element_id(
+        element_id = compute_and_register_element_id(
             "toggle" if type == CheckboxProto.StyleType.TOGGLE else "checkbox",
             user_key=key,
             label=label,
@@ -321,7 +321,6 @@ class CheckboxMixin:
         checkbox_state = register_widget(
             "checkbox",
             checkbox_proto,
-            user_key=key,
             on_change_handler=on_change,
             args=args,
             kwargs=kwargs,

streamlit/elements/widgets/color_picker.py

@@ -27,6 +27,7 @@ from streamlit.elements.lib.policies import (
 from streamlit.elements.lib.utils import (
     Key,
     LabelVisibility,
+    compute_and_register_element_id,
     get_label_visibility_proto_value,
     to_key,
 )
@@ -40,7 +41,6 @@ from streamlit.runtime.state import (
     WidgetKwargs,
     register_widget,
 )
-from streamlit.runtime.state.common import compute_element_id
 
 if TYPE_CHECKING:
     from streamlit.delta_generator import DeltaGenerator
@@ -186,7 +186,7 @@ class ColorPickerMixin:
         )
         maybe_raise_label_warnings(label, label_visibility)
 
-        element_id = compute_element_id(
+        element_id = compute_and_register_element_id(
             "color_picker",
             user_key=key,
             label=label,
@@ -241,7 +241,6 @@ class ColorPickerMixin:
         widget_state = register_widget(
             "color_picker",
             color_picker_proto,
-            user_key=key,
             on_change_handler=on_change,
             args=args,
             kwargs=kwargs,