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

streamlit/errors.py

@@ -15,10 +15,14 @@
 from __future__ import annotations
 
 import os
-from typing import Any, Literal
+from typing import TYPE_CHECKING, Any, Literal
 
 from streamlit import util
 
+if TYPE_CHECKING:
+    from collections.abc import Collection
+    from datetime import date, time
+
 
 class Error(Exception):
     """The base class for all exceptions thrown by Streamlit.
@@ -49,7 +53,7 @@ class FragmentStorageKeyError(Error, KeyError):
     pass
 
 
-class FragmentHandledException(Exception):
+class FragmentHandledException(Exception):  # noqa: N818
     """An exception that is raised by the fragment
     when it has handled the exception itself.
     """
@@ -57,15 +61,15 @@ class FragmentHandledException(Exception):
     pass
 
 
-class NoStaticFiles(Error):
+class NoStaticFiles(Error):  # noqa: N818
     pass
 
 
-class NoSessionContext(Error):
+class NoSessionContext(Error):  # noqa: N818
     pass
 
 
-class MarkdownFormattedException(Error):
+class MarkdownFormattedException(Error):  # noqa: N818
     """Exceptions with Markdown in their description.
 
     Instances of this class can use markdown in their messages, which will get
@@ -104,7 +108,7 @@ class StreamlitAuthError(StreamlitAPIException):
 class StreamlitDuplicateElementId(DuplicateWidgetID):
     """An exception raised when the auto-generated ID of an element is not unique."""
 
