streamlit-nightly 1.33.1.dev20240416__py2.py3-none-any.whl → 1.33.1.dev20240418__py2.py3-none-any.whl

streamlit/components/v1/custom_component.py

@@ -20,6 +20,7 @@ from typing import TYPE_CHECKING, Any
 from streamlit import _main, type_util
 from streamlit.components.types.base_custom_component import BaseCustomComponent
 from streamlit.elements.form import current_form_id
+from streamlit.elements.utils import check_cache_replay_rules
 from streamlit.errors import StreamlitAPIException
 from streamlit.proto.Components_pb2 import ArrowTable as ArrowTableProto
 from streamlit.proto.Components_pb2 import SpecialArg
@@ -101,6 +102,7 @@ PyArrow. To do so locally:
 And if you're using Streamlit Cloud, add "pyarrow" to your requirements.txt."""
             )
 
+        check_cache_replay_rules()
         # In addition to the custom kwargs passed to the component, we also
         # send the special 'default' and 'key' params to the component
         # frontend.

streamlit/delta_generator.py

@@ -528,11 +528,6 @@ class DeltaGenerator(
                 "call your fragment function inside a `with st.sidebar` context manager."
             )
 
-        # Warn if we're called from within a legacy @st.cache function
-        legacy_caching.maybe_show_cached_st_function_warning(dg, delta_type)
-        # Warn if we're called from within @st.memo or @st.singleton
-        caching.maybe_show_cached_st_function_warning(dg, delta_type)
-
         # Warn if an element is being changed but the user isn't running the streamlit server.
         _maybe_print_use_warning()
 

streamlit/elements/__init__.py

@@ -11,60 +11,3 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
-WIDGETS = [
-    "button",
-    "camera_input",
-    "chat_input",
-    "checkbox",
-    "color_picker",
-    "component_instance",
-    "download_button",
-    "file_uploader",
-    "form",
-    "multiselect",
-    "number_input",
-    "radio",
-    "selectbox",
-    "select_slider",
-    "slider",
-    "text_input",
-    "text_area",
-    "time_input",
-    "date_input",
-]
-NONWIDGET_ELEMENTS = [
-    "alert",
-    "arrow_area_chart",
-    "arrow_bar_chart",
-    "arrow_data_frame",
-    "arrow_line_chart",
-    "arrow_table",
-    "arrow_vega_lite_chart",
-    "audio",
-    "balloons",
-    "bokeh_chart",
-    "code",
-    "deck_gl_json_chart",
-    "doc_string",
-    "empty",
-    "exception",
-    "graphviz_chart",
-    "heading",
-    "iframe",
-    "imgs",
-    "json",
-    # link_button unlike button and download_button is not a widget. It only sends a
-    # forward messages to the frontend, and not sends trigger value back.
-    "link_button",
-    "markdown",
-    "metric",
-    "page_link",
-    "plotly_chart",
-    "progress",
-    "pyplot",
-    "snow",
-    "text",
-    "video",
-    "write",
-]

streamlit/elements/arrow_vega_lite.py

