streamlit-nightly 1.31.2.dev20240212__py2.py3-none-any.whl → 1.31.2.dev20240214__py2.py3-none-any.whl

streamlit/elements/pyplot.py

@@ -14,15 +14,14 @@
 
 """Streamlit support for Matplotlib PyPlot charts."""
 
-import io
-from typing import TYPE_CHECKING, Any, Optional, cast
+from __future__ import annotations
 
-from typing_extensions import Final
+import io
+from typing import TYPE_CHECKING, Any, cast
 
 import streamlit.elements.image as image_utils
 from streamlit import config
 from streamlit.errors import StreamlitDeprecationWarning
-from streamlit.logger import get_logger
 from streamlit.proto.Image_pb2 import ImageList as ImageListProto
 from streamlit.runtime.metrics_util import gather_metrics
 
@@ -31,18 +30,16 @@ if TYPE_CHECKING:
 
     from streamlit.delta_generator import DeltaGenerator
 
-LOGGER: Final = get_logger(__name__)
-
 
 class PyplotMixin:
     @gather_metrics("pyplot")
     def pyplot(
         self,
-        fig: Optional["Figure"] = None,
-        clear_figure: Optional[bool] = None,
+        fig: Figure | None = None,
+        clear_figure: bool | None = None,
         use_container_width: bool = True,
         **kwargs: Any,
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         """Display a matplotlib.pyplot figure.
 
         Parameters
@@ -117,7 +114,7 @@ class PyplotMixin:
         return self.dg._enqueue("imgs", image_list_proto)
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)
 
@@ -125,8 +122,8 @@ class PyplotMixin:
 def marshall(
     coordinates: str,
     image_list_proto: ImageListProto,
-    fig: Optional["Figure"] = None,
-    clear_figure: Optional[bool] = True,
+    fig: Figure | None = None,
+    clear_figure: bool | None = True,
     use_container_width: bool = True,
     **kwargs: Any,
 ) -> None:
@@ -183,7 +180,7 @@ def marshall(
 
 class PyplotGlobalUseWarning(StreamlitDeprecationWarning):
     def __init__(self) -> None:
-        super(PyplotGlobalUseWarning, self).__init__(
+        super().__init__(
             msg=self._get_message(), config_option="deprecation.showPyplotGlobalUse"
         )
 

streamlit/elements/snow.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 from typing import TYPE_CHECKING, cast
 
 from streamlit.proto.Snow_pb2 import Snow as SnowProto
@@ -23,7 +25,7 @@ if TYPE_CHECKING:
 
 class SnowMixin:
     @gather_metrics("snow")
-    def snow(self) -> "DeltaGenerator":
+    def snow(self) -> DeltaGenerator:
         """Draw celebratory snowfall.
 
         Example
@@ -40,6 +42,6 @@ class SnowMixin:
         return self.dg._enqueue("snow", snow_proto)
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)

streamlit/elements/spinner.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import contextlib
 import threading
 from typing import Iterator

streamlit/elements/text.py

@@ -12,7 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import TYPE_CHECKING, Optional, cast
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, cast
 
 from streamlit.proto.Text_pb2 import Text as TextProto
 from streamlit.runtime.metrics_util import gather_metrics
@@ -27,10 +29,10 @@ class TextMixin:
     @gather_metrics("text")
     def text(
         self,
-        body: "SupportsStr",
+        body: SupportsStr,
         *,  # keyword-only arguments:
-        help: Optional[str] = None,
-    ) -> "DeltaGenerator":
+        help: str | None = None,
+    ) -> DeltaGenerator:
         """Write fixed-width and preformatted text.
 
         Parameters
@@ -55,6 +57,6 @@ class TextMixin:
         return self.dg._enqueue("text", text_proto)
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)

streamlit/elements/toast.py

@@ -12,7 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import TYPE_CHECKING, Optional, cast
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, cast
 
 from streamlit.errors import StreamlitAPIException
 from streamlit.proto.Toast_pb2 import Toast as ToastProto
@@ -39,8 +41,8 @@ class ToastMixin:
         self,
         body: SupportsStr,
         *,  # keyword-only args:
-        icon: Optional[str] = None,
-    ) -> "DeltaGenerator":
+        icon: str | None = None,
+    ) -> DeltaGenerator:
         """Display a short message, known as a notification "toast".
         The toast appears in the app's bottom-right corner and disappears after four seconds.
 
