"""Module to save and retrieve Hive Text Recognition data."""

from typing import Any

from assets.models import (
    asset_final as asset_final_model,
    hive_text_recognition as hive_text_recognition_model,
)


def save_hive_text_recognition(payload: dict[str, Any]) -> None:
    """Save hive text recognition data.

    Args:
        payload: POST body containing asset_final_id and block_text.
    """
    asset_final_model.get_asset_final_by_id(payload["asset_final_id"])
    hive_text_recognition_model.create_hive_text_recognition(
        payload["asset_final_id"], payload["block_text"]
    )


def get_hive_text_recognition_by_product(
    product_id: int,
) -> dict[str, Any] | None:
    """Get hive text recognition result for a product.

    Args:
        product_id (int): Product id.

    Returns:
        dict: Dict with asset_final_id and block_text, or None if not found.
    """
    return hive_text_recognition_model.get_hive_text_recognition_by_product(product_id)