streamlit-nightly 1.37.2.dev20240815__tar.gz → 1.37.2.dev20240816__tar.gz

{streamlit_nightly-1.37.2.dev20240815/streamlit_nightly.egg-info → streamlit_nightly-1.37.2.dev20240816}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: streamlit-nightly
-Version: 1.37.2.dev20240815
+Version: 1.37.2.dev20240816
 Summary: A faster way to build and share data apps
 Home-page: https://streamlit.io
 Author: Snowflake Inc

{streamlit_nightly-1.37.2.dev20240815 → streamlit_nightly-1.37.2.dev20240816}/setup.py

@@ -21,7 +21,7 @@ from setuptools.command.install import install
 
 THIS_DIRECTORY = Path(__file__).parent
 
-VERSION = "1.37.2.dev20240815"  # PEP-440
+VERSION = "1.37.2.dev20240816"  # PEP-440
 
 # IMPORTANT: We should try very hard *not* to add dependencies to Streamlit.
 # And if you do add one, make the required version as general as possible:

{streamlit_nightly-1.37.2.dev20240815 → streamlit_nightly-1.37.2.dev20240816}/streamlit/components/v1/component_arrow.py

@@ -20,7 +20,7 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, Any
 
-from streamlit import dataframe_util, type_util
+from streamlit import dataframe_util
 from streamlit.elements.lib import pandas_styler_utils
 
 if TYPE_CHECKING:
@@ -122,7 +122,7 @@ def arrow_proto_to_dataframe(proto: ArrowTableProto) -> DataFrame:
 
     """
 
-    if type_util.is_pyarrow_version_less_than("14.0.1"):
+    if dataframe_util.is_pyarrow_version_less_than("14.0.1"):
         raise RuntimeError(
             "The installed pyarrow version is not compatible with this component. "
             "Please upgrade to 14.0.1 or higher: pip install -U pyarrow"

{streamlit_nightly-1.37.2.dev20240815 → streamlit_nightly-1.37.2.dev20240816}/streamlit/dataframe_util.py

@@ -22,16 +22,16 @@ import inspect
 import math
 import re
 from collections import ChainMap, UserDict, UserList, deque
-from collections.abc import ItemsView, Mapping
+from collections.abc import ItemsView
 from enum import Enum, EnumMeta, auto
 from types import MappingProxyType
 from typing import (
     TYPE_CHECKING,
     Any,
-    Dict,
     Final,
     Iterable,
     List,
+    Mapping,
     Protocol,
     Sequence,
     TypeVar,
@@ -44,13 +44,16 @@ from typing_extensions import TypeAlias, TypeGuard
 
 from streamlit import config, errors, logger, string_util
 from streamlit.type_util import (
+    CustomDict,
     NumpyShape,
     has_callable_attr,
     is_custom_dict,
     is_dataclass_instance,
     is_list_like,
     is_namedtuple,
+    is_pydantic_model,
     is_type,
+    is_version_less_than,
 )
 
 if TYPE_CHECKING:
@@ -144,9 +147,26 @@ class DataFrameGenericAlias(Protocol[V_co]):
     def iloc(self) -> _iLocIndexer: ...
 
 
+class PandasCompatible(Protocol):
+    """Protocol for Pandas compatible objects that have a `to_pandas` method."""
+
+    def to_pandas(self) -> DataFrame | Series: ...
+
+
+class DataframeInterchangeCompatible(Protocol):
+    """Protocol for objects support the dataframe-interchange protocol.
+
+    https://data-apis.org/dataframe-protocol/latest/index.html
+    """
+
+    def __dataframe__(self, allow_copy: bool) -> Any: ...
+
+
 OptionSequence: TypeAlias = Union[
     Iterable[V_co],
     DataFrameGenericAlias[V_co],
+    PandasCompatible,
+    DataframeInterchangeCompatible,
 ]
 
 # Various data types supported by our dataframe processing
@@ -158,10 +178,14 @@ Data: TypeAlias = Union[
     "Styler",
     "Index",
     "pa.Table",
+    "pa.Array",
     "np.ndarray[Any, np.dtype[Any]]",
     Iterable[Any],
-    Dict[Any, Any],
+    "Mapping[Any, Any]",
     DBAPICursor,
+    PandasCompatible,
+    DataframeInterchangeCompatible,
+    CustomDict,
     None,
 ]
 
@@ -204,6 +228,53 @@ class DataFormat(Enum):
     DUCKDB_RELATION = auto()  # DuckDB Relation
 
 
+def is_pyarrow_version_less_than(v: str) -> bool:
+    """Return True if the current Pyarrow version is less than the input version.
+
+    Parameters
+    ----------
+    v : str
+        Version string, e.g. "0.25.0"
+
+    Returns
+    -------
+    bool
+
+
+    Raises
+    ------
+    InvalidVersion
+        If the version strings are not valid.
+
+    """
+    import pyarrow as pa
+
+    return is_version_less_than(pa.__version__, v)
+
+
+def is_pandas_version_less_than(v: str) -> bool:
+    """Return True if the current Pandas version is less than the input version.
+
+    Parameters
+    ----------
+    v : str
+        Version string, e.g. "0.25.0"
+
+    Returns
+    -------
+    bool
+
+
+    Raises
+    ------
+    InvalidVersion
+        If the version strings are not valid.
+    """
+    import pandas as pd
+
+    return is_version_less_than(pd.__version__, v)
+
+
 def is_dataframe_like(obj: object) -> bool:
     """True if the object is a dataframe-like object.
 
