EidZddlmZddlmZmZddlmZddlm Z erddl m Z m Z ddl mZddZdd Z d dd Zdd Zdd ZddZddZy )ztData marshalling utilities for ArrowTable protobufs, which are used by CustomComponent for dataframe serialization. ) annotations) TYPE_CHECKINGAny)dataframe_util)pandas_styler_utils) DataFrameIndex) ArrowTablec<t|tr t|S|S)z;Convert a tuple to a list. Leave as is if it's not a tuple.) isinstancetuplelist)items i/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/streamlit/components/v1/component_arrow.py_maybe_tuple_to_listr s#D%04::d:cXtj|tjd S)aZConvert a DataFrame to Arrow IPC bytes for custom component v1. Uses ``convert_pandas_df_to_arrow_bytes`` with large-type downcasting enabled for pandas >= 3.0, where ``StringDtype`` is backed by ``large_string`` in Arrow. Third-party custom components bundling older Arrow JS libraries cannot decode these large type codes. z3.0.0)downcast_large_types)r convert_pandas_df_to_arrow_bytesis_pandas_version_less_than)dfs r$_convert_df_to_component_arrow_bytesr%s+  : : !/!K!KG!TT rNctj|rtj|||tj|}t ||j t||jt||y)a:Marshall data into an ArrowTable proto. Parameters ---------- proto : proto.ArrowTable Output. The protobuf for a Streamlit ArrowTable proto. data : pandas.DataFrame, pandas.Styler, numpy.ndarray, Iterable, dict, or None Something that is or can be converted to a dataframe. N) ris_pandas_stylerrmarshall_stylerconvert_anything_to_pandas_df_marshall_indexindex_marshall_columnscolumns_marshall_data)protodata default_uuidrs rmarshallr%3sZ&&t,++E4F  5 5d ;BE288$eRZZ(5"rcddl}ttt|j}|j |}t ||_y)aHMarshall pandas.DataFrame index into an ArrowTable proto. Parameters ---------- proto : proto.ArrowTable Output. The protobuf for a Streamlit ArrowTable proto. index : pd.Index Index to use for resulting frame. Will default to RangeIndex (0, 1, 2, ..., n) if no index is provided. rN)pandasrmaprvaluesrrr)r"rpd index_valuesindex_dfs rrrJs70%,,?@L||L)H6x@EKrcddl}ttt|j}|j |}t ||_y)aZMarshall pandas.DataFrame columns into an ArrowTable proto. Parameters ---------- proto : proto.ArrowTable Output. The protobuf for a Streamlit ArrowTable proto. columns : Index Column labels to use for resulting frame. Will default to RangeIndex (0, 1, 2, ..., n) if no column labels are provided. rN)r'rr(rr)rrr )r"r r* column_values columns_dfs rrr^s717>>BCMm,J8DEMrc$t||_y)zMarshall pandas.DataFrame data into an ArrowTable proto. Parameters ---------- proto : proto.ArrowTable Output. The protobuf for a Streamlit ArrowTable proto. df : pandas.DataFrame A dataframe to marshall. N)rr#)r"rs rr!r!rs6b9EJrctjdr tdddl}tj|j }tj|j }tj|j}|j|j|jjj|jjjS)zConvert ArrowTable proto to pandas.DataFrame. Parameters ---------- proto : proto.ArrowTable Output. pandas.DataFrame z14.0.1zThe installed pyarrow version is not compatible with this component. Please upgrade to 14.0.1 or higher: pip install -U pyarrowrN)rr ) ris_pyarrow_version_less_than RuntimeErrorr' convert_arrow_bytes_to_pandas_dfr#rr rto_numpyTtolist)r"r*r#rr s rarrow_proto_to_dataframer8s228< I    : :5:: FD  ; ;EKK HE==emmLG << nn  '')  "$$++-  r)rrreturnr)rrr9bytes)N)r"ArrowTableProtor#rr$z str | Noner9None)r"r;r Index[Any]r9r<)r"r;r r=r9r<)r"r;rrr9r<)r"r;r9r)__doc__ __future__rtypingrr streamlitrstreamlit.elements.librr'rr streamlit.proto.Components_pb2r r;rrr%rrr!r8rrrEsm#%$6'L; CG "%5? .A(E( :r