-    def __init__(self, element_type: str):
+    def __init__(self, element_type: str) -> None:
         super().__init__(
             f"There are multiple `{element_type}` elements with the same "
             "auto-generated ID. When this element is created, it is assigned an "
@@ -118,7 +122,7 @@ class StreamlitDuplicateElementId(DuplicateWidgetID):
 class StreamlitDuplicateElementKey(DuplicateWidgetID):
     """An exception raised when the key of an element is not unique."""
 
-    def __init__(self, user_key: str):
+    def __init__(self, user_key: str) -> None:
         super().__init__(
             f"There are multiple elements with the same `key='{user_key}'`. "
             "To fix this, please make sure that the `key` argument is unique for "
@@ -137,7 +141,7 @@ class StreamlitAPIWarning(StreamlitAPIException, Warning):
     instead.
     """
 
-    def __init__(self, *args):
+    def __init__(self, *args: Any) -> None:
         super().__init__(*args)
         import inspect
         import traceback
@@ -154,7 +158,7 @@ class StreamlitModuleNotFoundError(StreamlitAPIWarning):
     that is not one of our core dependencies.
     """
 
-    def __init__(self, module_name, *args):
+    def __init__(self, module_name: str, *args: Any) -> None:
         message = (
             f'This Streamlit command requires module "{module_name}" to be installed.'
         )
@@ -162,7 +166,7 @@ class StreamlitModuleNotFoundError(StreamlitAPIWarning):
 
 
 class LocalizableStreamlitException(StreamlitAPIException):
-    def __init__(self, message: str, **kwargs):
+    def __init__(self, message: str, **kwargs: Any) -> None:
         super().__init__((message).format(**kwargs))
         self._exec_kwargs = kwargs
 
@@ -171,23 +175,10 @@ class LocalizableStreamlitException(StreamlitAPIException):
         return self._exec_kwargs
 
 
-# st.set_page_config
-class StreamlitSetPageConfigMustBeFirstCommandError(LocalizableStreamlitException):
-    """Exception raised when the set_page_config command is not the first executed streamlit command."""
-
-    def __init__(self):
-        super().__init__(
-            "`set_page_config()` can only be called once per app page, "
-            "and must be called as the first Streamlit command in your script.\n\n"
-            "For more information refer to the [docs]"
-            "(https://docs.streamlit.io/develop/api-reference/configuration/st.set_page_config)."
-        )
-
-
 class StreamlitInvalidPageLayoutError(LocalizableStreamlitException):
     """Exception raised when an invalid value is specified for layout."""
 
-    def __init__(self, layout: str):
+    def __init__(self, layout: str) -> None:
         super().__init__(
             '`layout` must be `"centered"` or `"wide"` (got `"{layout}"`)',
             layout=layout,
@@ -197,9 +188,10 @@ class StreamlitInvalidPageLayoutError(LocalizableStreamlitException):
 class StreamlitInvalidSidebarStateError(LocalizableStreamlitException):
     """Exception raised when an invalid value is specified for `initial_sidebar_state`."""
 
-    def __init__(self, initial_sidebar_state: str):
+    def __init__(self, initial_sidebar_state: str) -> None:
         super().__init__(
-            '`initial_sidebar_state` must be `"auto"` or `"expanded"` or `"collapsed"` (got `"{initial_sidebar_state}"`)',
+            '`initial_sidebar_state` must be `"auto"` or `"expanded"` or '
+            '`"collapsed"` (got `"{initial_sidebar_state}"`)',
             initial_sidebar_state=initial_sidebar_state,
         )
 
@@ -207,7 +199,7 @@ class StreamlitInvalidSidebarStateError(LocalizableStreamlitException):
 class StreamlitInvalidMenuItemKeyError(LocalizableStreamlitException):
     """Exception raised when an invalid key is specified."""
 
-    def __init__(self, key: str):
+    def __init__(self, key: str) -> None:
         super().__init__(
             'We only accept the keys: `"Get help"`, `"Report a bug"`, and `"About"` (`"{key}"` is not a valid key.)',
             key=key,
@@ -217,7 +209,7 @@ class StreamlitInvalidMenuItemKeyError(LocalizableStreamlitException):
 class StreamlitInvalidURLError(LocalizableStreamlitException):
     """Exception raised when an invalid URL is specified for any of the menu items except for “About”."""
 
-    def __init__(self, url: str):
+    def __init__(self, url: str) -> None:
         super().__init__(
             '"{url}" is a not a valid URL. '
             'You must use a fully qualified domain beginning with "http://", "https://", or "mailto:".',
@@ -229,7 +221,7 @@ class StreamlitInvalidURLError(LocalizableStreamlitException):
 class StreamlitInvalidColumnSpecError(LocalizableStreamlitException):
     """Exception raised when no weights are specified, or a negative weight is specified."""
 
-    def __init__(self):
+    def __init__(self) -> None:
         super().__init__(
             "The `spec` argument to `st.columns` must be either a "
             "positive integer (number of columns) or a list of positive numbers (width ratios of the columns). "
@@ -241,7 +233,7 @@ class StreamlitInvalidColumnSpecError(LocalizableStreamlitException):
 class StreamlitInvalidVerticalAlignmentError(LocalizableStreamlitException):
     """Exception raised when an invalid value is specified for vertical_alignment."""
 
-    def __init__(self, vertical_alignment: str):
+    def __init__(self, vertical_alignment: str) -> None:
         super().__init__(
             'The `vertical_alignment` argument to `st.columns` must be `"top"`, `"center"`, or `"bottom"`. \n'
             "The argument passed was {vertical_alignment}.",
@@ -252,9 +244,9 @@ class StreamlitInvalidVerticalAlignmentError(LocalizableStreamlitException):
 class StreamlitInvalidColumnGapError(LocalizableStreamlitException):
     """Exception raised when an invalid value is specified for gap."""
 
-    def __init__(self, gap: str):
+    def __init__(self, gap: str) -> None:
         super().__init__(
-            'The `gap` argument to `st.columns` must be `"small"`, `"medium"`, or `"large"`. \n'
+            'The `gap` argument to `st.columns` must be `"small"`, `"medium"`, `"large"`, or `"none"`. \n'
             "The argument passed was {gap}.",
             gap=gap,
         )
@@ -264,7 +256,9 @@ class StreamlitInvalidColumnGapError(LocalizableStreamlitException):
 class StreamlitSelectionCountExceedsMaxError(LocalizableStreamlitException):
     """Exception raised when there are more default selections specified than the max allowable selections."""
 
-    def __init__(self, current_selections_count: int, max_selections_count: int):
+    def __init__(
+        self, current_selections_count: int, max_selections_count: int
+    ) -> None:
         super().__init__(
             "Multiselect has {current_selections_count} {current_selections_noun} "
             "selected but `max_selections` is set to {max_selections_count}. "
@@ -291,7 +285,7 @@ class StreamlitMixedNumericTypesError(LocalizableStreamlitException):
         min_value: int | float | None,
         max_value: int | float | None,
         step: int | float | None,
-    ):
+    ) -> None:
         value_type = None
         min_value_type = None
         max_value_type = None
@@ -327,7 +321,11 @@ class StreamlitMixedNumericTypesError(LocalizableStreamlitException):
 class StreamlitValueBelowMinError(LocalizableStreamlitException):
     """Exception raised when the `min_value` is greater than the `value`."""
 
-    def __init__(self, value: int | float, min_value: int | float):
+    def __init__(
+        self,
+        value: int | float | date | time,
+        min_value: int | float | date | time,
+    ) -> None:
         super().__init__(
             "The `value` {value} is less than the `min_value` {min_value}.",
             value=value,
@@ -338,7 +336,11 @@ class StreamlitValueBelowMinError(LocalizableStreamlitException):
 class StreamlitValueAboveMaxError(LocalizableStreamlitException):
     """Exception raised when the `max_value` is less than the `value`."""
 
-    def __init__(self, value: int | float, max_value: int | float):
+    def __init__(
+        self,
+        value: int | float | date | time,
+        max_value: int | float | date | time,
+    ) -> None:
         super().__init__(
             "The `value` {value} is greater than the `max_value` {max_value}.",
             value=value,
@@ -349,7 +351,7 @@ class StreamlitValueAboveMaxError(LocalizableStreamlitException):
 class StreamlitJSNumberBoundsError(LocalizableStreamlitException):
     """Exception raised when a number exceeds the Javascript limits."""
 
-    def __init__(self, message: str):
+    def __init__(self, message: str) -> None:
         super().__init__(message)
 
 
@@ -358,7 +360,7 @@ class StreamlitInvalidNumberFormatError(LocalizableStreamlitException):
     invalid characters.
     """
 
-    def __init__(self, format: str):
+    def __init__(self, format: str) -> None:
         super().__init__(
             "Format string for `st.number_input` contains invalid characters: {format}",
             format=format,
@@ -369,7 +371,7 @@ class StreamlitInvalidNumberFormatError(LocalizableStreamlitException):
 class StreamlitMissingPageLabelError(LocalizableStreamlitException):
     """Exception raised when a page_link is created without a label."""
 
-    def __init__(self):
+    def __init__(self) -> None:
         super().__init__(
             "The `label` param is required for external links used with `st.page_link` - please provide a `label`."
         )
@@ -380,7 +382,7 @@ class StreamlitPageNotFoundError(LocalizableStreamlitException):
 
     def __init__(
         self, page: str, main_script_directory: str, uses_pages_directory: bool
-    ):
+    ) -> None:
         directory = os.path.basename(main_script_directory)
 
         message = (
@@ -408,14 +410,16 @@ class StreamlitPageNotFoundError(LocalizableStreamlitException):
 class StreamlitFragmentWidgetsNotAllowedOutsideError(LocalizableStreamlitException):
     """Exception raised when the fragment attempts to write to an element outside of its container."""
 
-    def __init__(self):
+    def __init__(self) -> None:
         super().__init__("Fragments cannot write widgets to outside containers.")
 
 
 class StreamlitInvalidFormCallbackError(LocalizableStreamlitException):
-    """Exception raised a `on_change` callback is set on any element in a form except for the `st.form_submit_button`."""
+    """Exception raised a `on_change` callback is set on any element in a form except for
+    the `st.form_submit_button`.
+    """
 
-    def __init__(self):
+    def __init__(self) -> None:
         super().__init__(
             "Within a form, callbacks can only be defined on `st.form_submit_button`. "
             "Defining callbacks on other widgets inside a form is not allowed."
@@ -425,7 +429,7 @@ class StreamlitInvalidFormCallbackError(LocalizableStreamlitException):
 class StreamlitValueAssignmentNotAllowedError(LocalizableStreamlitException):
     """Exception raised when trying to set values where writes are not allowed."""
 
-    def __init__(self, key: str):
+    def __init__(self, key: str) -> None:
         super().__init__(
             "Values for the widget with `key` '{key}' cannot be set using `st.session_state`.",
             key=key,
@@ -433,7 +437,9 @@ class StreamlitValueAssignmentNotAllowedError(LocalizableStreamlitException):
 
 
 class StreamlitInvalidColorError(LocalizableStreamlitException):
-    def __init__(self, color):
+    def __init__(
+        self, color: str | Collection[Any] | tuple[int, int, int, int]
+    ) -> None:
         super().__init__(
             "This does not look like a valid color: {color}.\n\n"
             "Colors must be in one of the following formats:"
@@ -446,7 +452,7 @@ class StreamlitInvalidColorError(LocalizableStreamlitException):
 class StreamlitBadTimeStringError(LocalizableStreamlitException):
     """Exception Raised when a time string argument is passed that cannot be parsed."""
 
-    def __init__(self, time_string: str):
+    def __init__(self, time_string: str) -> None:
         super().__init__(
             "Time string doesn't look right. It should be formatted as"
             "`'1d2h34m'` or `2 days`, for example. Got: {time_string}",
@@ -457,14 +463,14 @@ class StreamlitBadTimeStringError(LocalizableStreamlitException):
 class StreamlitSecretNotFoundError(LocalizableStreamlitException, FileNotFoundError):
     """Exception raised when a secret cannot be found or parsed in the secrets.toml file."""
 
-    def __init__(self, message: str):
+    def __init__(self, message: str) -> None:
         super().__init__(message)
 
 
 class StreamlitInvalidWidthError(LocalizableStreamlitException):
     """Exception raised when an invalid width value is provided."""
 
-    def __init__(self, width: Any, allow_content: bool = False):
+    def __init__(self, width: Any, allow_content: bool = False) -> None:
         valid_values = "an integer (pixels) or 'stretch'"
         if allow_content:
             valid_values = "an integer (pixels), 'stretch', or 'content'"
@@ -474,3 +480,18 @@ class StreamlitInvalidWidthError(LocalizableStreamlitException):
             width=repr(width),
             valid_values=valid_values,
         )
+
+
+class StreamlitInvalidHeightError(LocalizableStreamlitException):
+    """Exception raised when an invalid height value is provided."""
+
+    def __init__(self, height: Any, allow_content: bool = False) -> None:
+        valid_values = "an integer (pixels) or 'stretch'"
+        if allow_content:
+            valid_values = "an integer (pixels), 'stretch', or 'content'"
+
+        super().__init__(
+            "Invalid height value: {height}. Height must be either {valid_values}.",
+            height=repr(height),
+            valid_values=valid_values,
+        )

streamlit/external/langchain/streamlit_callback_handler.py

@@ -36,6 +36,10 @@ This module is lazy-loaded.
 # by default.
 # mypy: disable-error-code="import-not-found, unused-ignore, misc"
 
+# Deactivate unused argument errors for this file since we need lots of
+# unused arguments to comply with the LangChain callback interface.
+# ruff: noqa: ARG002
+
 from __future__ import annotations
 
 import time
@@ -151,7 +155,7 @@ class LLMThought:
         labeler: LLMThoughtLabeler,
         expanded: bool,
         collapse_on_complete: bool,
-    ):
+    ) -> None:
         self._container = parent_container.status(
             labeler.get_initial_label(), expanded=expanded
         )
@@ -246,9 +250,10 @@ class LLMThought:
     def complete(self, final_label: str | None = None) -> None:
         """Finish the thought."""
         if final_label is None and self._state == LLMThoughtState.RUNNING_TOOL:
-            assert self._last_tool is not None, (
-                "_last_tool should never be null when _state == RUNNING_TOOL"
-            )
+            if self._last_tool is None:
+                raise RuntimeError(
+                    "_last_tool should never be null when _state == RUNNING_TOOL"
+                )
             final_label = self._labeler.get_tool_label(
                 self._last_tool, is_complete=True
             )
@@ -279,7 +284,7 @@ class StreamlitCallbackHandler(BaseCallbackHandler):
         expand_new_thoughts: bool = False,
         collapse_completed_thoughts: bool = False,
         thought_labeler: LLMThoughtLabeler | None = None,
-    ):
+    ) -> None:
         """Construct a new StreamlitCallbackHandler. This CallbackHandler is geared
         towards use with a LangChain Agent; it displays the Agent's LLM and tool-usage
         "thoughts" inside a series of Streamlit expanders.

streamlit/file_util.py

@@ -19,15 +19,19 @@ import errno
 import io
 import os
 from pathlib import Path
+from typing import IO, TYPE_CHECKING, Any, Final
 
 from streamlit import env_util, errors
 from streamlit.string_util import is_binary_string
 
+if TYPE_CHECKING:
+    from collections.abc import Generator
+
 # Configuration and credentials are stored inside the ~/.streamlit folder
-CONFIG_FOLDER_NAME = ".streamlit"
+CONFIG_FOLDER_NAME: Final = ".streamlit"
 
 # If enableStaticServing is enabled, static file served from the ./static folder
-APP_STATIC_FOLDER_NAME = "static"
+APP_STATIC_FOLDER_NAME: Final = "static"
 
 
 def get_encoded_file_data(
@@ -62,7 +66,7 @@ def get_encoded_file_data(
 
 
 @contextlib.contextmanager
-def streamlit_read(path, binary=False):
+def streamlit_read(path: str, binary: bool = False) -> Generator[IO[Any], None, None]:
     """Opens a context to read this file relative to the streamlit path.
 
     For example:
@@ -87,7 +91,7 @@ def streamlit_read(path, binary=False):
 
 
 @contextlib.contextmanager
-def streamlit_write(path, binary=False):
+def streamlit_write(path: str, binary: bool = False) -> Generator[IO[Any], None, None]:
     """Opens a file for writing within the streamlit path, and
     ensuring that the path exists.
 
@@ -128,12 +132,11 @@ def get_static_dir() -> str:
 
 def get_app_static_dir(main_script_path: str) -> str:
     """Get the folder where app static files live."""
-    main_script_path = Path(main_script_path)
-    static_dir = main_script_path.parent / APP_STATIC_FOLDER_NAME
+    static_dir = Path(main_script_path).parent / APP_STATIC_FOLDER_NAME
     return os.path.abspath(static_dir)
 
 
-def get_streamlit_file_path(*filepath) -> str:
+def get_streamlit_file_path(*filepath: str) -> str:
     """Return the full path to a file in ~/.streamlit.
 
     This doesn't guarantee that the file (or its directory) exists.
@@ -145,7 +148,7 @@ def get_streamlit_file_path(*filepath) -> str:
     return str(home / CONFIG_FOLDER_NAME / Path(*filepath))
 
 
-def get_project_streamlit_file_path(*filepath):
+def get_project_streamlit_file_path(*filepath: str) -> str:
     """Return the full path to a filepath in ${CWD}/.streamlit.
 
     This doesn't guarantee that the file (or its directory) exists.
@@ -153,6 +156,20 @@ def get_project_streamlit_file_path(*filepath):
     return str(Path.cwd() / CONFIG_FOLDER_NAME / Path(*filepath))
 
 
+def get_main_script_streamlit_file_path(main_script_path: str, filename: str) -> str:
+    """Return the full path to a file in the .streamlit folder relative to the
+    main script's path.
+
+    This doesn't guarantee that the file (or its directory) exists.
+    """
+
+    return str(
+        Path(os.path.abspath(os.path.dirname(main_script_path)))
+        / CONFIG_FOLDER_NAME
+        / filename
+    )
+
+
 def file_is_in_folder_glob(filepath: str, folderpath_glob: str) -> bool:
     """Test whether a file is in some folder with globbing support.
 
@@ -214,7 +231,7 @@ def file_in_pythonpath(filepath: str) -> bool:
     )
 
 
-def normalize_path_join(*args):
+def normalize_path_join(*args: str) -> str:
     """Return the normalized path of the joined path.
 
     Parameters
@@ -230,7 +247,7 @@ def normalize_path_join(*args):
     return os.path.normpath(os.path.join(*args))
 
 
-def get_main_script_directory(main_script):
+def get_main_script_directory(main_script: str) -> str:
     """Return the full path to the main script directory.
 
     Parameters

streamlit/git_util.py

@@ -16,10 +16,13 @@ from __future__ import annotations
 
 import os
 import re
-from typing import Any
+from typing import TYPE_CHECKING, cast
 
 from streamlit import util
 
+if TYPE_CHECKING:
+    from git import Commit, Remote, RemoteReference, Repo
+
 # Github has two URLs, one that is https and one that is ssh
 GITHUB_HTTP_URL = r"^https://(www\.)?github.com/(.+)/(.+)(?:.git)?$"
 GITHUB_SSH_URL = r"^git@github.com:(.+)/(.+)(?:.git)?$"
@@ -31,19 +34,17 @@ MIN_GIT_VERSION = (2, 7, 0)
 
 
 class GitRepo:
-    def __init__(self, path):
-        # If we have a valid repo, git_version will be a tuple of 3+ ints:
-        # (major, minor, patch, possible_additional_patch_number)
+    repo: Repo | None
+
+    def __init__(self, path: str) -> None:
+        # If we have a valid repo, git_version will be a tuple
+        # of 3+ ints: (major, minor, patch, possible_additional_patch_number)
         self.git_version: tuple[int, ...] | None = None
 
         try:
             import git
 
-            # GitPython is not fully typed, and mypy is outputting inconsistent
-            # type errors on Mac and Linux. We bypass type checking entirely
-            # by re-declaring the `git` import as an "Any".
-            git_package: Any = git
-            self.repo = git_package.Repo(path, search_parent_directories=True)
+            self.repo = git.Repo(path, search_parent_directories=True)
             self.git_version = self.repo.git.version_info
 
             if self.git_version >= MIN_GIT_VERSION:
@@ -69,8 +70,8 @@ class GitRepo:
         )
 
     @property
-    def tracking_branch(self):
-        if not self.is_valid():
+    def tracking_branch(self) -> RemoteReference | None:
+        if self.repo is None or not self.is_valid():
             return None
 
         if self.is_head_detached:
@@ -79,41 +80,45 @@ class GitRepo:
         return self.repo.active_branch.tracking_branch()
 
     @property
-    def untracked_files(self):
-        if not self.is_valid():
+    def untracked_files(self) -> list[str] | None:
+        if self.repo is None or not self.is_valid():
             return None
 
         return self.repo.untracked_files
 
     @property
-    def is_head_detached(self):
-        if not self.is_valid():
+    def is_head_detached(self) -> bool:
+        if self.repo is None or not self.is_valid():
             return False
 
         return self.repo.head.is_detached
 
     @property
-    def uncommitted_files(self):
-        if not self.is_valid():
+    def uncommitted_files(self) -> list[str] | None:
+        if self.repo is None or not self.is_valid():
             return None
 
-        return [item.a_path for item in self.repo.index.diff(None)]
+        return [cast("str", item.a_path) for item in self.repo.index.diff(None)]
 
     @property
-    def ahead_commits(self):
-        if not self.is_valid():
+    def ahead_commits(self) -> list[Commit] | None:
+        if self.repo is None or not self.is_valid():
             return None
 
         try:
-            remote, branch_name = self.get_tracking_branch_remote()
+            tracking_branch_info = self.get_tracking_branch_remote()
+            if tracking_branch_info is None:
+                return None
+
+            remote, branch_name = tracking_branch_info
             remote_branch = f"{remote.name}/{branch_name}"
 
             return list(self.repo.iter_commits(f"{remote_branch}..{branch_name}"))
         except Exception:
             return []
 
-    def get_tracking_branch_remote(self):
-        if not self.is_valid():
+    def get_tracking_branch_remote(self) -> tuple[Remote, str] | None:
+        if self.repo is None or not self.is_valid():
             return None
 
         tracking_branch = self.tracking_branch
@@ -145,7 +150,7 @@ class GitRepo:
 
         return False
 
-    def get_repo_info(self):
+    def get_repo_info(self) -> tuple[str, str, str] | None:
         if not self.is_valid():
             return None
 

streamlit/hello/animation_demo.py

@@ -42,22 +42,22 @@ def animation_demo() -> None:
     for frame_num, a in enumerate(np.linspace(0.0, 4 * np.pi, 100)):
         # Here were setting value for these two elements.
         progress_bar.progress(frame_num)
-        frame_text.text("Frame %i/100" % (frame_num + 1))
+        frame_text.text(f"Frame {frame_num + 1}/100")
 
         # Performing some fractal wizardry.
         c = separation * np.exp(1j * a)
-        Z = np.tile(x, (n, 1)) + 1j * np.tile(y, (1, m))
-        C = np.full((n, m), c)
-        M: Any = np.full((n, m), True, dtype=bool)
-        N = np.zeros((n, m))
+        z = np.tile(x, (n, 1)) + 1j * np.tile(y, (1, m))
+        c_matrix = np.full((n, m), c)
+        m_matrix: Any = np.full((n, m), True, dtype=bool)
+        n_matrix = np.zeros((n, m))
 
         for i in range(iterations):
-            Z[M] = Z[M] * Z[M] + C[M]
-            M[np.abs(Z) > 2] = False
-            N[M] = i
+            z[m_matrix] = z[m_matrix] * z[m_matrix] + c_matrix[m_matrix]
+            m_matrix[np.abs(z) > 2] = False
+            n_matrix[m_matrix] = i
 
         # Update the image placeholder by calling the image() function on it.
-        image.image(1.0 - (N / N.max()), use_container_width=True)
+        image.image(1.0 - (n_matrix / n_matrix.max()), use_container_width=True)
 
     # We clear elements by calling empty on them.
     progress_bar.empty()

streamlit/hello/dataframe_demo.py

@@ -21,15 +21,15 @@ import streamlit as st
 from streamlit.hello.utils import show_code
 
 
-def data_frame_demo():
+def data_frame_demo() -> None:
     @st.cache_data
-    def get_UN_data():
-        AWS_BUCKET_URL = "https://streamlit-demo-data.s3-us-west-2.amazonaws.com"
-        df = pd.read_csv(AWS_BUCKET_URL + "/agri.csv.gz")
+    def get_un_data() -> pd.DataFrame:
+        aws_bucket_url = "https://streamlit-demo-data.s3-us-west-2.amazonaws.com"
+        df = pd.read_csv(aws_bucket_url + "/agri.csv.gz")
         return df.set_index("Region")
 
     try:
-        df = get_UN_data()
+        df = get_un_data()
         countries = st.multiselect(
             "Choose countries", list(df.index), ["China", "United States of America"]
         )

streamlit/hello/hello.py

@@ -20,6 +20,7 @@ st.write(
     """
     Streamlit is an open-source app framework built specifically for
     machine learning and data science projects.
+
     **👈 Select a demo from the sidebar** to see some examples
     of what Streamlit can do!