@@ -86,6 +88,6 @@ class ToastMixin:
         return self.dg._enqueue("toast", toast_proto)
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)

streamlit/elements/utils.py

@@ -12,21 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 from enum import Enum, EnumMeta
-from typing import (
-    TYPE_CHECKING,
-    Any,
-    Hashable,
-    Iterable,
-    List,
-    Optional,
-    Sequence,
-    Tuple,
-    Type,
-    Union,
-    cast,
-    overload,
-)
+from typing import TYPE_CHECKING, Any, Hashable, Iterable, Sequence, cast, overload
 
 import streamlit
 from streamlit import config, runtime, type_util
@@ -43,8 +32,8 @@ if TYPE_CHECKING:
 
 
 def last_index_for_melted_dataframes(
-    data: Union["DataFrameCompatible", Any]
-) -> Optional[Hashable]:
+    data: DataFrameCompatible | Any,
+) -> Hashable | None:
     if type_util.is_dataframe_compatible(data):
         data = type_util.convert_anything_to_df(data)
 
@@ -54,9 +43,7 @@ def last_index_for_melted_dataframes(
     return None
 
 
-def check_callback_rules(
-    dg: "DeltaGenerator", on_change: Optional[WidgetCallback]
-) -> None:
+def check_callback_rules(dg: DeltaGenerator, on_change: WidgetCallback | None) -> None:
     if runtime.exists() and is_in_form(dg) and on_change is not None:
         raise StreamlitAPIException(
             "With forms, callbacks can only be defined on the `st.form_submit_button`."
@@ -73,7 +60,7 @@ st.chat_input, and st.form cannot be set using st.session_state.
 
 
 def check_session_state_rules(
-    default_value: Any, key: Optional[str], writes_allowed: bool = True
+    default_value: Any, key: str | None, writes_allowed: bool = True
 ) -> None:
     global _shown_default_value_warning
 
@@ -101,7 +88,7 @@ def check_session_state_rules(
 
 def get_label_visibility_proto_value(
     label_visibility_string: type_util.LabelVisibility,
-) -> "LabelVisibilityMessage.LabelVisibilityOptions.ValueType":
+) -> LabelVisibilityMessage.LabelVisibilityOptions.ValueType:
     """Returns one of LabelVisibilityMessage enum constants.py based on string value."""
 
     if label_visibility_string == "visible":
@@ -117,7 +104,7 @@ def get_label_visibility_proto_value(
 @overload
 def maybe_coerce_enum(
     register_widget_result: RegisterWidgetResult[Enum],
-    options: Type[Enum],
+    options: type[Enum],
     opt_sequence: Sequence[Any],
 ) -> RegisterWidgetResult[Enum]:
     ...
@@ -141,7 +128,7 @@ def maybe_coerce_enum(register_widget_result, options, opt_sequence):
     if not isinstance(register_widget_result.value, Enum):
         return register_widget_result
 
-    coerce_class: Optional[EnumMeta]
+    coerce_class: EnumMeta | None
     if isinstance(options, EnumMeta):
         coerce_class = options
     else:
@@ -159,19 +146,19 @@ def maybe_coerce_enum(register_widget_result, options, opt_sequence):
 # (https://github.com/python/typing/issues/548)
 @overload
 def maybe_coerce_enum_sequence(
-    register_widget_result: RegisterWidgetResult[List[T]],
+    register_widget_result: RegisterWidgetResult[list[T]],
     options: type_util.OptionSequence[T],
     opt_sequence: Sequence[T],
-) -> RegisterWidgetResult[List[T]]:
+) -> RegisterWidgetResult[list[T]]:
     ...
 
 
 @overload
 def maybe_coerce_enum_sequence(
-    register_widget_result: RegisterWidgetResult[Tuple[T, T]],
+    register_widget_result: RegisterWidgetResult[tuple[T, T]],
     options: type_util.OptionSequence[T],
     opt_sequence: Sequence[T],
-) -> RegisterWidgetResult[Tuple[T, T]]:
+) -> RegisterWidgetResult[tuple[T, T]]:
     ...
 
 
@@ -185,7 +172,7 @@ def maybe_coerce_enum_sequence(register_widget_result, options, opt_sequence):
         return register_widget_result
 
     # Extract the class to coerce
-    coerce_class: Optional[EnumMeta]
+    coerce_class: EnumMeta | None
     if isinstance(options, EnumMeta):
         coerce_class = options
     else:

streamlit/elements/widgets/button.py

@@ -18,9 +18,9 @@ import io
 import os
 from dataclasses import dataclass
 from textwrap import dedent
-from typing import TYPE_CHECKING, BinaryIO, Optional, TextIO, Union, cast
+from typing import TYPE_CHECKING, BinaryIO, Final, Literal, TextIO, Union, cast
 
-from typing_extensions import Final, Literal
+from typing_extensions import TypeAlias
 
 from streamlit import runtime, source_util
 from streamlit.elements.form import current_form_id, is_in_form
@@ -52,7 +52,7 @@ For more information, refer to the
 [documentation for forms](https://docs.streamlit.io/library/api-reference/control-flow/st.form).
 """
 
-DownloadButtonDataType = Union[str, bytes, TextIO, BinaryIO, io.RawIOBase]
+DownloadButtonDataType: TypeAlias = Union[str, bytes, TextIO, BinaryIO, io.RawIOBase]
 
 
 @dataclass
@@ -60,7 +60,7 @@ class ButtonSerde:
     def serialize(self, v: bool) -> bool:
         return bool(v)
 
-    def deserialize(self, ui_value: Optional[bool], widget_id: str = "") -> bool:
+    def deserialize(self, ui_value: bool | None, widget_id: str = "") -> bool:
         return ui_value or False
 
 
@@ -69,11 +69,11 @@ class ButtonMixin:
     def button(
         self,
         label: str,
-        key: Optional[Key] = None,
-        help: Optional[str] = None,
-        on_click: Optional[WidgetCallback] = None,
-        args: Optional[WidgetArgs] = None,
-        kwargs: Optional[WidgetKwargs] = None,
+        key: Key | None = None,
+        help: str | None = None,
+        on_click: WidgetCallback | None = None,
+        args: WidgetArgs | None = None,
+        kwargs: WidgetKwargs | None = None,
         *,  # keyword-only arguments:
         type: Literal["primary", "secondary"] = "secondary",
         disabled: bool = False,
@@ -179,13 +179,13 @@ class ButtonMixin:
         self,
         label: str,
         data: DownloadButtonDataType,
-        file_name: Optional[str] = None,
-        mime: Optional[str] = None,
-        key: Optional[Key] = None,
-        help: Optional[str] = None,
-        on_click: Optional[WidgetCallback] = None,
-        args: Optional[WidgetArgs] = None,
-        kwargs: Optional[WidgetKwargs] = None,
+        file_name: str | None = None,
+        mime: str | None = None,
+        key: Key | None = None,
+        help: str | None = None,
+        on_click: WidgetCallback | None = None,
+        args: WidgetArgs | None = None,
+        kwargs: WidgetKwargs | None = None,
         *,  # keyword-only arguments:
         type: Literal["primary", "secondary"] = "secondary",
         disabled: bool = False,
@@ -350,11 +350,11 @@ class ButtonMixin:
         label: str,
         url: str,
         *,
-        help: Optional[str] = None,
+        help: str | None = None,
         type: Literal["primary", "secondary"] = "secondary",
         disabled: bool = False,
         use_container_width: bool = False,
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         r"""Display a link button element.
 
         When clicked, a new tab will be opened to the specified URL. This will
@@ -437,7 +437,7 @@ class ButtonMixin:
         help: str | None = None,
         disabled: bool = False,
         use_container_width: bool | None = None,
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         """Display a link to another page in a multipage app or to an external page.
 
         If another page in a multipage app is specified, clicking ``st.page_link``
@@ -537,18 +537,18 @@ class ButtonMixin:
         self,
         label: str,
         data: DownloadButtonDataType,
-        file_name: Optional[str] = None,
-        mime: Optional[str] = None,
-        key: Optional[Key] = None,
-        help: Optional[str] = None,
-        on_click: Optional[WidgetCallback] = None,
-        args: Optional[WidgetArgs] = None,
-        kwargs: Optional[WidgetKwargs] = None,
+        file_name: str | None = None,
+        mime: str | None = None,
+        key: Key | None = None,
+        help: str | None = None,
+        on_click: WidgetCallback | None = None,
+        args: WidgetArgs | None = None,
+        kwargs: WidgetKwargs | None = None,
         *,  # keyword-only arguments:
         type: Literal["primary", "secondary"] = "secondary",
         disabled: bool = False,
         use_container_width: bool = False,
-        ctx: Optional[ScriptRunContext] = None,
+        ctx: ScriptRunContext | None = None,
     ) -> bool:
         key = to_key(key)
         check_session_state_rules(default_value=None, key=key, writes_allowed=False)
@@ -606,12 +606,12 @@ class ButtonMixin:
         self,
         label: str,
         url: str,
-        help: Optional[str],
+        help: str | None,
         *,  # keyword-only arguments:
         type: Literal["primary", "secondary"] = "secondary",
         disabled: bool = False,
         use_container_width: bool = False,
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         link_button_proto = LinkButtonProto()
         link_button_proto.label = label
         link_button_proto.url = url
@@ -633,7 +633,7 @@ class ButtonMixin:
         help: str | None = None,
         disabled: bool = False,
         use_container_width: bool | None = None,
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         page_link_proto = PageLinkProto()
         page_link_proto.disabled = disabled
 
@@ -692,17 +692,17 @@ class ButtonMixin:
     def _button(
         self,
         label: str,
-        key: Optional[str],
-        help: Optional[str],
+        key: str | None,
+        help: str | None,
         is_form_submitter: bool,
-        on_click: Optional[WidgetCallback] = None,
-        args: Optional[WidgetArgs] = None,
-        kwargs: Optional[WidgetKwargs] = None,
+        on_click: WidgetCallback | None = None,
+        args: WidgetArgs | None = None,
+        kwargs: WidgetKwargs | None = None,
         *,  # keyword-only arguments:
         type: Literal["primary", "secondary"] = "secondary",
         disabled: bool = False,
         use_container_width: bool = False,
-        ctx: Optional[ScriptRunContext] = None,
+        ctx: ScriptRunContext | None = None,
     ) -> bool:
         if not is_form_submitter:
             check_callback_rules(self.dg, on_click)
@@ -767,7 +767,7 @@ class ButtonMixin:
         return button_state.value
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)
 
@@ -776,8 +776,8 @@ def marshall_file(
     coordinates: str,
     data: DownloadButtonDataType,
     proto_download_button: DownloadButtonProto,
-    mimetype: Optional[str],
-    file_name: Optional[str] = None,
+    mimetype: str | None,
+    file_name: str | None = None,
 ) -> None:
     data_as_bytes: bytes
     if isinstance(data, str):

streamlit/elements/widgets/camera_input.py

@@ -12,9 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 from dataclasses import dataclass
 from textwrap import dedent
-from typing import TYPE_CHECKING, Optional, Union, cast
+from typing import TYPE_CHECKING, Union, cast
+
+from typing_extensions import TypeAlias
 
 from streamlit.elements.form import current_form_id
 from streamlit.elements.utils import (
@@ -41,7 +45,7 @@ from streamlit.type_util import Key, LabelVisibility, maybe_raise_label_warnings
 if TYPE_CHECKING:
     from streamlit.delta_generator import DeltaGenerator
 
-SomeUploadedSnapshotFile = Union[UploadedFile, DeletedFile, None]
+SomeUploadedSnapshotFile: TypeAlias = Union[UploadedFile, DeletedFile, None]
 
 
 @dataclass
@@ -64,7 +68,7 @@ class CameraInputSerde:
         return state_proto
 
     def deserialize(
-        self, ui_value: Optional[FileUploaderStateProto], widget_id: str
+        self, ui_value: FileUploaderStateProto | None, widget_id: str
     ) -> SomeUploadedSnapshotFile:
         upload_files = _get_upload_files(ui_value)
         if len(upload_files) == 0:
@@ -79,15 +83,15 @@ class CameraInputMixin:
     def camera_input(
         self,
         label: str,
-        key: Optional[Key] = None,
-        help: Optional[str] = None,
-        on_change: Optional[WidgetCallback] = None,
-        args: Optional[WidgetArgs] = None,
-        kwargs: Optional[WidgetKwargs] = None,
+        key: Key | None = None,
+        help: str | None = None,
+        on_change: WidgetCallback | None = None,
+        args: WidgetArgs | None = None,
+        kwargs: WidgetKwargs | None = None,
         *,  # keyword-only arguments:
         disabled: bool = False,
         label_visibility: LabelVisibility = "visible",
-    ) -> Optional[UploadedFile]:
+    ) -> UploadedFile | None:
         r"""Display a widget that returns pictures from the user's webcam.
 
         Parameters
@@ -180,16 +184,16 @@ class CameraInputMixin:
     def _camera_input(
         self,
         label: str,
-        key: Optional[Key] = None,
-        help: Optional[str] = None,
-        on_change: Optional[WidgetCallback] = None,
-        args: Optional[WidgetArgs] = None,
-        kwargs: Optional[WidgetKwargs] = None,
+        key: Key | None = None,
+        help: str | None = None,
+        on_change: WidgetCallback | None = None,
+        args: WidgetArgs | None = None,
+        kwargs: WidgetKwargs | None = None,
         *,  # keyword-only arguments:
         disabled: bool = False,
         label_visibility: LabelVisibility = "visible",
-        ctx: Optional[ScriptRunContext] = None,
-    ) -> Optional[UploadedFile]:
+        ctx: ScriptRunContext | None = None,
+    ) -> UploadedFile | None:
         key = to_key(key)
         check_callback_rules(self.dg, on_change)
         check_session_state_rules(default_value=None, key=key, writes_allowed=False)
@@ -238,6 +242,6 @@ class CameraInputMixin:
         return camera_input_state.value
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)

