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

streamlit/elements/widgets/select_slider.py

@@ -12,20 +12,11 @@
 # 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,
-    Any,
-    Callable,
-    Generic,
-    List,
-    Optional,
-    Sequence,
-    Tuple,
-    Union,
-    cast,
-)
+from typing import TYPE_CHECKING, Any, Callable, Generic, Sequence, Tuple, cast
 
 from typing_extensions import TypeGuard
 
@@ -64,30 +55,30 @@ if TYPE_CHECKING:
     from streamlit.delta_generator import DeltaGenerator
 
 
-def _is_range_value(value: Union[T, Sequence[T]]) -> TypeGuard[Sequence[T]]:
+def _is_range_value(value: T | Sequence[T]) -> TypeGuard[Sequence[T]]:
     return isinstance(value, (list, tuple))
 
 
 @dataclass
 class SelectSliderSerde(Generic[T]):
     options: Sequence[T]
-    value: List[int]
+    value: list[int]
     is_range_value: bool
 
-    def serialize(self, v: object) -> List[int]:
+    def serialize(self, v: object) -> list[int]:
         return self._as_index_list(v)
 
     def deserialize(
         self,
-        ui_value: Optional[List[int]],
+        ui_value: list[int] | None,
         widget_id: str = "",
-    ) -> Union[T, Tuple[T, T]]:
+    ) -> T | tuple[T, T]:
         if not ui_value:
             # Widget has not been used; fallback to the original value,
             ui_value = self.value
 
         # The widget always returns floats, so convert to ints before indexing
