o ,&]i$@sddlmZddlZddlmZddlmZmZmZddl m Z ddl m Z m Z ddlmZddlmZdd lmZdd lmZmZerJdd lmZeeZ d"d#ddZd$ddZd%ddZd&ddZeGd d!d!Z dS)') annotationsN) dataclass) TYPE_CHECKINGAnycast) ARROW_REF_KEY)convert_anything_to_arrow_bytesis_dataframe_like) get_logger) BidiComponent) MixedData)AttributeDictionarycalc_md5)BidiComponentStatedatadict[str, Any] arrow_blobsdict[str, bytes] | Nonereturnc Cs|duri}i}|D]>\}}t|rFzt|}t|}|||<t|i||<Wq tyE}ztd|||||<WYd}~q d}~ww|||<q |S)aExtract dataframe-like objects from a dictionary and replace them with placeholders. This function traverses the first level of a dictionary, detects any dataframe-like objects, stores their Arrow bytes representation in the `arrow_blobs` dictionary, and replaces them with JSON-serializable placeholder objects. Parameters ---------- data The dictionary to process. Only the first level is checked for dataframe-like objects. arrow_blobs The dictionary to store the extracted Arrow bytes in, keyed by a unique reference ID. Returns ------- dict[str, Any] A new dictionary with dataframe-like objects replaced by placeholders. NzAArrow serialization failed for key %r, keeping original value: %s)itemsr rrr Exception_LOGGERdebug)rrprocessed_datakeyvalue arrow_bytesref_iderr/var/www/html/IGF-ODF-V3/venv/lib/python3.10/site-packages/streamlit/components/v2/bidi_component/serialization.py_extract_dataframes_from_dict"s* r!rbidi_component_protoBidiComponentProtoNonecCsi}t|tr t||}n|}|rEt}zt||_Wnty-tt||_Ynw|D] \}}||j |_ q2|j |dSz t||_WdSty`tt||_YdSw)a4Serialize mixed data with automatic dataframe detection into a protobuf message. This function detects dataframe-like objects in the first level of a dictionary, extracts them into separate Arrow blobs, and populates a `MixedDataProto` protobuf message for efficient serialization. Parameters ---------- data The data structure to serialize. If it is a dictionary, its first level will be scanned for dataframe-like objects. bidi_component_proto The protobuf message to populate with the serialized data. N) isinstancedictr!MixedDataProtojsondumps TypeErrorstrrrrmixedCopyFrom)rr"rr mixed_protorrrrr serialize_mixed_data[s&    r/s str | NonecCs2|durdSzt|WStjy|YSw)a;Deserialize a JSON string, returning the string itself if it's not valid JSON. Parameters ---------- s The string to deserialize. Returns ------- Any The deserialized JSON object, or the original string if parsing fails. Returns `None` if the input is `None`. N)r(loadsJSONDecodeError)r0rrr handle_deserializes r4list[Any] | NonecCs(t|}|dur dSt|tr|S|gS)aDeserialize trigger aggregator payloads as a list. For bidirectional components, the frontend always sends a JSON array of payload objects. This deserializer normalizes older or singular payloads into a list while preserving ``None`` for cleared values. Parameters ---------- s The JSON string to deserialize, hopefully representing a list of payloads. Returns ------- list[Any] or None A list of payloads, or `None` if the input was `None`. N)r4r%list)r0rrrr deserialize_trigger_lists  r7c@s2eZdZUdZdZded<dd d ZdddZdS)BidiComponentSerdeaSerialization and deserialization logic for a bidirectional component. This class handles the conversion of component state between the frontend (JSON strings) and the backend (Python objects). The canonical shape is a flat mapping of state keys to values. Parameters ---------- default A dictionary of default values to be applied to the state when deserializing, if the corresponding keys are not already present. Nzdict[str, Any] | Nonedefaultui_valuestr | dict[str, Any] | Nonerrc Cst|tr|}n7t|tr=zt|}t|tr|ni}Wn!tjtfy<}ztjd||di}WYd}~nd}~wwi}|j durV|j D] \}}||vrU|||<qIt dt |}|S)a_Deserialize the component's state from a frontend value. Parameters ---------- ui_value The value received from the frontend, which can be a JSON string, a dictionary, or `None`. Returns ------- BidiComponentState The deserialized state as a flat mapping. z7Failed to deserialize component state from frontend: %s)exc_infoNr) r%r&r+r(r2r3r*rwarningr9rrr )selfr:deserialized_valueparsedr default_key default_valuestaterrr deserializes2    zBidiComponentSerde.deserializerrr+cCs t|S)aSerialize the component's state into a JSON string for the frontend. Parameters ---------- value The component state to serialize. Returns ------- str A JSON string representation of the value. )r(r))r>rrrr serializes zBidiComponentSerde.serialize)r:r;rr)rrrr+)__name__ __module__ __qualname____doc__r9__annotations__rDrErrrr r8s   0r8)N)rrrrrr)rrr"r#rr$)r0r1rr)r0r1rr5)! __future__rr( dataclassesrtypingrrr0streamlit.components.v2.bidi_component.constantsrstreamlit.dataframe_utilrr streamlit.loggerr !streamlit.proto.BidiComponent_pb2r r#r r'streamlit.utilr r,streamlit.components.v2.bidi_component.staterrFrr!r/r4r7r8rrrr s(        9 3