streamlit 1.45.1__py3-none-any.whl → 1.46.1__py3-none-any.whl

streamlit/runtime/state/session_state.py

@@ -84,7 +84,7 @@ class WStates(MutableMapping[str, Any]):
     states: dict[str, WState] = field(default_factory=dict)
     widget_metadata: dict[str, WidgetMetadata[Any]] = field(default_factory=dict)
 
-    def __repr__(self):
+    def __repr__(self) -> str:
         return util.repr_(self)
 
     def __getitem__(self, k: str) -> Any:
@@ -123,9 +123,9 @@ class WStates(MutableMapping[str, Any]):
             # Array types are messages with data in a `data` field
             value = value.data
         elif value_field_name == "json_value":
-            value = json.loads(value)
+            value = json.loads(cast("str", value))
 
-        deserialized = metadata.deserializer(value, metadata.id)
+        deserialized = metadata.deserializer(value)
 
         # Update metadata to reflect information from WidgetState proto
         self.set_widget_metadata(
@@ -147,7 +147,7 @@ class WStates(MutableMapping[str, Any]):
     def __len__(self) -> int:
         return len(self.states)
 
-    def __iter__(self):
+    def __iter__(self) -> Iterator[str]:
         # For this and many other methods, we can't simply delegate to the
         # states field, because we need to invoke `__getitem__` for any
         # values, to handle deserialization and unwrapping of values.
@@ -249,11 +249,10 @@ class WStates(MutableMapping[str, Any]):
         """Return a list of serialized widget values for each widget with a value."""
         states = [
             self.get_serialized(widget_id)
-            for widget_id in self.states.keys()
+            for widget_id in self.states
             if self.get_serialized(widget_id)
         ]
-        states = cast("list[WidgetStateProto]", states)
-        return states
+        return cast("list[WidgetStateProto]", states)
 
     def call_callback(self, widget_id: str) -> None:
         """Call the given widget's callback and return the callback's
@@ -262,14 +261,24 @@ class WStates(MutableMapping[str, Any]):
         If the widget doesn't exist, raise an Exception.
         """
         metadata = self.widget_metadata.get(widget_id)
-        assert metadata is not None
+
+        if metadata is None:
+            raise RuntimeError(f"Widget {widget_id} not found.")
+
         callback = metadata.callback
         if callback is None:
             return
 
         args = metadata.callback_args or ()
         kwargs = metadata.callback_kwargs or {}
-        callback(*args, **kwargs)
+
+        ctx = get_script_run_ctx()
+        if ctx and metadata.fragment_id is not None:
+            ctx.in_fragment_callback = True
+            callback(*args, **kwargs)
+            ctx.in_fragment_callback = False
+        else:
+            callback(*args, **kwargs)
 
 
 def _missing_key_error_message(key: str) -> str:
@@ -319,11 +328,11 @@ class KeyIdMapper:
         self._key_id_mapping.update(other._key_id_mapping)
         self._id_key_mapping.update(other._id_key_mapping)
 
-    def clear(self):
+    def clear(self) -> None:
         self._key_id_mapping.clear()
         self._id_key_mapping.clear()
 
-    def delete(self, key: str):
+    def delete(self, key: str) -> None:
         widget_id = self._key_id_mapping[key]
         del self._key_id_mapping[key]
         del self._id_key_mapping[widget_id]
@@ -364,7 +373,7 @@ class SessionState:
     # widget state at one point.
     query_params: QueryParams = field(default_factory=QueryParams)
 
-    def __repr__(self):
+    def __repr__(self) -> str:
         return util.repr_(self)
 
     # is it possible for a value to get through this without being deserialized?
@@ -376,7 +385,7 @@ class SessionState:
         for key_or_wid in self:
             try:
                 self._old_state[key_or_wid] = self[key_or_wid]
-            except KeyError:
+            except KeyError:  # noqa: PERF203
                 # handle key errors from widget state not having metadata gracefully
                 # https://github.com/streamlit/streamlit/issues/7206
                 pass
@@ -422,10 +431,10 @@ class SessionState:
         for widgets that don't have user_keys defined, and which aren't
         exposed to user code).
         """
-        old_keys = {self._get_widget_id(k) for k in self._old_state.keys()}
+        old_keys = {self._get_widget_id(k) for k in self._old_state}
         new_widget_keys = set(self._new_widget_state.keys())
         new_session_state_keys = {
-            self._get_widget_id(k) for k in self._new_session_state.keys()
+            self._get_widget_id(k) for k in self._new_session_state
         }
         return old_keys | new_widget_keys | new_session_state_keys
 
@@ -462,7 +471,10 @@ class SessionState:
 
         At least one of the arguments must have a value.
         """
-        assert user_key is not None or widget_id is not None
+        if user_key is None and widget_id is None:
+            raise ValueError(
+                "user_key and widget_id cannot both be None. This should never happen."
+            )
 
         if user_key is not None:
             try:
@@ -569,7 +581,7 @@ class SessionState:
         for wid in changed_widget_ids:
             try:
                 self._new_widget_state.call_callback(wid)
-            except RerunException:
+            except RerunException:  # noqa: PERF203
                 st.warning("Calling st.rerun() within a callback is a no-op.")
 
     def _widget_changed(self, widget_id: str) -> bool:
@@ -602,9 +614,10 @@ class SessionState:
             if metadata is not None:
                 if metadata.value_type == "trigger_value":
                     self._new_widget_state[state_id] = Value(False)
-                elif metadata.value_type == "string_trigger_value":
-                    self._new_widget_state[state_id] = Value(None)
-                elif metadata.value_type == "chat_input_value":
+                elif metadata.value_type in {
+                    "string_trigger_value",
+                    "chat_input_value",
+                }:
                     self._new_widget_state[state_id] = Value(None)
 
         for state_id in self._old_state:
@@ -612,9 +625,10 @@ class SessionState:
             if metadata is not None:
                 if metadata.value_type == "trigger_value":
                     self._old_state[state_id] = False
-                elif metadata.value_type == "string_trigger_value":
-                    self._old_state[state_id] = None
-                elif metadata.value_type == "chat_input_value":
+                elif metadata.value_type in {
+                    "string_trigger_value",
+                    "chat_input_value",
+                }:
                     self._old_state[state_id] = None
 
     def _remove_stale_widgets(self, active_widget_ids: set[str]) -> None:
@@ -683,7 +697,7 @@ class SessionState:
             # This is the first time the widget is registered, so we save its
             # value in widget state.
             deserializer = metadata.deserializer
-            initial_widget_value = deepcopy(deserializer(None, metadata.id))
+            initial_widget_value = deepcopy(deserializer(None))
             self._new_widget_state.set_from_value(widget_id, initial_widget_value)
 
         # Get the current value of the widget for use as its return value.
@@ -723,11 +737,15 @@ class SessionState:
         for k in self:
             try:
                 pickle.dumps(self[k])
-            except Exception as e:
-                err_msg = f"""Cannot serialize the value (of type `{type(self[k])}`) of '{k}' in st.session_state.
-                Streamlit has been configured to use [pickle](https://docs.python.org/3/library/pickle.html) to
-                serialize session_state values. Please convert the value to a pickle-serializable type. To learn
-                more about this behavior, see [our docs](https://docs.streamlit.io/knowledge-base/using-streamlit/serializable-session-state). """
+            except Exception as e:  # noqa: PERF203
+                err_msg = (
+                    f"Cannot serialize the value (of type `{type(self[k])}`) of '{k}' in "
+                    "st.session_state. Streamlit has been configured to use "
+                    "[pickle](https://docs.python.org/3/library/pickle.html) to "
+                    "serialize session_state values. Please convert the value to a "
+                    "pickle-serializable type. To learn more about this behavior, "
+                    "see [our docs](https://docs.streamlit.io/knowledge-base/using-streamlit/serializable-session-state)."
+                )
                 raise UnserializableSessionStateError(err_msg) from e
 
     def maybe_check_serializable(self) -> None:
@@ -751,14 +769,14 @@ def _is_stale_widget(
 ) -> bool:
     if not metadata:
         return True
-    elif metadata.id in active_widget_ids:
-        return False
+
     # If we're running 1 or more fragments, but this widget is unrelated to any of the
     # fragments that we're running, then it should not be marked as stale as its value
     # may still be needed for a future fragment run or full script run.
-    elif fragment_ids_this_run and metadata.fragment_id not in fragment_ids_this_run:
-        return False
-    return True
+    return not (
+        metadata.id in active_widget_ids
+        or (fragment_ids_this_run and metadata.fragment_id not in fragment_ids_this_run)
+    )
 
 
 @dataclass

streamlit/runtime/stats.py

@@ -63,7 +63,7 @@ class CacheStat(NamedTuple):
 def group_stats(stats: list[CacheStat]) -> list[CacheStat]:
     """Group a list of CacheStats by category_name and cache_name and sum byte_length."""
 
-    def key_function(individual_stat):
+    def key_function(individual_stat: CacheStat) -> tuple[str, str]:
         return individual_stat.category_name, individual_stat.cache_name
 
     result: list[CacheStat] = []
@@ -90,7 +90,7 @@ class CacheStatsProvider(Protocol):
 
 
 class StatsManager:
-    def __init__(self):
+    def __init__(self) -> None:
         self._cache_stats_providers: list[CacheStatsProvider] = []
 
     def register_provider(self, provider: CacheStatsProvider) -> None:

streamlit/runtime/uploaded_file_manager.py