-        return_value: Tuple[T, T] = cast(
+        return_value: tuple[T, T] = cast(
             Tuple[T, T],
             tuple(map(lambda x: self.options[int(x)], ui_value)),
         )
@@ -95,7 +86,7 @@ class SelectSliderSerde(Generic[T]):
         # If the original value was a list/tuple, so will be the output (and vice versa)
         return return_value if self.is_range_value else return_value[0]
 
-    def _as_index_list(self, v: object) -> List[int]:
+    def _as_index_list(self, v: object) -> list[int]:
         if _is_range_value(v):
             slider_value = [index_(self.options, val) for val in v]
             start, end = slider_value
@@ -114,15 +105,15 @@ class SelectSliderMixin:
         options: OptionSequence[T] = (),
         value: object = None,
         format_func: Callable[[Any], Any] = 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",
-    ) -> Union[T, Tuple[T, T]]:
+    ) -> T | tuple[T, T]:
         r"""
         Display a slider widget to select items from a list.
 
@@ -251,15 +242,15 @@ class SelectSliderMixin:
         options: OptionSequence[T] = (),
         value: object = None,
         format_func: Callable[[Any], Any] = 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,
         disabled: bool = False,
         label_visibility: LabelVisibility = "visible",
-        ctx: Optional[ScriptRunContext] = None,
-    ) -> Union[T, Tuple[T, T]]:
+        ctx: ScriptRunContext | None = None,
+    ) -> T | tuple[T, T]:
         key = to_key(key)
         check_callback_rules(self.dg, on_change)
         check_session_state_rules(default_value=value, key=key)
@@ -270,7 +261,7 @@ class SelectSliderMixin:
         if len(opt) == 0:
             raise StreamlitAPIException("The `options` argument needs to be non-empty")
 
-        def as_index_list(v: object) -> List[int]:
+        def as_index_list(v: object) -> list[int]:
             if _is_range_value(v):
                 slider_value = [index_(opt, val) for val in v]
                 start, end = slider_value
@@ -349,6 +340,6 @@ class SelectSliderMixin:
         return widget_state.value
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)

streamlit/elements/widgets/selectbox.py

@@ -301,6 +301,6 @@ class SelectboxMixin:
         return widget_state.value
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)

streamlit/elements/widgets/slider.py

@@ -12,23 +12,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 from dataclasses import dataclass
 from datetime import date, datetime, time, timedelta, timezone, tzinfo
 from numbers import Integral, Real
 from textwrap import dedent
-from typing import (
-    TYPE_CHECKING,
-    Any,
-    List,
-    Optional,
-    Sequence,
-    Tuple,
-    TypeVar,
-    Union,
-    cast,
-)
+from typing import TYPE_CHECKING, Any, Final, Sequence, Tuple, TypeVar, Union, cast
 
-from typing_extensions import Final, TypeAlias
+from typing_extensions import TypeAlias
 
 from streamlit.elements.form import current_form_id
 from streamlit.elements.utils import (
@@ -119,7 +111,7 @@ def _datetime_to_micros(dt: datetime) -> int:
     return _delta_to_micros(utc_dt - UTC_EPOCH)
 
 
-def _micros_to_datetime(micros: int, orig_tz: Optional[tzinfo]) -> datetime:
+def _micros_to_datetime(micros: int, orig_tz: tzinfo | None) -> datetime:
     """Restore times/datetimes to original timezone (dates are always naive)"""
     utc_dt = UTC_EPOCH + timedelta(microseconds=micros)
     # Add the original timezone. No conversion is required here,
@@ -129,12 +121,12 @@ def _micros_to_datetime(micros: int, orig_tz: Optional[tzinfo]) -> datetime:
 
 @dataclass
 class SliderSerde:
-    value: List[float]
+    value: list[float]
     data_type: int
     single_value: bool
-    orig_tz: Optional[tzinfo]
+    orig_tz: tzinfo | None
 
-    def deserialize(self, ui_value: Optional[List[float]], widget_id: str = ""):
+    def deserialize(self, ui_value: list[float] | None, widget_id: str = ""):
         if ui_value is not None:
             val: Any = ui_value
         else:
@@ -157,7 +149,7 @@ class SliderSerde:
             ]
         return val[0] if self.single_value else tuple(val)
 
-    def serialize(self, v: Any) -> List[Any]:
+    def serialize(self, v: Any) -> list[Any]:
         range_value = isinstance(v, (list, tuple))
         value = list(v) if range_value else [v]
         if self.data_type == SliderProto.DATE:
@@ -174,16 +166,16 @@ class SliderMixin:
     def slider(
         self,
         label: str,
-        min_value: Optional[SliderScalar] = None,
-        max_value: Optional[SliderScalar] = None,
-        value: Optional[SliderValue] = None,
-        step: Optional[Step] = None,
-        format: Optional[str] = None,
-        key: Optional[Key] = None,
-        help: Optional[str] = None,
-        on_change: Optional[WidgetCallback] = None,
-        args: Optional[WidgetArgs] = None,
-        kwargs: Optional[WidgetKwargs] = None,
+        min_value: SliderScalar | None = None,
+        max_value: SliderScalar | None = None,
+        value: SliderValue | None = None,
+        step: Step | None = None,
+        format: str | None = 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",
@@ -359,17 +351,17 @@ class SliderMixin:
         min_value=None,
         max_value=None,
         value=None,
-        step: Optional[Step] = None,
-        format: Optional[str] = None,
-        key: Optional[Key] = None,
-        help: Optional[str] = None,
-        on_change: Optional[WidgetCallback] = None,
-        args: Optional[WidgetArgs] = None,
-        kwargs: Optional[WidgetKwargs] = None,
+        step: Step | None = None,
+        format: str | None = 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,
+        ctx: ScriptRunContext | None = None,
     ) -> SliderReturn:
         key = to_key(key)
         check_callback_rules(self.dg, on_change)
@@ -673,6 +665,6 @@ class SliderMixin:
         return cast(SliderReturn, widget_state.value)
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)

streamlit/elements/widgets/text_widgets.py

@@ -16,11 +16,8 @@ from __future__ import annotations
 
 from dataclasses import dataclass
 from textwrap import dedent
-from typing import cast, overload
+from typing import TYPE_CHECKING, Literal, cast, overload
 
-from typing_extensions import Literal
-
-import streamlit
 from streamlit.elements.form import current_form_id
 from streamlit.elements.utils import (
     check_callback_rules,
@@ -47,6 +44,9 @@ from streamlit.type_util import (
     to_key,
 )
 
+if TYPE_CHECKING:
+    from streamlit.delta_generator import DeltaGenerator
+
 
 @dataclass
 class TextInputSerde:
@@ -581,6 +581,6 @@ class TextWidgetsMixin:
         return widget_state.value
 
     @property
-    def dg(self) -> "streamlit.delta_generator.DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
-        return cast("streamlit.delta_generator.DeltaGenerator", self)
+        return cast("DeltaGenerator", self)

streamlit/elements/widgets/time_widgets.py

@@ -20,7 +20,7 @@ from textwrap import dedent
 from typing import (
     TYPE_CHECKING,
     Any,
-    List,
+    Final,
     Literal,
     Sequence,
     Tuple,
@@ -61,8 +61,8 @@ DateWidgetReturn: TypeAlias = Union[
     date, Tuple[()], Tuple[date], Tuple[date, date], None
 ]
 
-DEFAULT_STEP_MINUTES = 15
-ALLOWED_DATE_FORMATS = re.compile(
+DEFAULT_STEP_MINUTES: Final = 15
+ALLOWED_DATE_FORMATS: Final = re.compile(
     r"^(YYYY[/.\-]MM[/.\-]DD|DD[/.\-]MM[/.\-]YYYY|MM[/.\-]DD[/.\-]YYYY)$"
 )
 
@@ -86,8 +86,8 @@ def _adjust_years(input_date: date, years: int) -> date:
 
 def _parse_date_value(
     value: DateValue | Literal["today"] | Literal["default_value_today"],
-) -> Tuple[List[date] | None, bool]:
-    parsed_dates: List[date]
+) -> tuple[list[date] | None, bool]:
+    parsed_dates: list[date]
     range_value: bool = False
     if value is None:
         return None, range_value
@@ -172,7 +172,7 @@ class _DateInputValues:
         value: DateValue | Literal["today"] | Literal["default_value_today"],
         min_value: SingleDateValue,
         max_value: SingleDateValue,
-    ) -> "_DateInputValues":
+    ) -> _DateInputValues:
         parsed_value, is_range = _parse_date_value(value=value)
         return cls(
             value=parsed_value,
@@ -249,7 +249,7 @@ class DateInputSerde:
             return return_value[0]
         return cast(DateWidgetReturn, tuple(return_value))
 
-    def serialize(self, v: DateWidgetReturn) -> List[str]:
+    def serialize(self, v: DateWidgetReturn) -> list[str]:
         if v is None:
             return []
 
@@ -271,7 +271,7 @@ class TimeWidgetsMixin:
         *,  # keyword-only arguments:
         disabled: bool = False,
         label_visibility: LabelVisibility = "visible",
-        step: Union[int, timedelta] = timedelta(minutes=DEFAULT_STEP_MINUTES),
+        step: int | timedelta = timedelta(minutes=DEFAULT_STEP_MINUTES),
     ) -> time:
         pass
 
@@ -288,7 +288,7 @@ class TimeWidgetsMixin:
         *,  # keyword-only arguments:
         disabled: bool = False,
         label_visibility: LabelVisibility = "visible",
-        step: Union[int, timedelta] = timedelta(minutes=DEFAULT_STEP_MINUTES),
+        step: int | timedelta = timedelta(minutes=DEFAULT_STEP_MINUTES),
     ) -> time | None:
         pass
 
@@ -305,7 +305,7 @@ class TimeWidgetsMixin:
         *,  # keyword-only arguments:
         disabled: bool = False,
         label_visibility: LabelVisibility = "visible",
-        step: Union[int, timedelta] = timedelta(minutes=DEFAULT_STEP_MINUTES),
+        step: int | timedelta = timedelta(minutes=DEFAULT_STEP_MINUTES),
     ) -> time | None:
         r"""Display a time input widget.
 
