"""AI Audio Detection logic."""

from typing import Any

from assets.models import hive_segment as hive_segment_model, ows_track


def get_ai_generated_audio_results_by_product(product_id: int) -> list[dict[str, Any]]:
    """Get AI generated audio results for product."""
    tracks_result = ows_track.get_tracks_by_product_id(product_id)
    tracks = tracks_result["items"]

    if not tracks:
        return []

    track_ids = [track["tuid"] for track in tracks]

    ai_audio_results = hive_segment_model.get_ai_generated_audio_results_by_product(
        product_id, track_ids
    )
    return ai_audio_results