streamlit-nightly 1.33.1.dev20240409__py2.py3-none-any.whl → 1.33.1.dev20240414__py2.py3-none-any.whl

streamlit/__init__.py

@@ -72,6 +72,10 @@ from streamlit.delta_generator import (
     event_dg as _event_dg,
     bottom_dg as _bottom_dg,
 )
+from streamlit.elements.dialog_decorator import (
+    # rename so that it is available as st.dialog
+    dialog_decorator as experimental_dialog,
+)
 from streamlit.runtime.caching import (
     cache_resource as _cache_resource,
     cache_data as _cache_data,
@@ -129,7 +133,6 @@ _config.on_config_parsed(_update_logger, True)
 secrets = _secrets_singleton
 
 # DeltaGenerator methods:
-
 _main = _main_dg
 sidebar = _sidebar_dg
 _event = _event_dg

streamlit/delta_generator.py

@@ -26,6 +26,7 @@ from typing import (
     Final,
     Hashable,
     Iterable,
+    List,
     Literal,
     NoReturn,
     TypeVar,
@@ -314,9 +315,9 @@ class DeltaGenerator(
         if self == self._main_dg:
             # We're being invoked via an `st.foo` pattern - use the current
             # `with` dg (aka the top of the stack).
-            current_stack = dg_stack.get()
-            if len(current_stack) > 1:
-                return current_stack[-1]
+            last_context_stack_dg = get_last_dg_added_to_context_stack()
+            if last_context_stack_dg is not None:
+                return last_context_stack_dg
 
         # We're being invoked via an `st.sidebar.foo` pattern - ignore the
         # current `with` dg.
@@ -606,34 +607,7 @@ class DeltaGenerator(
         block_type = block_proto.WhichOneof("type")
         # Convert the generator to a list, so we can use it multiple times.
         ancestor_block_types = list(dg._ancestor_block_types)
-
-        if block_type == "column":
-            num_of_parent_columns = self._count_num_of_parent_columns(
-                ancestor_block_types
-            )
-            if (
-                self._root_container == RootContainer.SIDEBAR
-                and num_of_parent_columns > 0
-            ):
-                raise StreamlitAPIException(
-                    "Columns cannot be placed inside other columns in the sidebar. This is only possible in the main area of the app."
-                )
-            if num_of_parent_columns > 1:
-                raise StreamlitAPIException(
-                    "Columns can only be placed inside other columns up to one level of nesting."
-                )
-        if block_type == "chat_message" and block_type in ancestor_block_types:
-            raise StreamlitAPIException(
-                "Chat messages cannot nested inside other chat messages."
-            )
-        if block_type == "expandable" and block_type in ancestor_block_types:
-            raise StreamlitAPIException(
-                "Expanders may not be nested inside other expanders."
-            )
-        if block_type == "popover" and block_type in ancestor_block_types:
-            raise StreamlitAPIException(
-                "Popovers may not be nested inside other popovers."
-            )
+        _check_nested_element_violation(self, block_type, ancestor_block_types)
 
         if dg._root_container is None or dg._cursor is None:
             return dg
@@ -684,11 +658,9 @@ class DeltaGenerator(
     def _arrow_add_rows(
         self: DG,
         data: Data = None,
-        **kwargs: DataFrame
-        | npt.NDArray[Any]
-        | Iterable[Any]
-        | dict[Hashable, Any]
-        | None,
+        **kwargs: (
+            DataFrame | npt.NDArray[Any] | Iterable[Any] | dict[Hashable, Any] | None
+        ),
     ) -> DG | None:
         """Concatenate a dataframe to the bottom of the current one.
 
@@ -814,6 +786,20 @@ dg_stack: ContextVar[tuple[DeltaGenerator, ...]] = ContextVar(
 )
 
 
+def get_last_dg_added_to_context_stack() -> DeltaGenerator | None:
+    """Get the last added DeltaGenerator of the stack in the current context.
+
+    Returns None if the stack has only one element or is empty for whatever reason.
+    """
+    current_stack = dg_stack.get()
+    # If set to "> 0" and thus return the only delta generator in the stack - which logically makes more sense -, some unit tests
+    # fail. It looks like the reason is that they create their own main delta generator but do not populate the dg_stack correctly. However, to be on the safe-side,
+    # we keep the logic but leave the comment as shared knowledge for whoever will look into this in the future.
+    if len(current_stack) > 1:
+        return current_stack[-1]
+    return None
+
+
 def _prep_data_for_add_rows(
     data: Data,
     delta_type: str,
@@ -927,3 +913,35 @@ def _writes_directly_to_sidebar(dg: DG) -> bool:
     in_sidebar = any(a._root_container == RootContainer.SIDEBAR for a in dg._ancestors)
     has_container = bool(len(list(dg._ancestor_block_types)))
     return in_sidebar and not has_container
+
+
+def _check_nested_element_violation(
+    dg: DeltaGenerator, block_type: str | None, ancestor_block_types: List[BlockType]
+) -> None:
+    """Check if elements are nested in a forbidden way.
+
+    Raises
+    ------
+      StreamlitAPIException: throw if an invalid element nesting is detected.
+    """
+
+    if block_type == "column":
+        num_of_parent_columns = dg._count_num_of_parent_columns(ancestor_block_types)
+        if dg._root_container == RootContainer.SIDEBAR and num_of_parent_columns > 0:
+            raise StreamlitAPIException(
+                "Columns cannot be placed inside other columns in the sidebar. This is only possible in the main area of the app."
+            )
+        if num_of_parent_columns > 1:
+            raise StreamlitAPIException(
+                "Columns can only be placed inside other columns up to one level of nesting."
+            )
+    if block_type == "chat_message" and block_type in ancestor_block_types:
+        raise StreamlitAPIException(
+            "Chat messages cannot nested inside other chat messages."
+        )
+    if block_type == "expandable" and block_type in ancestor_block_types:
+        raise StreamlitAPIException(
+            "Expanders may not be nested inside other expanders."
+        )
+    if block_type == "popover" and block_type in ancestor_block_types:
+        raise StreamlitAPIException("Popovers may not be nested inside other popovers.")

streamlit/elements/dialog_decorator.py

@@ -0,0 +1,166 @@
+# Copyright (c) Streamlit Inc. (2018-2022) Snowflake Inc. (2022-2024)
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# 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.
+
+from __future__ import annotations
+
+from functools import wraps
+from typing import Callable, TypeVar, cast, overload
+
+from streamlit.delta_generator import event_dg, get_last_dg_added_to_context_stack
+from streamlit.elements.lib.dialog import DialogWidth
+from streamlit.errors import StreamlitAPIException
+from streamlit.runtime.fragment import fragment as _fragment
+from streamlit.runtime.metrics_util import gather_metrics
+
+
+def _assert_no_nested_dialogs() -> None:
+    """Check the current stack for existing DeltaGenerator's of type 'dialog'.
+    Note that the check like this only works when Dialog is called as a context manager, as this populates the dg_stack in delta_generator correctly.
+
+    This does not detect the edge case in which someone calls, for example, `with st.sidebar` inside of a dialog function and opens a dialog in there,
+    as `with st.sidebar` pushes the new DeltaGenerator to the stack. In order to check for that edge case, we could try to check all DeltaGenerators in the stack,
+    and not only the last one. Since we deem this to be an edge case, we lean towards simplicity here.
+
+    Raises
+    ------
+    StreamlitAPIException
+        Raised if the user tries to nest dialogs inside of each other.
+    """
+    last_dg_in_current_context = get_last_dg_added_to_context_stack()
+    if last_dg_in_current_context and "dialog" in set(
+        last_dg_in_current_context._ancestor_block_types
+    ):
+        raise StreamlitAPIException("Dialogs may not be nested inside other dialogs.")
+
+
+F = TypeVar("F", bound=Callable[..., None])
+
+
+def _dialog_decorator(
+    non_optional_func: F, title: str, *, width: DialogWidth = "small"
+) -> F:
+    if title is None or title == "":
+        raise StreamlitAPIException(
+            'A non-empty `title` argument has to be provided for dialogs, for example `@st.experimental_dialog("Example Title")`.'
+        )
+
+    @wraps(non_optional_func)
+    def wrap(*args, **kwargs) -> None:
+        _assert_no_nested_dialogs()
+        # Call the Dialog on the event_dg because it lives outside of the normal
+        # Streamlit UI flow. For example, if it is called from the sidebar, it should not
+        # inherit the sidebar theming.
+        dialog = event_dg._dialog(title=title, dismissible=True, width=width)
+        dialog.open()
+
+        @_fragment
+        def dialog_content() -> None:
+            # if the dialog should be closed, st.rerun() has to be called (same behavior as with st.fragment)
+            _ = non_optional_func(*args, **kwargs)
+            return None
+
+        with dialog:
+            return dialog_content()
+
+    return cast(F, wrap)
+
+
+@overload
+def dialog_decorator(title: str, *, width: DialogWidth = "small") -> Callable[[F], F]:
+    ...
+
+
+# 'title' can be a function since `dialog_decorator` is a decorator. We just call it 'title' here though
+# to make the user-doc more friendly as we want the user to pass a title, not a function.
+# The user is supposed to call it like @st.dialog("my_title") , which makes 'title' a positional arg, hence
+# this 'trick'. The overload is required to have a good type hint for the decorated function args.
+@overload
+def dialog_decorator(title: F | None, *, width: DialogWidth = "small") -> F:
+    ...
+
+
+@gather_metrics("experimental_dialog")
+def dialog_decorator(
+    title: F | None | str = "", *, width: DialogWidth = "small"
+) -> F | Callable[[F], F]:
+    r"""Decorate a function to mark it as a Streamlit dialog. When the decorated function is called, a dialog element is inserted with the function's body as the content.
+
+    The decorated function can hold multiple elements which are rendered inside of a modal when the decorated function is called.
+    The decorated function is using `st.experimental_fragment`, which means that interacting with elements inside of the dialog will
+    only re-run the dialog function.
+
+    The decorated function can accept arguments that can be passed when it is called.
+
+    Dismissing a dialog does not cause an app re-run.
+    You can close the dialog programmatically by executing `st.rerun()` explicitly inside of the decorated function.
+
+    In order to pass state from dialog widgets to the app, you can leverage `st.session_state`.
+
+    .. warning::
+        Currently, a dialog may not open another dialog.
+        Also, only one dialog-decorated function may be called in a script run, which means that only one dialog can be open at any given time.
+
+    Parameters
+    ----------
+    title : str
+        A string that will be used as the dialog's title. It cannot be empty.
+    width : "small", "large"
+        The width of the dialog. Defaults to "small".
+
+    Returns
+    -------
+    A decorated function that, when called, inserts a dialog element context container. The container itself contains the decorated function's elements.
+
+    Examples
+    --------
+    You can annotate a function to mark it as a Streamlit dialog function and pass arguments to it. You can either dismiss the dialog via the ESC-key or the X-button or close it programmatically and trigger a re-run by using `st.rerun()`.
+    Leverage `st.session_state` if you want to pass dialog widget states to the overall app:
+
+    >>> import streamlit as st
+    >>>
+    >>> @st.experimental_dialog("Streamlit Example Dialog")
+    >>> def example_dialog(some_arg: str, some_other_arg: int):
+    >>>     st.write(f"You passed following args: {some_arg} | {some_other_arg}")
+    >>>     # interacting with the text_input only re-runs `example_dialog`
+    >>>     some_text_input = st.text_input("Type something:", key="example_dialog_some_text_input")
+    >>>     # following write is updated when chaning the text_input inside the dialog
+    >>>     st.write(f"You wrote '{some_text_input}' in the dialog")
+    >>>     if st.button("Close the dialog"):
+    >>>         st.rerun()
+    >>>
+    >>> if st.button("Open dialog"):
+    >>>     example_dialog("Some string arg", 42)
+    >>>
+    >>> # following write is updated with the dialog's text input when the dialog was opened, the text input was interacted with and a re-run was triggered, e.g. by clicking the Close-button defined in `example_dialog`
+    >>> st.write(f"You wrote '{st.session_state.get('example_dialog_some_text_input', '')}' in the dialog")
+
+    """
+
+    func_or_title = title
+    if func_or_title is None:
+        # Support passing the params via function decorator
+        def wrapper(f: F) -> F:
+            return _dialog_decorator(non_optional_func=f, title="", width=width)
+
+        return wrapper
+    elif type(func_or_title) is str:
+        # Support passing the params via function decorator
+        def wrapper(f: F) -> F:
+            title: str = func_or_title
+            return _dialog_decorator(non_optional_func=f, title=title, width=width)
+
+        return wrapper
+
+    func: F = cast(F, func_or_title)
+    return _dialog_decorator(func, "", width=width)

streamlit/elements/image.py

@@ -296,9 +296,11 @@ def _ensure_image_size_and_format(
     if width > 0 and actual_width > width:
         # We need to resize the image.
         new_height = int(1.0 * actual_height * width / actual_width)
-        pil_image = pil_image.resize(
-            (width, new_height), resample=Image.Resampling.BILINEAR
-        )
+        # pillow reexports Image.Resampling.BILINEAR as Image.BILINEAR for backwards
+        # compatibility reasons, so we use the reexport to support older pillow
+        # versions. The types don't seem to reflect this, though, hence the type: ignore
+        # below.
+        pil_image = pil_image.resize((width, new_height), resample=Image.BILINEAR)  # type: ignore[attr-defined]
         return _PIL_to_bytes(pil_image, format=image_format, quality=90)
 
     if pil_image.format != image_format:

streamlit/elements/layouts.py

@@ -24,6 +24,7 @@ from streamlit.runtime.metrics_util import gather_metrics
 
 if TYPE_CHECKING:
     from streamlit.delta_generator import DeltaGenerator
+    from streamlit.elements.lib.dialog import Dialog
     from streamlit.elements.lib.mutable_status_container import StatusContainer
 
 SpecType: TypeAlias = Union[int, Sequence[Union[int, float]]]
@@ -710,6 +711,24 @@ class LayoutsMixin:
             self.dg, label=label, expanded=expanded, state=state
         )
 
+    def _dialog(
+        self,
+        title: str,
+        *,
+        dismissible: bool = True,
+        width: Literal["small", "large"] = "small",
+    ) -> "Dialog":
+        """Inserts the dialog container.
+
+        Marked as internal because it is used by the dialog_decorator and is not supposed to be used directly.
+        The dialog_decorator also has a more descriptive docstring since it is user-facing.
+        """
+
+        # We need to import Dialog here to avoid a circular import
+        from streamlit.elements.lib.dialog import Dialog
+
+        return Dialog._create(self.dg, title, dismissible=dismissible, width=width)
+
     @property
     def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""

streamlit/elements/lib/dialog.py

@@ -0,0 +1,148 @@
+# Copyright (c) Streamlit Inc. (2018-2022) Snowflake Inc. (2022-2024)
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# 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.
+
+from __future__ import annotations
+
+import time
+from types import TracebackType
+from typing import Literal, cast
+
+from typing_extensions import TypeAlias
+
+from streamlit.cursor import Cursor
+from streamlit.delta_generator import DeltaGenerator, _enqueue_message
+from streamlit.errors import StreamlitAPIException
+from streamlit.proto.Block_pb2 import Block as BlockProto
+from streamlit.proto.ForwardMsg_pb2 import ForwardMsg
+from streamlit.runtime.scriptrunner import get_script_run_ctx
+
+DialogWidth: TypeAlias = Literal["small", "large"]
+
+
+def _process_dialog_width_input(
+    width: DialogWidth,
+) -> BlockProto.Dialog.DialogWidth.ValueType:
+    """Maps the user-provided literal to a value of the DialogWidth proto enum.
+
+    Returns the mapped enum field for "small" by default and otherwise the mapped type.
+    """
+    if width == "large":
+        return BlockProto.Dialog.DialogWidth.LARGE
+
+    return BlockProto.Dialog.DialogWidth.SMALL
+
+
+def _assert_first_dialog_to_be_opened(should_open: bool) -> None:
+    """Check whether a dialog has already been opened in the same script run.
+
+    Only one dialog is supposed to be opened. The check is implemented in a way
+    that for a script run, the open function can only be called once.
+    One dialog at a time is a product decision and not a technical one.
+
+    Raises
+    ------
+    StreamlitAPIException
+        Raised when a dialog has already been opened in the current script run.
+    """
+    script_run_ctx = get_script_run_ctx()
+    # We don't reset the ctx.has_dialog_opened when the flag is False because
+    # it is reset in a new scriptrun anyways. If the execution model ever changes,
+    # this might need to change.
+    if should_open and script_run_ctx:
+        if script_run_ctx.has_dialog_opened:
+            raise StreamlitAPIException(
+                "Only one dialog is allowed to be opened at the same time. Please make sure to not call a dialog-decorated function more than once in a script run."
+            )
+        script_run_ctx.has_dialog_opened = True
+
+
+class Dialog(DeltaGenerator):
+    @staticmethod
+    def _create(
+        parent: DeltaGenerator,
+        title: str,
+        *,
+        dismissible: bool = True,
+        width: DialogWidth = "small",
+    ) -> Dialog:
+        block_proto = BlockProto()
+        block_proto.dialog.title = title
+        block_proto.dialog.dismissible = dismissible
+        block_proto.dialog.width = _process_dialog_width_input(width)
+
+        # We store the delta path here, because in _update we enqueue a new proto message to update the
+        # open status. Without this, the dialog content is gone when the _update message is sent
+        delta_path: list[int] = (
+            parent._active_dg._cursor.delta_path if parent._active_dg._cursor else []
+        )
+        dialog = cast(Dialog, parent._block(block_proto=block_proto, dg_type=Dialog))
+
+        dialog._delta_path = delta_path
+        dialog._current_proto = block_proto
+        # We add a sleep here to give the web app time to react to the update. Otherwise,
+        #  we might run into issues where the dialog cannot be opened again after closing
+        time.sleep(0.05)
+        return dialog
+
+    def __init__(
+        self,
+        root_container: int | None,
+        cursor: Cursor | None,
+        parent: DeltaGenerator | None,
+        block_type: str | None,
+    ):
+        super().__init__(root_container, cursor, parent, block_type)
+
+        # Initialized in `_create()`:
+        self._current_proto: BlockProto | None = None
+        self._delta_path: list[int] | None = None
+
+    def _update(self, should_open: bool):
+        """Send an updated proto message to indicate the open-status for the dialog."""
+
+        assert self._current_proto is not None, "Dialog not correctly initialized!"
+        assert self._delta_path is not None, "Dialog not correctly initialized!"
+        _assert_first_dialog_to_be_opened(should_open)
+        msg = ForwardMsg()
+        msg.metadata.delta_path[:] = self._delta_path
+        msg.delta.add_block.CopyFrom(self._current_proto)
+        msg.delta.add_block.dialog.is_open = should_open
+
+        self._current_proto = msg.delta.add_block
+
+        # We add a sleep here to give the web app time to react to the update. Otherwise,
+        #  we might run into issues where the dialog cannot be opened again after closing
+        time.sleep(0.05)
+        _enqueue_message(msg)
+
+    def open(self) -> None:
+        self._update(True)
+
+    def close(self) -> None:
+        self._update(False)
+
+    def __enter__(self) -> Dialog:  # type: ignore[override]
+        # This is a little dubious: we're returning a different type than
+        # our superclass' `__enter__` function. Maybe DeltaGenerator.__enter__
+        # should always return `self`?
+        super().__enter__()
+        return self
+
+    def __exit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_val: BaseException | None,
+        exc_tb: TracebackType | None,
+    ) -> Literal[False]:
+        return super().__exit__(exc_type, exc_val, exc_tb)

streamlit/proto/Block_pb2.py

@@ -13,7 +13,7 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1bstreamlit/proto/Block.proto\"\xfd\x06\n\x05\x42lock\x12#\n\x08vertical\x18\x01 \x01(\x0b\x32\x0f.Block.VerticalH\x00\x12\'\n\nhorizontal\x18\x02 \x01(\x0b\x32\x11.Block.HorizontalH\x00\x12\x1f\n\x06\x63olumn\x18\x03 \x01(\x0b\x32\r.Block.ColumnH\x00\x12\'\n\nexpandable\x18\x04 \x01(\x0b\x32\x11.Block.ExpandableH\x00\x12\x1b\n\x04\x66orm\x18\x05 \x01(\x0b\x32\x0b.Block.FormH\x00\x12,\n\rtab_container\x18\x06 \x01(\x0b\x32\x13.Block.TabContainerH\x00\x12\x19\n\x03tab\x18\x07 \x01(\x0b\x32\n.Block.TabH\x00\x12*\n\x0c\x63hat_message\x18\t \x01(\x0b\x32\x12.Block.ChatMessageH\x00\x12!\n\x07popover\x18\n \x01(\x0b\x32\x0e.Block.PopoverH\x00\x12\x13\n\x0b\x61llow_empty\x18\x08 \x01(\x08\x1a*\n\x08Vertical\x12\x0e\n\x06\x62order\x18\x01 \x01(\x08\x12\x0e\n\x06height\x18\x02 \x01(\r\x1a\x19\n\nHorizontal\x12\x0b\n\x03gap\x18\x01 \x01(\t\x1a%\n\x06\x43olumn\x12\x0e\n\x06weight\x18\x01 \x01(\x01\x12\x0b\n\x03gap\x18\x02 \x01(\t\x1aM\n\nExpandable\x12\r\n\x05label\x18\x01 \x01(\t\x12\x15\n\x08\x65xpanded\x18\x02 \x01(\x08H\x00\x88\x01\x01\x12\x0c\n\x04icon\x18\x03 \x01(\tB\x0b\n\t_expanded\x1a@\n\x04\x46orm\x12\x0f\n\x07\x66orm_id\x18\x01 \x01(\t\x12\x17\n\x0f\x63lear_on_submit\x18\x02 \x01(\x08\x12\x0e\n\x06\x62order\x18\x03 \x01(\x08\x1a\x0e\n\x0cTabContainer\x1a\x14\n\x03Tab\x12\r\n\x05label\x18\x01 \x01(\t\x1aU\n\x07Popover\x12\r\n\x05label\x18\x01 \x01(\t\x12\x1b\n\x13use_container_width\x18\x02 \x01(\x08\x12\x0c\n\x04help\x18\x03 \x01(\t\x12\x10\n\x08\x64isabled\x18\x04 \x01(\x08\x1a\x8d\x01\n\x0b\x43hatMessage\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0e\n\x06\x61vatar\x18\x02 \x01(\t\x12\x32\n\x0b\x61vatar_type\x18\x03 \x01(\x0e\x32\x1d.Block.ChatMessage.AvatarType\",\n\nAvatarType\x12\t\n\x05IMAGE\x10\x00\x12\t\n\x05\x45MOJI\x10\x01\x12\x08\n\x04ICON\x10\x02\x42\x06\n\x04typeB*\n\x1c\x63om.snowflake.apps.streamlitB\nBlockProtob\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1bstreamlit/proto/Block.proto\"\xbe\x08\n\x05\x42lock\x12#\n\x08vertical\x18\x01 \x01(\x0b\x32\x0f.Block.VerticalH\x00\x12\'\n\nhorizontal\x18\x02 \x01(\x0b\x32\x11.Block.HorizontalH\x00\x12\x1f\n\x06\x63olumn\x18\x03 \x01(\x0b\x32\r.Block.ColumnH\x00\x12\'\n\nexpandable\x18\x04 \x01(\x0b\x32\x11.Block.ExpandableH\x00\x12\x1b\n\x04\x66orm\x18\x05 \x01(\x0b\x32\x0b.Block.FormH\x00\x12,\n\rtab_container\x18\x06 \x01(\x0b\x32\x13.Block.TabContainerH\x00\x12\x19\n\x03tab\x18\x07 \x01(\x0b\x32\n.Block.TabH\x00\x12*\n\x0c\x63hat_message\x18\t \x01(\x0b\x32\x12.Block.ChatMessageH\x00\x12!\n\x07popover\x18\n \x01(\x0b\x32\x0e.Block.PopoverH\x00\x12\x1f\n\x06\x64ialog\x18\x0b \x01(\x0b\x32\r.Block.DialogH\x00\x12\x13\n\x0b\x61llow_empty\x18\x08 \x01(\x08\x1a*\n\x08Vertical\x12\x0e\n\x06\x62order\x18\x01 \x01(\x08\x12\x0e\n\x06height\x18\x02 \x01(\r\x1a\x19\n\nHorizontal\x12\x0b\n\x03gap\x18\x01 \x01(\t\x1a%\n\x06\x43olumn\x12\x0e\n\x06weight\x18\x01 \x01(\x01\x12\x0b\n\x03gap\x18\x02 \x01(\t\x1aM\n\nExpandable\x12\r\n\x05label\x18\x01 \x01(\t\x12\x15\n\x08\x65xpanded\x18\x02 \x01(\x08H\x00\x88\x01\x01\x12\x0c\n\x04icon\x18\x03 \x01(\tB\x0b\n\t_expanded\x1a\x9d\x01\n\x06\x44ialog\x12\r\n\x05title\x18\x01 \x01(\t\x12\x13\n\x0b\x64ismissible\x18\x02 \x01(\x08\x12(\n\x05width\x18\x03 \x01(\x0e\x32\x19.Block.Dialog.DialogWidth\x12\x14\n\x07is_open\x18\x04 \x01(\x08H\x00\x88\x01\x01\"#\n\x0b\x44ialogWidth\x12\t\n\x05SMALL\x10\x00\x12\t\n\x05LARGE\x10\x01\x42\n\n\x08_is_open\x1a@\n\x04\x46orm\x12\x0f\n\x07\x66orm_id\x18\x01 \x01(\t\x12\x17\n\x0f\x63lear_on_submit\x18\x02 \x01(\x08\x12\x0e\n\x06\x62order\x18\x03 \x01(\x08\x1a\x0e\n\x0cTabContainer\x1a\x14\n\x03Tab\x12\r\n\x05label\x18\x01 \x01(\t\x1aU\n\x07Popover\x12\r\n\x05label\x18\x01 \x01(\t\x12\x1b\n\x13use_container_width\x18\x02 \x01(\x08\x12\x0c\n\x04help\x18\x03 \x01(\t\x12\x10\n\x08\x64isabled\x18\x04 \x01(\x08\x1a\x8d\x01\n\x0b\x43hatMessage\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0e\n\x06\x61vatar\x18\x02 \x01(\t\x12\x32\n\x0b\x61vatar_type\x18\x03 \x01(\x0e\x32\x1d.Block.ChatMessage.AvatarType\",\n\nAvatarType\x12\t\n\x05IMAGE\x10\x00\x12\t\n\x05\x45MOJI\x10\x01\x12\x08\n\x04ICON\x10\x02\x42\x06\n\x04typeB*\n\x1c\x63om.snowflake.apps.streamlitB\nBlockProtob\x06proto3')
 
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'streamlit.proto.Block_pb2', globals())
@@ -22,25 +22,29 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n\034com.snowflake.apps.streamlitB\nBlockProto'
   _BLOCK._serialized_start=32
-  _BLOCK._serialized_end=925
-  _BLOCK_VERTICAL._serialized_start=395
-  _BLOCK_VERTICAL._serialized_end=437
-  _BLOCK_HORIZONTAL._serialized_start=439
-  _BLOCK_HORIZONTAL._serialized_end=464
-  _BLOCK_COLUMN._serialized_start=466
-  _BLOCK_COLUMN._serialized_end=503
-  _BLOCK_EXPANDABLE._serialized_start=505
-  _BLOCK_EXPANDABLE._serialized_end=582
-  _BLOCK_FORM._serialized_start=584
-  _BLOCK_FORM._serialized_end=648
-  _BLOCK_TABCONTAINER._serialized_start=650
-  _BLOCK_TABCONTAINER._serialized_end=664
-  _BLOCK_TAB._serialized_start=666
-  _BLOCK_TAB._serialized_end=686
-  _BLOCK_POPOVER._serialized_start=688
-  _BLOCK_POPOVER._serialized_end=773
-  _BLOCK_CHATMESSAGE._serialized_start=776
-  _BLOCK_CHATMESSAGE._serialized_end=917
-  _BLOCK_CHATMESSAGE_AVATARTYPE._serialized_start=873
-  _BLOCK_CHATMESSAGE_AVATARTYPE._serialized_end=917
+  _BLOCK._serialized_end=1118
+  _BLOCK_VERTICAL._serialized_start=428
+  _BLOCK_VERTICAL._serialized_end=470
+  _BLOCK_HORIZONTAL._serialized_start=472
+  _BLOCK_HORIZONTAL._serialized_end=497
+  _BLOCK_COLUMN._serialized_start=499
+  _BLOCK_COLUMN._serialized_end=536
+  _BLOCK_EXPANDABLE._serialized_start=538
+  _BLOCK_EXPANDABLE._serialized_end=615
+  _BLOCK_DIALOG._serialized_start=618
+  _BLOCK_DIALOG._serialized_end=775
+  _BLOCK_DIALOG_DIALOGWIDTH._serialized_start=728
+  _BLOCK_DIALOG_DIALOGWIDTH._serialized_end=763
+  _BLOCK_FORM._serialized_start=777
+  _BLOCK_FORM._serialized_end=841
+  _BLOCK_TABCONTAINER._serialized_start=843
+  _BLOCK_TABCONTAINER._serialized_end=857
+  _BLOCK_TAB._serialized_start=859
+  _BLOCK_TAB._serialized_end=879
+  _BLOCK_POPOVER._serialized_start=881
+  _BLOCK_POPOVER._serialized_end=966
+  _BLOCK_CHATMESSAGE._serialized_start=969
+  _BLOCK_CHATMESSAGE._serialized_end=1110
+  _BLOCK_CHATMESSAGE_AVATARTYPE._serialized_start=1066
+  _BLOCK_CHATMESSAGE_AVATARTYPE._serialized_end=1110
 # @@protoc_insertion_point(module_scope)

streamlit/proto/Block_pb2.pyi

@@ -96,6 +96,42 @@ class Block(google.protobuf.message.Message):
         def ClearField(self, field_name: typing_extensions.Literal["_expanded", b"_expanded", "expanded", b"expanded", "icon", b"icon", "label", b"label"]) -> None: ...
         def WhichOneof(self, oneof_group: typing_extensions.Literal["_expanded", b"_expanded"]) -> typing_extensions.Literal["expanded"] | None: ...
 
+    class Dialog(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        class _DialogWidth:
+            ValueType = typing.NewType("ValueType", builtins.int)
+            V: typing_extensions.TypeAlias = ValueType
+
+        class _DialogWidthEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Block.Dialog._DialogWidth.ValueType], builtins.type):  # noqa: F821
+            DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+            SMALL: Block.Dialog._DialogWidth.ValueType  # 0
+            LARGE: Block.Dialog._DialogWidth.ValueType  # 1
+
+        class DialogWidth(_DialogWidth, metaclass=_DialogWidthEnumTypeWrapper): ...
+        SMALL: Block.Dialog.DialogWidth.ValueType  # 0
+        LARGE: Block.Dialog.DialogWidth.ValueType  # 1
+
+        TITLE_FIELD_NUMBER: builtins.int
+        DISMISSIBLE_FIELD_NUMBER: builtins.int
+        WIDTH_FIELD_NUMBER: builtins.int
+        IS_OPEN_FIELD_NUMBER: builtins.int
+        title: builtins.str
+        dismissible: builtins.bool
+        width: global___Block.Dialog.DialogWidth.ValueType
+        is_open: builtins.bool
+        def __init__(
+            self,
+            *,
+            title: builtins.str = ...,
+            dismissible: builtins.bool = ...,
+            width: global___Block.Dialog.DialogWidth.ValueType = ...,
+            is_open: builtins.bool | None = ...,
+        ) -> None: ...
+        def HasField(self, field_name: typing_extensions.Literal["_is_open", b"_is_open", "is_open", b"is_open"]) -> builtins.bool: ...
+        def ClearField(self, field_name: typing_extensions.Literal["_is_open", b"_is_open", "dismissible", b"dismissible", "is_open", b"is_open", "title", b"title", "width", b"width"]) -> None: ...
+        def WhichOneof(self, oneof_group: typing_extensions.Literal["_is_open", b"_is_open"]) -> typing_extensions.Literal["is_open"] | None: ...
+
     class Form(google.protobuf.message.Message):
         DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -196,6 +232,7 @@ class Block(google.protobuf.message.Message):
     TAB_FIELD_NUMBER: builtins.int
     CHAT_MESSAGE_FIELD_NUMBER: builtins.int
     POPOVER_FIELD_NUMBER: builtins.int
+    DIALOG_FIELD_NUMBER: builtins.int
     ALLOW_EMPTY_FIELD_NUMBER: builtins.int
     @property
     def vertical(self) -> global___Block.Vertical: ...
@@ -215,6 +252,8 @@ class Block(google.protobuf.message.Message):
     def chat_message(self) -> global___Block.ChatMessage: ...
     @property
     def popover(self) -> global___Block.Popover: ...
+    @property
+    def dialog(self) -> global___Block.Dialog: ...
     allow_empty: builtins.bool
     def __init__(
         self,
@@ -228,10 +267,11 @@ class Block(google.protobuf.message.Message):
         tab: global___Block.Tab | None = ...,
         chat_message: global___Block.ChatMessage | None = ...,
         popover: global___Block.Popover | None = ...,
+        dialog: global___Block.Dialog | None = ...,
         allow_empty: builtins.bool = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["chat_message", b"chat_message", "column", b"column", "expandable", b"expandable", "form", b"form", "horizontal", b"horizontal", "popover", b"popover", "tab", b"tab", "tab_container", b"tab_container", "type", b"type", "vertical", b"vertical"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["allow_empty", b"allow_empty", "chat_message", b"chat_message", "column", b"column", "expandable", b"expandable", "form", b"form", "horizontal", b"horizontal", "popover", b"popover", "tab", b"tab", "tab_container", b"tab_container", "type", b"type", "vertical", b"vertical"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["vertical", "horizontal", "column", "expandable", "form", "tab_container", "tab", "chat_message", "popover"] | None: ...
+    def HasField(self, field_name: typing_extensions.Literal["chat_message", b"chat_message", "column", b"column", "dialog", b"dialog", "expandable", b"expandable", "form", b"form", "horizontal", b"horizontal", "popover", b"popover", "tab", b"tab", "tab_container", b"tab_container", "type", b"type", "vertical", b"vertical"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["allow_empty", b"allow_empty", "chat_message", b"chat_message", "column", b"column", "dialog", b"dialog", "expandable", b"expandable", "form", b"form", "horizontal", b"horizontal", "popover", b"popover", "tab", b"tab", "tab_container", b"tab_container", "type", b"type", "vertical", b"vertical"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["vertical", "horizontal", "column", "expandable", "form", "tab_container", "tab", "chat_message", "popover", "dialog"] | None: ...
 
 global___Block = Block