@@ -425,7 +425,7 @@ class TimeWidgetsMixin:
         *,  # keyword-only arguments:
         disabled: bool = False,
         label_visibility: LabelVisibility = "visible",
-        step: Union[int, timedelta] = timedelta(minutes=DEFAULT_STEP_MINUTES),
+        step: int | timedelta = timedelta(minutes=DEFAULT_STEP_MINUTES),
         ctx: ScriptRunContext | None = None,
     ) -> time | None:
         key = to_key(key)
@@ -703,7 +703,7 @@ class TimeWidgetsMixin:
         parsed_min_date = parse_date_deterministic(min_value)
         parsed_max_date = parse_date_deterministic(max_value)
 
-        parsed: str | None | List[str | None]
+        parsed: str | None | list[str | None]
         if value == "today" or value == "default_value_today" or value is None:
             parsed = None
         elif isinstance(value, (datetime, date)):
@@ -804,6 +804,6 @@ class TimeWidgetsMixin:
         return widget_state.value
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)

streamlit/elements/write.py

@@ -19,18 +19,7 @@ import inspect
 import json
 import types
 from io import StringIO
-from typing import (
-    TYPE_CHECKING,
-    Any,
-    Callable,
-    Final,
-    Generator,
-    Iterable,
-    List,
-    Tuple,
-    Type,
-    cast,
-)
+from typing import TYPE_CHECKING, Any, Callable, Final, Generator, Iterable, List, cast
 
 from streamlit import type_util
 from streamlit.errors import StreamlitAPIException
