streamlit-nightly 1.45.2.dev20250513__py3-none-any.whl → 1.45.2.dev20250515__py3-none-any.whl

streamlit/commands/execution_control.py

@@ -64,38 +64,38 @@ def _new_fragment_id_queue(
     if scope == "app":
         return []
 
-    else:  # scope == "fragment"
-        curr_queue = ctx.fragment_ids_this_run
-
-        # If st.rerun(scope="fragment") is called during a full script run, we raise an
-        # exception. This occurs, of course, if st.rerun(scope="fragment") is called
-        # outside of a fragment, but it somewhat surprisingly occurs if it gets called
-        # from within a fragment during a run of the full script. While this behavior may
-        # be surprising, it seems somewhat reasonable given that the correct behavior of
-        # calling st.rerun(scope="fragment") in this situation is unclear to me:
-        #   * Rerunning just the fragment immediately may cause weirdness down the line
-        #     as any part of the script that occurs after the fragment will not be
-        #     executed.
-        #   * Waiting until the full script run completes before rerunning the fragment
-        #     seems odd (even if we normally do this before running a fragment not
-        #     triggered by st.rerun()) because it defers the execution of st.rerun().
-        #   * Rerunning the full app feels incorrect as we're seemingly ignoring the
-        #     `scope` argument.
-        # With these issues and given that it seems pretty unnatural to have a
-        # fragment-scoped rerun happen during a full script run to begin with, it seems
-        # reasonable to just disallow this completely for now.
-        if not curr_queue:
-            raise StreamlitAPIException(
-                'scope="fragment" can only be specified from `@st.fragment`-decorated '
-                "functions during fragment reruns."
-            )
-
-        new_queue = list(dropwhile(lambda x: x != ctx.current_fragment_id, curr_queue))
-        assert new_queue, (
-            "Could not find current_fragment_id in fragment_id_queue. This should never happen."
+    # > scope == "fragment"
+    curr_queue = ctx.fragment_ids_this_run
+
+    # If st.rerun(scope="fragment") is called during a full script run, we raise an
+    # exception. This occurs, of course, if st.rerun(scope="fragment") is called
+    # outside of a fragment, but it somewhat surprisingly occurs if it gets called
+    # from within a fragment during a run of the full script. While this behavior may
+    # be surprising, it seems somewhat reasonable given that the correct behavior of
+    # calling st.rerun(scope="fragment") in this situation is unclear to me:
+    #   * Rerunning just the fragment immediately may cause weirdness down the line
+    #     as any part of the script that occurs after the fragment will not be
+    #     executed.
+    #   * Waiting until the full script run completes before rerunning the fragment
+    #     seems odd (even if we normally do this before running a fragment not
+    #     triggered by st.rerun()) because it defers the execution of st.rerun().
+    #   * Rerunning the full app feels incorrect as we're seemingly ignoring the
+    #     `scope` argument.
+    # With these issues and given that it seems pretty unnatural to have a
+    # fragment-scoped rerun happen during a full script run to begin with, it seems
+    # reasonable to just disallow this completely for now.
+    if not curr_queue:
+        raise StreamlitAPIException(
+            'scope="fragment" can only be specified from `@st.fragment`-decorated '
+            "functions during fragment reruns."
         )
 
-        return new_queue
+    new_queue = list(dropwhile(lambda x: x != ctx.current_fragment_id, curr_queue))
+    assert new_queue, (
+        "Could not find current_fragment_id in fragment_id_queue. This should never happen."
+    )
+
+    return new_queue
 
 
 @gather_metrics("rerun")

streamlit/connections/sql_connection.py

@@ -319,13 +319,16 @@ class SQLConnection(BaseConnection["Engine"]):
             import pandas as pd
 
             instance = self._instance.connect()
-            return pd.read_sql(
-                text(sql),
-                instance,
-                index_col=index_col,
-                chunksize=chunksize,
-                params=params,
-                **kwargs,
+            return cast(
+                "DataFrame",
+                pd.read_sql(
+                    text(sql),
+                    instance,
+                    index_col=index_col,
+                    chunksize=chunksize,
+                    params=params,
+                    **kwargs,
+                ),
             )
 
         # We modify our helper function's `__qualname__` here to work around default

streamlit/connections/util.py

@@ -21,7 +21,7 @@
 from __future__ import annotations
 
 import os
-from typing import TYPE_CHECKING, Any, cast
+from typing import TYPE_CHECKING, Any
 
 if TYPE_CHECKING:
     from collections.abc import Collection
@@ -92,6 +92,6 @@ def running_in_sis() -> bool:
             is_in_stored_procedure,
         )
 
-        return cast("bool", is_in_stored_procedure())
+        return is_in_stored_procedure()  # type: ignore
     except ModuleNotFoundError:
         return False

streamlit/elements/heading.py

@@ -262,10 +262,9 @@ class HeadingMixin:
         ]
         if divider in valid_colors:
             return cast("str", divider)
-        else:
-            raise StreamlitAPIException(
-                f"Divider parameter has invalid value: `{divider}`. Please choose from: {', '.join(valid_colors)}."
-            )
+        raise StreamlitAPIException(
+            f"Divider parameter has invalid value: `{divider}`. Please choose from: {', '.join(valid_colors)}."
+        )
 
     @staticmethod
     def _create_heading_proto(

streamlit/elements/html.py

@@ -113,10 +113,9 @@ class HtmlMixin:
             # If true, there are only style tags - send html to the event container
             html_proto.body = html_content
             return self._event_dg._enqueue("html", html_proto)
-        else:
-            # Otherwise, send the html to the main container as normal
-            html_proto.body = html_content
-            return self.dg._enqueue("html", html_proto)
+        # Otherwise, send the html to the main container as normal
+        html_proto.body = html_content
+        return self.dg._enqueue("html", html_proto)
 
     @property
     def dg(self) -> DeltaGenerator:

streamlit/elements/lib/built_in_chart_utils.py

@@ -559,7 +559,7 @@ def _melt_data(
 
     # Arrow has problems with object types after melting two different dtypes
     # > pyarrow.lib.ArrowTypeError: "Expected a <TYPE> object, got a object"
-    fixed_df = dataframe_util.fix_arrow_incompatible_column_types(
+    return dataframe_util.fix_arrow_incompatible_column_types(
         melted_df,
         selected_columns=[
             *columns_to_leave_alone,
@@ -568,8 +568,6 @@ def _melt_data(
         ],
     )
 
-    return fixed_df
-
 
 def _maybe_reset_index_in_place(
     df: pd.DataFrame, x_column: str | None, y_column_list: list[str]

streamlit/elements/lib/image_utils.py

@@ -335,9 +335,8 @@ def image_to_url(
         url = runtime.get_instance().media_file_mgr.add(image_data, mimetype, image_id)
         caching.save_media_data(image_data, mimetype, image_id)
         return url
-    else:
-        # When running in "raw mode", we can't access the MediaFileManager.
-        return ""
+    # When running in "raw mode", we can't access the MediaFileManager.
+    return ""
 
 
 def _4d_to_list_3d(array: npt.NDArray[Any]) -> list[npt.NDArray[Any]]:

streamlit/elements/lib/options_selector_utils.py

@@ -92,8 +92,7 @@ def get_default_indices(
     indexable_options: Sequence[T], default: Sequence[Any] | Any | None = None
 ) -> list[int]:
     default_indices = check_and_convert_to_indices(indexable_options, default)
-    default_indices = default_indices if default_indices is not None else []
-    return default_indices
+    return default_indices if default_indices is not None else []
 
 
 E1 = TypeVar("E1", bound=Enum)
@@ -186,7 +185,11 @@ def maybe_coerce_enum(
 ) -> RegisterWidgetResult[T]: ...
 
 
-def maybe_coerce_enum(register_widget_result, options, opt_sequence):
+def maybe_coerce_enum(
+    register_widget_result: RegisterWidgetResult[Any],
+    options: OptionSequence[Any],
+    opt_sequence: Sequence[Any],
+) -> RegisterWidgetResult[Any]:
     """Maybe Coerce a RegisterWidgetResult with an Enum member value to
     RegisterWidgetResult[option] if option is an EnumType, otherwise just return
     the original RegisterWidgetResult.
@@ -228,7 +231,11 @@ def maybe_coerce_enum_sequence(
 ) -> RegisterWidgetResult[tuple[T, T]]: ...
 
 
-def maybe_coerce_enum_sequence(register_widget_result, options, opt_sequence):
+def maybe_coerce_enum_sequence(
+    register_widget_result: RegisterWidgetResult[list[Any] | tuple[Any, ...]],
+    options: OptionSequence[Any],
+    opt_sequence: Sequence[Any],
+) -> RegisterWidgetResult[list[Any] | tuple[Any, ...]]:
     """Maybe Coerce a RegisterWidgetResult with a sequence of Enum members as value
     to RegisterWidgetResult[Sequence[option]] if option is an EnumType, otherwise just
     return the original RegisterWidgetResult.

streamlit/elements/lib/pandas_styler_utils.py

@@ -213,9 +213,7 @@ def _pandas_style_to_css(
     selector = ", ".join(selectors)
 
     declaration_block = "; ".join(declarations)
-    rule_set = selector + " { " + declaration_block + " }"
-
-    return rule_set
+    return selector + " { " + declaration_block + " }"
 
 
 def _marshall_display_values(

streamlit/elements/lib/subtitle_utils.py

@@ -104,9 +104,7 @@ def _srt_to_vtt(srt_data: str | bytes) -> bytes:
     # Add WebVTT file header
     vtt_content = "WEBVTT\n\n" + vtt_data
     # Convert the vtt content to bytes
-    vtt_content = vtt_content.strip().encode("utf-8")
-
-    return vtt_content
+    return vtt_content.strip().encode("utf-8")
 
 
 def _handle_string_or_path_data(data_or_path: str | Path) -> bytes:
@@ -123,14 +121,14 @@ def _handle_string_or_path_data(data_or_path: str | Path) -> bytes:
         with open(data_or_path, "rb") as file:
             content = file.read()
         return _srt_to_vtt(content) if file_extension == ".srt" else content
-    elif isinstance(data_or_path, Path):
-        raise ValueError(f"File {data_or_path} does not exist.")
+    if isinstance(data_or_path, Path):
+        raise ValueError(f"File {data_or_path} does not exist.")  # noqa: TRY004
 
     content_string = data_or_path.strip()
 
     if content_string.startswith("WEBVTT") or content_string == "":
         return content_string.encode("utf-8")
-    elif _is_srt(content_string):
+    if _is_srt(content_string):
         return _srt_to_vtt(content_string)
     raise ValueError("The provided string neither matches valid VTT nor SRT format.")
 
@@ -173,6 +171,5 @@ def process_subtitle_data(
         )
         caching.save_media_data(subtitle_data, "text/vtt", coordinates)
         return file_url
-    else:
-        # When running in "raw mode", we can't access the MediaFileManager.
-        return ""
+    # When running in "raw mode", we can't access the MediaFileManager.
+    return ""

streamlit/elements/map.py

@@ -357,8 +357,7 @@ def _get_lat_or_lon_col_name(
                 f"Map data must contain a {human_readable_name} column named: "
                 f"{formatted_allowed_col_name}. Existing columns: {formmated_col_names}"
             )
-        else:
-            col_name = candidate_col_name
+        col_name = candidate_col_name
 
     # Check that the column is well-formed.
     # IMPLEMENTATION NOTE: We can't use isnull().values.any() because .values can return

streamlit/elements/markdown.py

@@ -22,7 +22,7 @@ from streamlit.string_util import clean_text, validate_icon_or_emoji
 from streamlit.type_util import SupportsStr, is_sympy_expression
 
 if TYPE_CHECKING:
-    import sympy  # type: ignore
+    import sympy
 
     from streamlit.delta_generator import DeltaGenerator
 

streamlit/elements/plotly_chart.py

@@ -29,7 +29,7 @@ from typing import (
     overload,
 )
 
-from typing_extensions import TypeAlias
+from typing_extensions import Required, TypeAlias
 
 from streamlit import type_util
 from streamlit.deprecation_util import show_deprecation_warning
@@ -159,10 +159,10 @@ class PlotlySelectionState(TypedDict, total=False):
 
     """
 
-    points: list[dict[str, Any]]
-    point_indices: list[int]
-    box: list[dict[str, Any]]
-    lasso: list[dict[str, Any]]
+    points: Required[list[dict[str, Any]]]
+    point_indices: Required[list[int]]
+    box: Required[list[dict[str, Any]]]
+    lasso: Required[list[dict[str, Any]]]
 
 
 class PlotlyState(TypedDict, total=False):
@@ -205,7 +205,7 @@ class PlotlyState(TypedDict, total=False):
 
     """
 
-    selection: PlotlySelectionState
+    selection: Required[PlotlySelectionState]
 
 
 @dataclass
@@ -231,7 +231,7 @@ class PlotlyChartSelectionSerde:
         )
 
         if "selection" not in selection_state:
-            selection_state = empty_selection_state
+            selection_state = empty_selection_state  # type: ignore[unreachable]
 
         return cast("PlotlyState", AttributeDictionary(selection_state))
 
@@ -537,8 +537,7 @@ class PlotlyMixin:
 
             self.dg._enqueue("plotly_chart", plotly_chart_proto)
             return cast("PlotlyState", widget_state.value)
-        else:
-            return self.dg._enqueue("plotly_chart", plotly_chart_proto)
+        return self.dg._enqueue("plotly_chart", plotly_chart_proto)
 
     @property
     def dg(self) -> DeltaGenerator:

streamlit/elements/toast.py

@@ -31,8 +31,7 @@ def validate_text(toast_text: SupportsStr) -> SupportsStr:
         raise StreamlitAPIException(
             "Toast body cannot be blank - please provide a message."
         )
-    else:
-        return toast_text
+    return toast_text
 
 
 class ToastMixin:

streamlit/elements/vega_charts.py

@@ -31,7 +31,7 @@ from typing import (
     overload,
 )
 
-from typing_extensions import TypeAlias
+from typing_extensions import Required, TypeAlias
 
 from streamlit import dataframe_util, type_util
 from streamlit.elements.lib import dicttools
@@ -218,7 +218,7 @@ class VegaLiteState(TypedDict, total=False):
 
     """
 
-    selection: AttributeDictionary
+    selection: Required[AttributeDictionary]
 
 
 @dataclass
@@ -242,7 +242,7 @@ class VegaLiteStateSerde:
         )
 
         if "selection" not in selection_state:
-            selection_state = empty_selection_state
+            selection_state = empty_selection_state  # type: ignore[unreachable]
 
         return cast("VegaLiteState", AttributeDictionary(selection_state))
 
@@ -1479,7 +1479,7 @@ class VegaChartsMixin:
         use_container_width: bool | None = None,
         theme: Literal["streamlit"] | None = "streamlit",
         key: Key | None = None,
-        on_select: Literal["ignore"],  # No default value here to make it work with mypy
+        on_select: Literal["ignore"] = "ignore",
         selection_mode: str | Iterable[str] | None = None,
     ) -> DeltaGenerator: ...
 
@@ -1492,7 +1492,7 @@ class VegaChartsMixin:
         use_container_width: bool | None = None,
         theme: Literal["streamlit"] | None = "streamlit",
         key: Key | None = None,
-        on_select: Literal["rerun"] | WidgetCallback = "rerun",
+        on_select: Literal["rerun"] | WidgetCallback,
         selection_mode: str | Iterable[str] | None = None,
     ) -> VegaLiteState: ...
 
@@ -1638,7 +1638,7 @@ class VegaChartsMixin:
         use_container_width: bool | None = None,
         theme: Literal["streamlit"] | None = "streamlit",
         key: Key | None = None,
-        on_select: Literal["ignore"],  # No default value here to make it work with mypy
+        on_select: Literal["ignore"] = "ignore",
         selection_mode: str | Iterable[str] | None = None,
         **kwargs: Any,
     ) -> DeltaGenerator: ...
@@ -1653,7 +1653,7 @@ class VegaChartsMixin:
         use_container_width: bool | None = None,
         theme: Literal["streamlit"] | None = "streamlit",
         key: Key | None = None,
-        on_select: Literal["rerun"] | WidgetCallback = "rerun",
+        on_select: Literal["rerun"] | WidgetCallback,
         selection_mode: str | Iterable[str] | None = None,
         **kwargs: Any,
     ) -> VegaLiteState: ...

streamlit/elements/widgets/chat.py

@@ -127,7 +127,7 @@ def _process_avatar_input(
 
     if avatar is None:
         return AvatarType.ICON, ""
-    elif isinstance(avatar, str) and avatar in {item.value for item in PresetNames}:
+    if isinstance(avatar, str) and avatar in {item.value for item in PresetNames}:
         # On the frontend, we only support "assistant" and "user" for the avatar.
         return (
             AvatarType.ICON,
@@ -137,25 +137,24 @@ def _process_avatar_input(
                 else "user"
             ),
         )
-    elif isinstance(avatar, str) and is_emoji(avatar):
+    if isinstance(avatar, str) and is_emoji(avatar):
         return AvatarType.EMOJI, avatar
 
-    elif isinstance(avatar, str) and avatar.startswith(":material"):
+    if isinstance(avatar, str) and avatar.startswith(":material"):
         return AvatarType.ICON, validate_material_icon(avatar)
-    else:
-        try:
-            return AvatarType.IMAGE, image_to_url(
-                avatar,
-                width=WidthBehavior.ORIGINAL,
-                clamp=False,
-                channels="RGB",
-                output_format="auto",
-                image_id=delta_path,
-            )
-        except Exception as ex:
-            raise StreamlitAPIException(
-                "Failed to load the provided avatar value as an image."
-            ) from ex
+    try:
+        return AvatarType.IMAGE, image_to_url(
+            avatar,
+            width=WidthBehavior.ORIGINAL,
+            clamp=False,
+            channels="RGB",
+            output_format="auto",
+            image_id=delta_path,
+        )
+    except Exception as ex:
+        raise StreamlitAPIException(
+            "Failed to load the provided avatar value as an image."
+        ) from ex
 
 
 def _pop_upload_files(
@@ -208,16 +207,15 @@ class ChatInputSerde:
             return None
         if not self.accept_files:
             return ui_value.data
-        else:
-            uploaded_files = _pop_upload_files(ui_value.file_uploader_state)
-            for file in uploaded_files:
-                if self.allowed_types and not isinstance(file, DeletedFile):
-                    enforce_filename_restriction(file.name, self.allowed_types)
-
-            return ChatInputValue(
-                text=ui_value.data,
-                files=uploaded_files,
-            )
+        uploaded_files = _pop_upload_files(ui_value.file_uploader_state)
+        for file in uploaded_files:
+            if self.allowed_types and not isinstance(file, DeletedFile):
+                enforce_filename_restriction(file.name, self.allowed_types)
+
+        return ChatInputValue(
+            text=ui_value.data,
+            files=uploaded_files,
+        )
 
     def serialize(self, v: str | None) -> ChatInputValueProto:
         return ChatInputValueProto(data=v)

streamlit/elements/widgets/file_uploader.py

@@ -127,7 +127,7 @@ class FileUploaderSerde:
 
         if not files:
             return state_proto
-        elif not isinstance(files, list):
+        if not isinstance(files, list):
             files = [files]
 
         for f in files:
@@ -170,7 +170,7 @@ class FileUploaderMixin:
         disabled: bool = False,
         label_visibility: LabelVisibility = "visible",
         width: WidthWithoutContent = "stretch",
-    ) -> list[UploadedFile] | None: ...
+    ) -> list[UploadedFile]: ...
 
     # 1. type is given as not a keyword-only argument
     # 2. accept_multiple_files = False or omitted
@@ -213,7 +213,7 @@ class FileUploaderMixin:
         disabled: bool = False,
         label_visibility: LabelVisibility = "visible",
         width: WidthWithoutContent = "stretch",
-    ) -> list[UploadedFile] | None: ...
+    ) -> list[UploadedFile]: ...
 
     # 1. type is skipped or a keyword argument
     # 2. accept_multiple_files = False or omitted
@@ -497,7 +497,7 @@ class FileUploaderMixin:
 
         if isinstance(widget_state.value, DeletedFile):
             return None
-        elif isinstance(widget_state.value, list):
+        if isinstance(widget_state.value, list):
             return [f for f in widget_state.value if not isinstance(f, DeletedFile)]
 
         return widget_state.value

streamlit/elements/widgets/number_input.py

@@ -116,7 +116,7 @@ class NumberInputMixin:
     def number_input(
         self,
         label: str,
-        min_value: int | None = None,
+        min_value: None = None,
         *,
         max_value: int,
         value: IntOrNone | Literal["min"] = "min",
@@ -166,8 +166,8 @@ class NumberInputMixin:
     def number_input(
         self,
         label: str,
-        min_value: int | None = None,
-        max_value: int | None = None,
+        min_value: None = None,
+        max_value: None = None,
         value: IntOrNone | Literal["min"] = "min",
         *,
         step: int,

streamlit/elements/widgets/radio.py

@@ -18,6 +18,8 @@ from dataclasses import dataclass
 from textwrap import dedent
 from typing import TYPE_CHECKING, Any, Callable, Generic, cast, overload
 
+from typing_extensions import Never
+
 from streamlit.dataframe_util import OptionSequence, convert_anything_to_list
 from streamlit.elements.lib.form_utils import current_form_id
 from streamlit.elements.lib.options_selector_utils import index_, maybe_coerce_enum
@@ -79,6 +81,25 @@ class RadioSerde(Generic[T]):
 
 
 class RadioMixin:
+    @overload
+    def radio(
+        self,
+        label: str,
+        options: Sequence[Never],
+        index: int = 0,
+        format_func: Callable[[Any], Any] = str,
+        key: Key | None = None,
+        help: str | None = None,
+        on_change: WidgetCallback | None = None,
+        args: WidgetArgs | None = None,
+        kwargs: WidgetKwargs | None = None,
+        *,  # keyword-only args:
+        disabled: bool = False,
+        horizontal: bool = False,
+        captions: Sequence[str] | None = None,
+        label_visibility: LabelVisibility = "visible",
+    ) -> None: ...
+
     @overload
     def radio(
         self,
@@ -341,12 +362,11 @@ class RadioMixin:
         def handle_captions(caption: str | None) -> str:
             if caption is None:
                 return ""
-            elif isinstance(caption, str):
+            if isinstance(caption, str):
                 return caption
-            else:
-                raise StreamlitAPIException(
-                    f"Radio captions must be strings. Passed type: {type(caption).__name__}"
-                )
+            raise StreamlitAPIException(
+                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:

streamlit/elements/widgets/select_slider.py

@@ -102,13 +102,12 @@ class SelectSliderSerde(Generic[T]):
             if start > end:
                 slider_value = [end, start]
             return slider_value
-        else:
-            return [index_(self.options, v)]
+        return [index_(self.options, v)]
 
 
 class SelectSliderMixin:
     @overload
-    def select_slider(  # type: ignore[overload-overlap]
+    def select_slider(
         self,
         label: str,
         options: OptionSequence[T],
@@ -125,18 +124,6 @@ class SelectSliderMixin:
         width: WidthWithoutContent = "stretch",
     ) -> tuple[T, T]: ...
 
-    # The overload-overlap error given by mypy here stems from
-    # the fact that
-    #
-    # > opt:List[object] = [1, 2, "3"]
-    # > select_slider("foo", options=opt, value=[1, 2])
-    #
-    # matches both overloads; "opt" matches
-    # OptionsSequence[T] in each case, binding T to object.
-    # However, the list[int] type of "value" can be interpreted
-    # as subtype of object, or as a subtype of List[object],
-    # meaning it matches both signatures.
-
     @overload
     def select_slider(
         self,
@@ -367,15 +354,14 @@ class SelectSliderMixin:
                 if start > end:
                     slider_value = [end, start]
                 return slider_value
-            else:
-                # Simplify future logic by always making value a list
-                try:
-                    return [index_(opt, v)]
-                except ValueError:
-                    if value is not None:
-                        raise
-
-                    return [0]
+            # Simplify future logic by always making value a list
+            try:
+                return [index_(opt, v)]
+            except ValueError:
+                if value is not None:
+                    raise
+
+                return [0]
 
         # Convert element to index of the elements
         slider_value = as_index_list(value)