@@ -65,7 +65,7 @@ class UploadedFile(io.BytesIO):
     initialized with `bytes`.
     """
 
-    def __init__(self, record: UploadedFileRec, file_urls: FileURLsProto):
+    def __init__(self, record: UploadedFileRec, file_urls: FileURLsProto) -> None:
         # BytesIO's copy-on-write semantics doesn't seem to be mentioned in
         # the Python docs - possibly because it's a CPython-only optimization
         # and not guaranteed to be in other Python runtimes. But it's detailed

streamlit/runtime/websocket_session_manager.py

@@ -67,9 +67,11 @@ class WebsocketSessionManager(SessionManager):
         existing_session_id: str | None = None,
         session_id_override: str | None = None,
     ) -> str:
-        assert not (existing_session_id and session_id_override), (
-            "Only one of existing_session_id and session_id_override should be truthy"
-        )
+        if existing_session_id and session_id_override:
+            raise RuntimeError(
+                "Only one of existing_session_id and session_id_override should be truthy. "
+                "This should never happen."
+            )
 
         if existing_session_id in self._active_session_info_by_id:
             _LOGGER.warning(
@@ -109,9 +111,11 @@ class WebsocketSessionManager(SessionManager):
             "Created new session for client %s. Session ID: %s", id(client), session.id
         )
 
-        assert session.id not in self._active_session_info_by_id, (
-            f"session.id '{session.id}' registered multiple times!"
-        )
+        if session.id in self._active_session_info_by_id:
+            raise RuntimeError(
+                f"session.id '{session.id}' registered multiple times. "
+                "This should never happen."
+            )
 
         self._active_session_info_by_id[session.id] = ActiveSessionInfo(client, session)
         return session.id

streamlit/source_util.py

@@ -15,7 +15,7 @@
 from __future__ import annotations
 
 import re
-from typing import TYPE_CHECKING, TypedDict
+from typing import TYPE_CHECKING, TextIO, TypedDict
 
 from typing_extensions import NotRequired, TypeAlias
 
@@ -38,7 +38,7 @@ class PageInfo(TypedDict):
     url_pathname: NotRequired[str]
 
 
-def open_python_file(filename: str):
+def open_python_file(filename: str) -> TextIO:
     """Open a read-only Python file taking proper care of its encoding.
 
     In Python 3, we would like all files to be opened with utf-8 encoding.
@@ -52,8 +52,7 @@ def open_python_file(filename: str):
         # Open file respecting PEP263 encoding. If no encoding header is
         # found, opens as utf-8.
         return tokenize.open(filename)
-    else:
-        return open(filename, encoding="utf-8")
+    return open(filename, encoding="utf-8")
 
 
 PAGE_FILENAME_REGEX = re.compile(r"([0-9]*)[_ -]*(.*)\.py")
@@ -62,9 +61,12 @@ PAGE_FILENAME_REGEX = re.compile(r"([0-9]*)[_ -]*(.*)\.py")
 def page_sort_key(script_path: Path) -> tuple[float, str]:
     matches = re.findall(PAGE_FILENAME_REGEX, script_path.name)
 
-    # Failing this assert should only be possible if script_path isn't a Python
+    # Failing this should only be possible if script_path isn't a Python
     # file, which should never happen.
-    assert len(matches) > 0, f"{script_path} is not a Python file"
+    if len(matches) == 0:
+        raise ValueError(
+            f"{script_path} is not a Python file. This should never happen."
+        )
 
     [(number, label)] = matches
     label = label.lower()

streamlit/static/index.html

@@ -25,21 +25,7 @@
     <link rel="shortcut icon" href="./favicon.png" />
     <link
       rel="preload"
-      href="./static/media/SourceSansPro-Regular.DZLUzqI4.woff2"
-      as="font"
-      type="font/woff2"
-      crossorigin
-    />
-    <link
-      rel="preload"
-      href="./static/media/SourceSansPro-SemiBold.sKQIyTMz.woff2"
-      as="font"
-      type="font/woff2"
-      crossorigin
-    />
-    <link
-      rel="preload"
-      href="./static/media/SourceSansPro-Bold.-6c9oR8J.woff2"
+      href="./static/media/SourceSansVF-Upright.ttf.BsWL4Kly.woff2"
       as="font"
       type="font/woff2"
       crossorigin
@@ -51,8 +37,8 @@
     <script>
       window.prerenderReady = false
     </script>
-    <script type="module" crossorigin src="./static/js/index.C1z8KpLA.js"></script>
-    <link rel="stylesheet" crossorigin href="./static/css/index.DqDwtg6_.css">
+    <script type="module" crossorigin src="./static/js/index.C1NIn1Y2.js"></script>
+    <link rel="stylesheet" crossorigin href="./static/css/index.CJVRHjQZ.css">
   </head>
   <body>
     <noscript>You need to enable JavaScript to run this app.</noscript>