@@ -45,7 +34,7 @@ if TYPE_CHECKING:
 
 
 # Special methods:
-HELP_TYPES: Final[Tuple[Type[Any], ...]] = (
+HELP_TYPES: Final[tuple[type[Any], ...]] = (
     types.BuiltinFunctionType,
     types.BuiltinMethodType,
     types.FunctionType,
@@ -53,9 +42,9 @@ HELP_TYPES: Final[Tuple[Type[Any], ...]] = (
     types.ModuleType,
 )
 
-_LOGGER = get_logger(__name__)
+_LOGGER: Final = get_logger(__name__)
 
-_TEXT_CURSOR = "▕"
+_TEXT_CURSOR: Final = "▕"
 
 
 class StreamingOutput(List[Any]):
@@ -66,7 +55,7 @@ class WriteMixin:
     @gather_metrics("write_stream")
     def write_stream(
         self, stream: Callable[..., Any] | Generator[Any, Any, Any] | Iterable[Any]
-    ) -> List[Any] | str:
+    ) -> list[Any] | str:
         """Stream a generator, iterable, or stream-like sequence to the app.
 
         ``st.write_stream`` iterates through the given sequences and writes all
@@ -99,14 +88,14 @@ class WriteMixin:
         >>> import streamlit as st
         >>>
         >>> _LOREM_IPSUM = \"\"\"
-        >>> Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor
+        >>> Lorem ipsum dolor sit amet, **consectetur adipiscing** elit, sed do eiusmod tempor
         >>> incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis
         >>> nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat.
         >>> \"\"\"
         >>>
         >>>
         >>> def stream_data():
-        >>>     for word in _LOREM_IPSUM.split():
+        >>>     for word in _LOREM_IPSUM.split(" "):
         >>>         yield word + " "
         >>>         time.sleep(0.02)
         >>>
@@ -115,7 +104,7 @@ class WriteMixin:
         >>>         columns=["a", "b", "c", "d", "e", "f", "g", "h", "i", "j"],
         >>>     )
         >>>
-        >>>     for word in _LOREM_IPSUM.split():
+        >>>     for word in _LOREM_IPSUM.split(" "):
         >>>         yield word + " "
         >>>         time.sleep(0.02)
         >>>
@@ -133,14 +122,14 @@ class WriteMixin:
         # not be passed in here.
         if isinstance(stream, str) or type_util.is_dataframe_like(stream):
             raise StreamlitAPIException(
-                "`st.stream_write` expects a generator or stream-like object as input "
+                "`st.write_stream` expects a generator or stream-like object as input "
                 f"not {type(stream)}. Please use `st.write` instead for "
                 "this data type."
             )
 
         stream_container: DeltaGenerator | None = None
         streamed_response: str = ""
-        written_content: List[Any] = StreamingOutput()
+        written_content: list[Any] = StreamingOutput()
 
         def flush_stream_response():
             """Write the full response to the app."""
@@ -149,7 +138,7 @@ class WriteMixin:
 
             if streamed_response and stream_container:
                 # Replace the stream_container element the full response
-                stream_container.write(streamed_response)
+                stream_container.markdown(streamed_response)
                 written_content.append(streamed_response)
                 stream_container = None
                 streamed_response = ""
@@ -168,12 +157,15 @@ class WriteMixin:
         # Iterate through the generator and write each chunk to the app
         # with a type writer effect.
         for chunk in stream:  # type: ignore
-            if type_util.is_type(
-                chunk, "openai.types.chat.chat_completion_chunk.ChatCompletionChunk"
-            ):
+            if type_util.is_openai_chunk(chunk):
                 # Try to convert OpenAI chat completion chunk to a string:
                 try:
-                    chunk = chunk.choices[0].delta.content or ""
+                    if len(chunk.choices) == 0:
+                        # The choices list can be empty. E.g. when using the
+                        # AzureOpenAI client, the first chunk will always be empty.
+                        chunk = ""
+                    else:
+                        chunk = chunk.choices[0].delta.content or ""
                 except AttributeError as err:
                     raise StreamlitAPIException(
                         "Failed to parse the OpenAI ChatCompletionChunk. "
@@ -203,7 +195,7 @@ class WriteMixin:
                     first_text = True
                 streamed_response += chunk
                 # Only add the streaming symbol on the second text chunk
-                stream_container.write(
+                stream_container.markdown(
                     streamed_response + ("" if first_text else _TEXT_CURSOR),
                 )
             elif callable(chunk):
@@ -346,7 +338,7 @@ class WriteMixin:
                 kwargs,
             )
 
-        string_buffer: List[str] = []
+        string_buffer: list[str] = []
 
         # This bans some valid cases like: e = st.empty(); e.write("a", "b").
         # BUT: 1) such cases are rare, 2) this rule is easy to understand,
@@ -476,6 +468,6 @@ class WriteMixin:
         flush_buffer()
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)

streamlit/env_util.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import os
 import platform
 import re
@@ -23,7 +25,7 @@ IS_DARWIN = _system == "Darwin"
 IS_LINUX_OR_BSD = (_system == "Linux") or ("BSD" in _system)
 
 
-def is_pex():
+def is_pex() -> bool:
     """Return if streamlit running in pex.
 
     Pex modifies sys.path so the pex file is the first path and that's
@@ -34,7 +36,7 @@ def is_pex():
     return False
 
 
-def is_repl():
+def is_repl() -> bool:
     """Return True if running in the Python REPL."""
     import inspect
 
@@ -52,7 +54,7 @@ def is_repl():
     return False
 
 
-def is_executable_in_path(name):
+def is_executable_in_path(name: str) -> bool:
     """Check if executable is in OS path."""
     from shutil import which
 

streamlit/error_util.py

@@ -12,15 +12,19 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
+from typing import Final
+
 import streamlit as st
 from streamlit import config
 from streamlit.errors import UncaughtAppException
 from streamlit.logger import get_logger
 
-_LOGGER = get_logger(__name__)
+_LOGGER: Final = get_logger(__name__)
 
 
-def _print_rich_exception(e: BaseException):
+def _print_rich_exception(e: BaseException) -> None:
     from rich import box, panel
 
     # Monkey patch the panel to use our custom box style
@@ -31,7 +35,7 @@ def _print_rich_exception(e: BaseException):
             box=box.Box("────\n    \n────\n    \n────\n────\n    \n────\n"),
             **kwargs,
         ):
-            super(ConfigurablePanel, self).__init__(renderable, box, **kwargs)
+            super().__init__(renderable, box, **kwargs)
 
     from rich import traceback as rich_traceback
 

streamlit/errors.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 from streamlit import util
 
 
@@ -89,7 +91,7 @@ class StreamlitAPIWarning(StreamlitAPIException, Warning):
     """
 
     def __init__(self, *args):
-        super(StreamlitAPIWarning, self).__init__(*args)
+        super().__init__(*args)
         import inspect
         import traceback