@@ -298,11 +369,7 @@ def is_snowpark_row_list(obj: object) -> bool:
 
 def is_pyspark_data_object(obj: object) -> bool:
     """True if obj is of type pyspark.sql.dataframe.DataFrame"""
-    return (
-        is_type(obj, _PYSPARK_DF_TYPE_STR)
-        and hasattr(obj, "toPandas")
-        and callable(obj.toPandas)
-    )
+    return is_type(obj, _PYSPARK_DF_TYPE_STR) and has_callable_attr(obj, "toPandas")
 
 
 def is_dask_object(obj: object) -> bool:
@@ -461,7 +528,7 @@ def convert_anything_to_pandas_df(
 
     Parameters
     ----------
-    data : any
+    data : dataframe-, array-, or collections-like object
         The data to convert to a Pandas DataFrame.
 
     max_unevaluated_rows: int
@@ -623,6 +690,16 @@ def convert_anything_to_pandas_df(
     if has_callable_attr(data, "to_pandas"):
         return pd.DataFrame(data.to_pandas())
 
+    # Check for dataframe interchange protocol
+    # Only available in pandas >= 1.5.0
+    # https://pandas.pydata.org/docs/whatsnew/v1.5.0.html#dataframe-interchange-protocol-implementation
+    if (
+        has_callable_attr(data, "__dataframe__")
+        and is_pandas_version_less_than("1.5.0") is False
+    ):
+        data_df = pd.api.interchange.from_dataframe(data)
+        return data_df.copy() if ensure_copy else data_df
+
     # Support for generator functions
     if inspect.isgeneratorfunction(data):
         data = _fix_column_naming(
@@ -657,7 +734,9 @@ def convert_anything_to_pandas_df(
         return _dict_to_pandas_df(dataclasses.asdict(data))
 
     # Support for dict-like objects
-    if isinstance(data, (ChainMap, MappingProxyType, UserDict)):
+    if isinstance(data, (ChainMap, MappingProxyType, UserDict)) or is_pydantic_model(
+        data
+    ):
         return _dict_to_pandas_df(dict(data))
 
     # Try to convert to pandas.DataFrame. This will raise an error is df is not
@@ -787,7 +866,7 @@ def convert_anything_to_arrow_bytes(
 
     Parameters
     ----------
-    data : any
+    data : dataframe-, array-, or collections-like object
         The data to convert to Arrow bytes.
 
     max_unevaluated_rows: int
@@ -805,19 +884,7 @@ def convert_anything_to_arrow_bytes(
     if isinstance(data, pa.Table):
         return convert_arrow_table_to_arrow_bytes(data)
 
-    if is_pandas_data_object(data):
-        # All pandas data objects should be handled via our pandas
-        # conversion logic. We are already calling it here
-        # to ensure that its not handled via the interchange
-        # protocol support below.
-        df = convert_anything_to_pandas_df(data, max_unevaluated_rows)
-        return convert_pandas_df_to_arrow_bytes(df)
-
-    if is_polars_dataframe(data):
-        return convert_arrow_table_to_arrow_bytes(data.to_arrow())
-
-    if is_polars_series(data):
-        return convert_arrow_table_to_arrow_bytes(data.to_frame().to_arrow())
+    # TODO(lukasmasuch): Add direct conversion to Arrow for supported formats here
 
     # Fallback: try to convert to pandas DataFrame
     # and then to Arrow bytes.
@@ -835,7 +902,7 @@ def convert_anything_to_sequence(obj: OptionSequence[V_co]) -> list[V_co]:
     Parameters
     ----------
 
-    obj : OptionSequence
+    obj : dataframe-, array-, or collections-like object
         The object to convert to a list.
 
     Returns
@@ -1141,6 +1208,7 @@ def determine_data_format(input_data: Any) -> DataFormat:
         or is_dataclass_instance(input_data)
         or is_namedtuple(input_data)
         or is_custom_dict(input_data)
+        or is_pydantic_model(input_data)
     ):
         return DataFormat.KEY_VALUE_DICT
     elif isinstance(input_data, (ItemsView, enumerate)):

{streamlit_nightly-1.37.2.dev20240815 → streamlit_nightly-1.37.2.dev20240816}/streamlit/elements/json.py

@@ -21,7 +21,12 @@ from typing import TYPE_CHECKING, Any, cast
 
 from streamlit.proto.Json_pb2 import Json as JsonProto
 from streamlit.runtime.metrics_util import gather_metrics
-from streamlit.type_util import is_custom_dict, is_namedtuple
+from streamlit.type_util import (
+    is_custom_dict,
+    is_list_like,
+    is_namedtuple,
+    is_pydantic_model,
+)
 
 if TYPE_CHECKING:
     from streamlit.delta_generator import DeltaGenerator
@@ -86,11 +91,13 @@ class JsonMixin:
         if is_namedtuple(body):
             body = body._asdict()
 
-        if isinstance(body, (map, enumerate)):
-            body = list(body)
+        if isinstance(
+            body, (ChainMap, types.MappingProxyType, UserDict)
+        ) or is_pydantic_model(body):
+            body = dict(body)  # type: ignore
 
-        if isinstance(body, (ChainMap, types.MappingProxyType, UserDict)):
-            body = dict(body)
+        if is_list_like(body):
+            body = list(body)
 
         if not isinstance(body, str):
             try:

{streamlit_nightly-1.37.2.dev20240815 → streamlit_nightly-1.37.2.dev20240816}/streamlit/elements/widgets/button_group.py

@@ -23,11 +23,9 @@ from typing import (
     Sequence,
     TypeVar,
     cast,
-    get_args,
+    overload,
 )
 
-from typing_extensions import TypeAlias
-
 from streamlit.elements.form_utils import current_form_id
 from streamlit.elements.lib.options_selector_utils import (
     convert_to_sequence_and_check_comparable,
@@ -80,8 +78,6 @@ _STAR_ICON: Final = ":material/star:"
 # in base64 format and send it over the wire as an image.
 _SELECTED_STAR_ICON: Final = ":material/star_filled:"
 
-_FeedbackOptions: TypeAlias = Literal["thumbs", "faces", "stars"]
-
 
 class FeedbackSerde:
     """Uses the MultiSelectSerde under-the-hood, but accepts a single index value
@@ -114,7 +110,7 @@ class FeedbackSerde:
 
 
 def get_mapped_options(
-    feedback_option: _FeedbackOptions,
+    feedback_option: Literal["thumbs", "faces", "stars"],
 ) -> tuple[list[ButtonGroupProto.Option], list[int]]:
     # options object understandable by the web app
     options: list[ButtonGroupProto.Option] = []
@@ -168,10 +164,33 @@ def _build_proto(
 
 
 class ButtonGroupMixin:
+    @overload  # These overloads are not documented in the docstring, at least not at this time, on the theory that most people won't know what it means. And the Literals here are a subclass of int anyway.
+    # Usually, we would make a type alias for Literal["thumbs", "faces", "stars"]; but, in this case, we don't use it in too many other places, and it's a more helpful autocomplete if we just enumerate the values explicitly, so a decision has been made to keep it as not an alias.
+    def feedback(
+        self,
+        options: Literal["thumbs"] = ...,
+        *,
+        key: str | None = None,
+        disabled: bool = False,
+        on_change: WidgetCallback | None = None,
+        args: Any | None = None,
+        kwargs: Any | None = None,
+    ) -> Literal[0, 1] | None: ...
+    @overload
+    def feedback(
+        self,
+        options: Literal["faces", "stars"] = ...,
+        *,
+        key: str | None = None,
+        disabled: bool = False,
+        on_change: WidgetCallback | None = None,
+        args: Any | None = None,
+        kwargs: Any | None = None,
+    ) -> Literal[0, 1, 2, 3, 4] | None: ...
     @gather_metrics("feedback")
     def feedback(
         self,
-        options: _FeedbackOptions = "thumbs",
+        options: Literal["thumbs", "faces", "stars"] = "thumbs",
         *,
         key: str | None = None,
         disabled: bool = False,
@@ -262,7 +281,7 @@ class ButtonGroupMixin:
 
         """
 
-        if not isinstance(options, list) and options not in get_args(_FeedbackOptions):
+        if options not in ["thumbs", "faces", "stars"]:
             raise StreamlitAPIException(
                 "The options argument to st.feedback must be one of "
                 "['thumbs', 'faces', 'stars']. "

{streamlit_nightly-1.37.2.dev20240815 → streamlit_nightly-1.37.2.dev20240816}/streamlit/elements/write.py

@@ -18,6 +18,7 @@ import dataclasses
 import inspect
 import types
 from collections import ChainMap, UserDict, UserList
+from collections.abc import ItemsView, KeysView, ValuesView
 from io import StringIO
 from typing import (
     TYPE_CHECKING,
@@ -258,13 +259,13 @@ class WriteMixin:
             - write(string)         : Prints the formatted Markdown string, with
                 support for LaTeX expression, emoji shortcodes, and colored text.
                 See docs for st.markdown for more.
-            - write(data_frame)     : Displays any dataframe-compatible value
-                as read-only table.
+            - write(dataframe)      : Displays any dataframe-like object in a table.
+            - write(dict)           : Displays dict-like in an interactive viewer.
+            - write(list)           : Displays list-like in an interactive viewer.
             - write(error)          : Prints an exception specially.
             - write(func)           : Displays information about a function.
             - write(module)         : Displays information about the module.
             - write(class)          : Displays information about a class.
-            - write(dict)           : Displays dict in an interactive widget.
             - write(mpl_fig)        : Displays a Matplotlib figure.
             - write(generator)      : Streams the output of a generator.
             - write(openai.Stream)  : Streams the output of an OpenAI stream.
@@ -276,8 +277,10 @@ class WriteMixin:
             - write(bokeh_fig)      : Displays a Bokeh figure.
             - write(sympy_expr)     : Prints SymPy expression using LaTeX.
             - write(htmlable)       : Prints _repr_html_() for the object if available.
+            - write(db_cursor)      : Displays DB API 2.0 cursor results in a table.
             - write(obj)            : Prints str(obj) if otherwise unknown.
 
+
         unsafe_allow_html : bool
             Whether to render HTML within ``*args``. This only applies to
             strings or objects falling back on ``_repr_html_()``. If this is
@@ -457,10 +460,14 @@ class WriteMixin:
                         UserDict,
                         ChainMap,
                         UserList,
+                        ItemsView,
+                        KeysView,
+                        ValuesView,
                     ),
                 )
                 or type_util.is_custom_dict(arg)
                 or type_util.is_namedtuple(arg)
+                or type_util.is_pydantic_model(arg)
             ):
                 flush_buffer()
                 self.dg.json(arg)
@@ -495,6 +502,14 @@ class WriteMixin:
             ):
                 # We either explicitly allow HTML or infer it's not HTML
                 self.dg.markdown(repr_html, unsafe_allow_html=unsafe_allow_html)
+            elif type_util.has_callable_attr(
+                arg, "to_pandas"
+            ) or type_util.has_callable_attr(arg, "__dataframe__"):
+                # This object can very likely be converted to a DataFrame
+                # using the to_pandas, to_arrow, or the dataframe interchange
+                # protocol.
+                flush_buffer()
+                self.dg.dataframe(arg)
             else:
                 stringified_arg = str(arg)
 

{streamlit_nightly-1.37.2.dev20240815 → streamlit_nightly-1.37.2.dev20240816}/streamlit/type_util.py

@@ -112,6 +112,11 @@ def is_type(obj: object, fqn_type_pattern: str | re.Pattern[str]) -> bool:
         return fqn_type_pattern.match(fqn_type) is not None
 
 
+def _is_type_instance(obj: object, type_to_check: str) -> bool:
+    """Check if instance of type without importing expensive modules."""
+    return type_to_check in [get_fqn(t) for t in type(obj).__mro__]
+
+
 def get_fqn(the_type: type) -> str:
     """Get module.type_name for a given type."""
     return f"{the_type.__module__}.{the_type.__qualname__}"
@@ -287,6 +292,18 @@ def is_pydeck(obj: object) -> TypeGuard[Deck]:
     return is_type(obj, "pydeck.bindings.deck.Deck")
 
 
+def is_pydantic_model(obj) -> bool:
+    """True if input looks like a Pydantic model instance."""
+
+    if isinstance(obj, type):
+        # The obj is a class, but we
+        # only want to check for instances
+        # of Pydantic models, so we return False.
+        return False
+
+    return _is_type_instance(obj, "pydantic.main.BaseModel")
+
+
 def is_custom_dict(obj: object) -> TypeGuard[CustomDict]:
     """True if input looks like one of the Streamlit custom dictionaries."""
     from streamlit.runtime.context import StreamlitCookies, StreamlitHeaders
@@ -364,53 +381,6 @@ def check_python_comparable(seq: Sequence[Any]) -> None:
         )
 
 
-def is_pandas_version_less_than(v: str) -> bool:
-    """Return True if the current Pandas version is less than the input version.
-
-    Parameters
-    ----------
-    v : str
-        Version string, e.g. "0.25.0"
-
-    Returns
-    -------
-    bool
-
-
-    Raises
-    ------
-    InvalidVersion
-        If the version strings are not valid.
-    """
-    import pandas as pd
-
-    return is_version_less_than(pd.__version__, v)
-
-
-def is_pyarrow_version_less_than(v: str) -> bool:
-    """Return True if the current Pyarrow version is less than the input version.
-
-    Parameters
-    ----------
-    v : str
-        Version string, e.g. "0.25.0"
-
-    Returns
-    -------
-    bool
-
-
-    Raises
-    ------
-    InvalidVersion
-        If the version strings are not valid.
-
-    """
-    import pyarrow as pa
-
-    return is_version_less_than(pa.__version__, v)
-
-
 def is_altair_version_less_than(v: str) -> bool:
     """Return True if the current Altair version is less than the input version.
 

{streamlit_nightly-1.37.2.dev20240815 → streamlit_nightly-1.37.2.dev20240816/streamlit_nightly.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: streamlit-nightly
-Version: 1.37.2.dev20240815
+Version: 1.37.2.dev20240816
 Summary: A faster way to build and share data apps
 Home-page: https://streamlit.io
 Author: Snowflake Inc