streamlit/elements/widgets/chat.py

@@ -11,13 +11,12 @@
 # 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 dataclasses import dataclass
 from enum import Enum
-from typing import TYPE_CHECKING, Optional, Tuple, cast
-
-from typing_extensions import Literal
+from typing import TYPE_CHECKING, Literal, cast
 
 from streamlit import runtime
 from streamlit.elements.form import is_in_form
@@ -53,7 +52,7 @@ class PresetNames(str, Enum):
 
 def _process_avatar_input(
     avatar: str | AtomicImage | None, delta_path: str
-) -> Tuple[BlockProto.ChatMessage.AvatarType.ValueType, str]:
+) -> tuple[BlockProto.ChatMessage.AvatarType.ValueType, str]:
     """Detects the avatar type and prepares the avatar data for the frontend.
 
     Parameters
@@ -102,7 +101,7 @@ def _process_avatar_input(
 @dataclass
 class ChatInputSerde:
     def deserialize(
-        self, ui_value: Optional[StringTriggerValueProto], widget_id: str = ""
+        self, ui_value: StringTriggerValueProto | None, widget_id: str = ""
     ) -> str | None:
         if ui_value is None or not ui_value.HasField("data"):
             return None
@@ -120,7 +119,7 @@ class ChatMixin:
         name: Literal["user", "assistant", "ai", "human"] | str,
         *,
         avatar: Literal["user", "assistant"] | str | AtomicImage | None = None,
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         """Insert a chat message container.
 
         To add elements to the returned container, you can use ``with`` notation
@@ -366,6 +365,6 @@ class ChatMixin:
         return widget_state.value if not widget_state.value_changed else None
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)