ranking.py

"""
ranking.py
Helper functions for IMDB-based scoring of candidates.
"""

from typing import Dict, Any


def score_candidate(
    imdb_rating: float,
    genre_match_score: float = 0.0,
    disliked_genre_penalty: float = 0.0,
    runtime_match_score: float = 0.0,
) -> float:
    """
    Compute a simple numeric score for a candidate.

    You can call this from a custom tool if you want to do ranking
    programmatically, or just keep it as documentation for your scoring scheme.
    """
    # Basic weighting (you can tweak these multipliers)
    score = 0.0
    score += imdb_rating * 2.0
    score += genre_match_score * 1.5
    score -= disliked_genre_penalty * 2.0
    score += runtime_match_score * 1.0
    return score