"""
DataFrame utility functions for handling serialized DataFrames.

This module provides helper functions to work with DataFrames that may
come from Temporal serialization (already as DataFrame) or from legacy
code (as dict).
"""

from typing import Any

from pandas import DataFrame


def ensure_dataframe(data: Any) -> DataFrame:
    """
    Ensure that data is a DataFrame, converting from dict if necessary.

    This function handles both cases:
    - Data already deserialized as DataFrame (from Temporal codec)
    - Data as dict (legacy format or non-DataFrame serialization)

    Args:
        data: Data that should be a DataFrame (can be DataFrame or dict)

    Returns:
        DataFrame: The data as a pandas DataFrame
    """
    if isinstance(data, DataFrame):
        return data
    return DataFrame(data)