@@ -155,7 +155,14 @@ def marshall(
         raise ValueError("Vega-Lite charts require a non-empty spec dict.")
 
     if "autosize" not in spec:
-        spec["autosize"] = {"type": "fit", "contains": "padding"}
+        # type fit does not work for many chart types. This change focuses
+        # on vconcat with use_container_width=True as there are unintended
+        # consequences of changing the default autosize for all charts.
+        # fit-x fits the width and height can be adjusted.
+        if "vconcat" in spec and use_container_width:
+            spec["autosize"] = {"type": "fit-x", "contains": "padding"}
+        else:
+            spec["autosize"] = {"type": "fit", "contains": "padding"}
 
     # Pull data out of spec dict when it's in a 'datasets' key:
     #   marshall(proto, {datasets: {foo: df1, bar: df2}, ...})

streamlit/elements/form.py

@@ -194,11 +194,15 @@ class FormMixin:
 
         """
         # Import this here to avoid circular imports.
-        from streamlit.elements.utils import check_session_state_rules
+        from streamlit.elements.utils import (
+            check_cache_replay_rules,
+            check_session_state_rules,
+        )
 
         if is_in_form(self.dg):
             raise StreamlitAPIException("Forms cannot be nested in other forms.")
 
+        check_cache_replay_rules()
         check_session_state_rules(default_value=None, key=key, writes_allowed=False)
 
         # A form is uniquely identified by its key.

streamlit/elements/spinner.py

@@ -42,20 +42,10 @@ def spinner(text: str = "In progress...", *, _cache: bool = False) -> Iterator[N
     >>> st.success('Done!')
 
     """
-    import streamlit.runtime.caching as caching
-    import streamlit.runtime.legacy_caching.caching as legacy_caching
     from streamlit.proto.Spinner_pb2 import Spinner as SpinnerProto
     from streamlit.string_util import clean_text
 
-    # @st.cache optionally uses spinner for long-running computations.
-    # Normally, streamlit warns the user when they call st functions
-    # from within an @st.cache'd function. But we do *not* want to show
-    # these warnings for spinner's message, so we create and mutate this
-    # message delta within the "suppress_cached_st_function_warning"
-    # context.
-    with legacy_caching.suppress_cached_st_function_warning():
-        with caching.suppress_cached_st_function_warning():
-            message = st.empty()
+    message = st.empty()
 
     # Set the message 0.5 seconds in the future to avoid annoying
     # flickering if this spinner runs too quickly.
@@ -68,12 +58,10 @@ def spinner(text: str = "In progress...", *, _cache: bool = False) -> Iterator[N
         def set_message():
             with display_message_lock:
                 if display_message:
-                    with legacy_caching.suppress_cached_st_function_warning():
-                        with caching.suppress_cached_st_function_warning():
-                            spinner_proto = SpinnerProto()
-                            spinner_proto.text = clean_text(text)
-                            spinner_proto.cache = _cache
-                            message._enqueue("spinner", spinner_proto)
+                    spinner_proto = SpinnerProto()
+                    spinner_proto.text = clean_text(text)
+                    spinner_proto.cache = _cache
+                    message._enqueue("spinner", spinner_proto)
 
         add_script_run_ctx(threading.Timer(DELAY_SECS, set_message)).start()
 
@@ -83,16 +71,14 @@ def spinner(text: str = "In progress...", *, _cache: bool = False) -> Iterator[N
         if display_message_lock:
             with display_message_lock:
                 display_message = False
-        with legacy_caching.suppress_cached_st_function_warning():
-            with caching.suppress_cached_st_function_warning():
-                if "chat_message" in set(message._active_dg._ancestor_block_types):
-                    # Temporary stale element fix:
-                    # For chat messages, we are resetting the spinner placeholder to an
-                    # empty container instead of an empty placeholder (st.empty) to have
-                    # it removed from the delta path. Empty containers are ignored in the
-                    # frontend since they are configured with allow_empty=False. This
-                    # prevents issues with stale elements caused by the spinner being
-                    # rendered only in some situations (e.g. for caching).
-                    message.container()
-                else:
-                    message.empty()
+            if "chat_message" in set(message._active_dg._ancestor_block_types):
+                # Temporary stale element fix:
+                # For chat messages, we are resetting the spinner placeholder to an
+                # empty container instead of an empty placeholder (st.empty) to have
+                # it removed from the delta path. Empty containers are ignored in the
+                # frontend since they are configured with allow_empty=False. This
+                # prevents issues with stale elements caused by the spinner being
+                # rendered only in some situations (e.g. for caching).
+                message.container()
+            else:
+                message.empty()

streamlit/elements/utils.py

@@ -20,7 +20,7 @@ from typing import TYPE_CHECKING, Any, Hashable, Iterable, Sequence, cast, overl
 import streamlit
 from streamlit import config, runtime, type_util
 from streamlit.elements.form import is_in_form
-from streamlit.errors import StreamlitAPIException
+from streamlit.errors import StreamlitAPIException, StreamlitAPIWarning
 from streamlit.proto.LabelVisibilityMessage_pb2 import LabelVisibilityMessage
 from streamlit.runtime.state import WidgetCallback, get_session_state
 from streamlit.runtime.state.common import RegisterWidgetResult
@@ -86,6 +86,41 @@ def check_session_state_rules(
         _shown_default_value_warning = True
 
 
+class CachedWidgetWarning(StreamlitAPIWarning):
+    def __init__(self):
+        super().__init__(
+            """
+Your script uses a widget command in a cached function
+(function decorated with `@st.cache_data` or `@st.cache_resource`).
+This code will only be called when we detect a cache "miss",
+which can lead to unexpected results.
+
+How to fix this:
+* Move all widget commands outside the cached function.
+* Or, if you know what you're doing, use `experimental_allow_widgets=True`
+in the cache decorator to enable widget replay and suppress this warning.
+"""
+        )
+
+
+def check_cache_replay_rules() -> None:
+    """Check if a widget is allowed to be used in the current context.
+    More specifically, this checks if the current context is inside a
+    cached function that disallows widget usage. If so, it raises a warning.
+
+    If there are other similar checks in the future, we could extend this
+    function to check for those as well. And rename it to check_widget_usage_rules.
+    """
+    if runtime.exists():
+        from streamlit.runtime.scriptrunner.script_run_context import get_script_run_ctx
+
+        ctx = get_script_run_ctx()
+        if ctx and ctx.disallow_cached_widget_usage:
+            # We use an exception here to show a proper stack trace
+            # that indicates to the user where the issue is.
+            streamlit.exception(CachedWidgetWarning())
+
+
 def get_label_visibility_proto_value(
     label_visibility_string: type_util.LabelVisibility,
 ) -> LabelVisibilityMessage.LabelVisibilityOptions.ValueType:

streamlit/elements/widgets/button.py

@@ -24,7 +24,6 @@ from typing_extensions import TypeAlias
 
 from streamlit import runtime, source_util
 from streamlit.elements.form import current_form_id, is_in_form
-from streamlit.elements.utils import check_callback_rules, check_session_state_rules
 from streamlit.errors import StreamlitAPIException
 from streamlit.file_util import get_main_script_directory, normalize_path_join
 from streamlit.proto.Button_pb2 import Button as ButtonProto
@@ -551,7 +550,17 @@ class ButtonMixin:
         ctx: ScriptRunContext | None = None,
     ) -> bool:
         key = to_key(key)
+
+        # Importing these functions here to avoid circular imports
+        from streamlit.elements.utils import (
+            check_cache_replay_rules,
+            check_callback_rules,
+            check_session_state_rules,
+        )
+
+        check_cache_replay_rules()
         check_session_state_rules(default_value=None, key=key, writes_allowed=False)
+        check_callback_rules(self.dg, on_click)
 
         id = compute_widget_id(
             "download_button",
@@ -704,8 +713,18 @@ class ButtonMixin:
         use_container_width: bool = False,
         ctx: ScriptRunContext | None = None,
     ) -> bool:
+        key = to_key(key)
+
+        # Importing these functions here to avoid circular imports
+        from streamlit.elements.utils import (
+            check_cache_replay_rules,
+            check_callback_rules,
+            check_session_state_rules,
+        )
+
         if not is_form_submitter:
             check_callback_rules(self.dg, on_click)
+        check_cache_replay_rules()
         check_session_state_rules(default_value=None, key=key, writes_allowed=False)
 
         id = compute_widget_id(

streamlit/elements/widgets/camera_input.py

@@ -22,6 +22,7 @@ from typing_extensions import TypeAlias
 
 from streamlit.elements.form import current_form_id
 from streamlit.elements.utils import (
+    check_cache_replay_rules,
     check_callback_rules,
     check_session_state_rules,
     get_label_visibility_proto_value,
@@ -195,6 +196,8 @@ class CameraInputMixin:
         ctx: ScriptRunContext | None = None,
     ) -> UploadedFile | None:
         key = to_key(key)
+
+        check_cache_replay_rules()
         check_callback_rules(self.dg, on_change)
         check_session_state_rules(default_value=None, key=key, writes_allowed=False)
         maybe_raise_label_warnings(label, label_visibility)

streamlit/elements/widgets/chat.py

@@ -21,7 +21,6 @@ from typing import TYPE_CHECKING, Literal, cast
 from streamlit import runtime
 from streamlit.elements.form import is_in_form
 from streamlit.elements.image import AtomicImage, WidthBehaviour, image_to_url
-from streamlit.elements.utils import check_callback_rules, check_session_state_rules
 from streamlit.errors import StreamlitAPIException
 from streamlit.proto.Block_pb2 import Block as BlockProto
 from streamlit.proto.ChatInput_pb2 import ChatInput as ChatInputProto
@@ -289,6 +288,14 @@ class ChatMixin:
         # We default to an empty string here and disallow user choice intentionally
         default = ""
         key = to_key(key)
+
+        from streamlit.elements.utils import (
+            check_cache_replay_rules,
+            check_callback_rules,
+            check_session_state_rules,
+        )
+
+        check_cache_replay_rules()
         check_callback_rules(self.dg, on_submit)
         check_session_state_rules(default_value=default, key=key, writes_allowed=False)
 

streamlit/elements/widgets/checkbox.py

@@ -20,6 +20,7 @@ from typing import TYPE_CHECKING, cast
 
 from streamlit.elements.form import current_form_id
 from streamlit.elements.utils import (
+    check_cache_replay_rules,
     check_callback_rules,
     check_session_state_rules,
     get_label_visibility_proto_value,
@@ -274,11 +275,12 @@ class CheckboxMixin:
         ctx: ScriptRunContext | None = None,
     ) -> bool:
         key = to_key(key)
+
+        check_cache_replay_rules()
         check_callback_rules(self.dg, on_change)
         check_session_state_rules(
             default_value=None if value is False else value, key=key
         )
-
         maybe_raise_label_warnings(label, label_visibility)
 
         id = compute_widget_id(

streamlit/elements/widgets/color_picker.py

@@ -22,6 +22,7 @@ from typing import cast
 import streamlit
 from streamlit.elements.form import current_form_id
 from streamlit.elements.utils import (
+    check_cache_replay_rules,
     check_callback_rules,
     check_session_state_rules,
     get_label_visibility_proto_value,
@@ -169,6 +170,8 @@ class ColorPickerMixin:
         ctx: ScriptRunContext | None = None,
     ) -> str:
         key = to_key(key)
+
+        check_cache_replay_rules()
         check_callback_rules(self.dg, on_change)
         check_session_state_rules(default_value=value, key=key)
         maybe_raise_label_warnings(label, label_visibility)

streamlit/elements/widgets/data_editor.py

@@ -55,7 +55,6 @@ from streamlit.elements.lib.column_config_utils import (
     update_column_config,
 )
 from streamlit.elements.lib.pandas_styler_utils import marshall_styler
-from streamlit.elements.utils import check_callback_rules, check_session_state_rules
 from streamlit.errors import StreamlitAPIException
 from streamlit.proto.Arrow_pb2 import Arrow as ArrowProto
 from streamlit.runtime.metrics_util import gather_metrics
@@ -774,10 +773,20 @@ class DataEditorMixin:
            height: 350px
 
         """
+        # Lazy-loaded import
         import pandas as pd
         import pyarrow as pa
 
+        # Import here to avoid cyclic import warning
+        from streamlit.elements.utils import (
+            check_cache_replay_rules,
+            check_callback_rules,
+            check_session_state_rules,
+        )
+
         key = to_key(key)
+
+        check_cache_replay_rules()
         check_callback_rules(self.dg, on_change)
         check_session_state_rules(default_value=None, key=key, writes_allowed=False)
 

streamlit/elements/widgets/file_uploader.py

@@ -23,6 +23,7 @@ from typing_extensions import TypeAlias
 from streamlit import config
 from streamlit.elements.form import current_form_id
 from streamlit.elements.utils import (
+    check_cache_replay_rules,
     check_callback_rules,
     check_session_state_rules,
     get_label_visibility_proto_value,
@@ -398,6 +399,8 @@ class FileUploaderMixin:
         ctx: ScriptRunContext | None = None,
     ) -> UploadedFile | list[UploadedFile] | None:
         key = to_key(key)
+
+        check_cache_replay_rules()
         check_callback_rules(self.dg, on_change)
         check_session_state_rules(default_value=None, key=key, writes_allowed=False)
         maybe_raise_label_warnings(label, label_visibility)

streamlit/elements/widgets/multiselect.py

@@ -20,6 +20,7 @@ from typing import TYPE_CHECKING, Any, Callable, Generic, Sequence, cast, overlo
 
 from streamlit.elements.form import current_form_id
 from streamlit.elements.utils import (
+    check_cache_replay_rules,
     check_callback_rules,
     check_session_state_rules,
     get_label_visibility_proto_value,
@@ -82,15 +83,19 @@ def _check_and_convert_to_indices(
             default_values, "pandas.core.series.Series"
         ):
             default_values = list(cast(Sequence[Any], default_values))
-        elif not default_values or default_values in opt:
-            default_values = [default_values]
-        else:
+        elif (
+            isinstance(default_values, (tuple, set))
+            or default_values
+            and default_values not in opt
+        ):
             default_values = list(default_values)
-
+        else:
+            default_values = [default_values]
     for value in default_values:
         if value not in opt:
             raise StreamlitAPIException(
-                "Every Multiselect default value must exist in options"
+                f"The default value '{value}' is part of the options. "
+                "Please make sure that every default values also exists in the options."
             )
 
     return [opt.index(value) for value in default_values]
@@ -282,12 +287,14 @@ class MultiSelectMixin:
         ctx: ScriptRunContext | None = None,
     ) -> list[T]:
         key = to_key(key)
+
+        check_cache_replay_rules()
         check_callback_rules(self.dg, on_change)
         check_session_state_rules(default_value=default, key=key)
+        maybe_raise_label_warnings(label, label_visibility)
 
         opt = ensure_indexable(options)
         check_python_comparable(opt)
-        maybe_raise_label_warnings(label, label_visibility)
 
         indices = _check_and_convert_to_indices(opt, default)
 

streamlit/elements/widgets/number_input.py

@@ -23,6 +23,7 @@ from typing_extensions import TypeAlias
 
 from streamlit.elements.form import current_form_id
 from streamlit.elements.utils import (
+    check_cache_replay_rules,
     check_callback_rules,
     check_session_state_rules,
     get_label_visibility_proto_value,
@@ -36,6 +37,7 @@ from streamlit.runtime.state import (
     WidgetArgs,
     WidgetCallback,
     WidgetKwargs,
+    get_session_state,
     register_widget,
 )
 from streamlit.runtime.state.common import compute_widget_id
@@ -277,6 +279,8 @@ class NumberInputMixin:
         ctx: ScriptRunContext | None = None,
     ) -> Number | None:
         key = to_key(key)
+
+        check_cache_replay_rules()
         check_callback_rules(self.dg, on_change)
         check_session_state_rules(
             default_value=value if value != "min" else None, key=key
@@ -320,6 +324,10 @@ class NumberInputMixin:
                 f"\n`step` has {type(step).__name__} type."
             )
 
+        session_state = get_session_state().filtered_state
+        if key is not None and key in session_state and session_state[key] is None:
+            value = None
+
         if value == "min":
             if min_value is not None:
                 value = min_value

streamlit/elements/widgets/radio.py

@@ -20,6 +20,7 @@ from typing import TYPE_CHECKING, Any, Callable, Generic, Sequence, cast
 
 from streamlit.elements.form import current_form_id
 from streamlit.elements.utils import (
+    check_cache_replay_rules,
     check_callback_rules,
     check_session_state_rules,
     get_label_visibility_proto_value,
@@ -33,6 +34,7 @@ from streamlit.runtime.state import (
     WidgetArgs,
     WidgetCallback,
     WidgetKwargs,
+    get_session_state,
     register_widget,
 )
 from streamlit.runtime.state.common import compute_widget_id, save_for_app_testing
@@ -249,9 +251,12 @@ class RadioMixin:
         ctx: ScriptRunContext | None,
     ) -> T | None:
         key = to_key(key)
+
+        check_cache_replay_rules()
         check_callback_rules(self.dg, on_change)
         check_session_state_rules(default_value=None if index == 0 else index, key=key)
         maybe_raise_label_warnings(label, label_visibility)
+
         opt = ensure_indexable(options)
         check_python_comparable(opt)
 
@@ -289,6 +294,10 @@ class RadioMixin:
                     f"Radio captions must be strings. Passed type: {type(caption).__name__}"
                 )
 
+        session_state = get_session_state().filtered_state
+        if key is not None and key in session_state and session_state[key] is None:
+            index = None
+
         radio_proto = RadioProto()
         radio_proto.id = id
         radio_proto.label = label

streamlit/elements/widgets/select_slider.py

@@ -22,6 +22,7 @@ from typing_extensions import TypeGuard
 
 from streamlit.elements.form import current_form_id
 from streamlit.elements.utils import (
+    check_cache_replay_rules,
     check_callback_rules,
     check_session_state_rules,
     get_label_visibility_proto_value,
@@ -256,9 +257,12 @@ class SelectSliderMixin:
         ctx: ScriptRunContext | None = None,
     ) -> T | tuple[T, T]:
         key = to_key(key)
+
+        check_cache_replay_rules()
         check_callback_rules(self.dg, on_change)
         check_session_state_rules(default_value=value, key=key)
         maybe_raise_label_warnings(label, label_visibility)
+
         opt = ensure_indexable(options)
         check_python_comparable(opt)
 

streamlit/elements/widgets/selectbox.py

@@ -19,6 +19,7 @@ from typing import TYPE_CHECKING, Any, Callable, Generic, Sequence, cast
 
 from streamlit.elements.form import current_form_id
 from streamlit.elements.utils import (
+    check_cache_replay_rules,
     check_callback_rules,
     check_session_state_rules,
     get_label_visibility_proto_value,
@@ -32,6 +33,7 @@ from streamlit.runtime.state import (
     WidgetArgs,
     WidgetCallback,
     WidgetKwargs,
+    get_session_state,
     register_widget,
 )
 from streamlit.runtime.state.common import compute_widget_id, save_for_app_testing
@@ -229,9 +231,10 @@ class SelectboxMixin:
         ctx: ScriptRunContext | None = None,
     ) -> T | None:
         key = to_key(key)
+
+        check_cache_replay_rules()
         check_callback_rules(self.dg, on_change)
         check_session_state_rules(default_value=None if index == 0 else index, key=key)
-
         maybe_raise_label_warnings(label, label_visibility)
 
         opt = ensure_indexable(options)
@@ -260,6 +263,10 @@ class SelectboxMixin:
                 "Selectbox index must be between 0 and length of options"
             )
 
+        session_state = get_session_state().filtered_state
+        if key is not None and key in session_state and session_state[key] is None:
+            index = None
+
         selectbox_proto = SelectboxProto()
         selectbox_proto.id = id
         selectbox_proto.label = label

streamlit/elements/widgets/slider.py

@@ -24,6 +24,7 @@ from typing_extensions import TypeAlias
 
 from streamlit.elements.form import current_form_id
 from streamlit.elements.utils import (
+    check_cache_replay_rules,
     check_callback_rules,
     check_session_state_rules,
     get_label_visibility_proto_value,
@@ -364,9 +365,10 @@ class SliderMixin:
         ctx: ScriptRunContext | None = None,
     ) -> SliderReturn:
         key = to_key(key)
+
+        check_cache_replay_rules()
         check_callback_rules(self.dg, on_change)
         check_session_state_rules(default_value=value, key=key)
-
         maybe_raise_label_warnings(label, label_visibility)
 
         id = compute_widget_id(