diff --git a/src/diffmage/evaluation/models.py b/src/diffmage/evaluation/models.py index 418f3cc..eba2f4a 100644 --- a/src/diffmage/evaluation/models.py +++ b/src/diffmage/evaluation/models.py @@ -43,9 +43,11 @@ def get_quality_level(score: float) -> str: @staticmethod def get_rating_color(score: float) -> str: """Color for the quality rating""" - if score > ScoreThresholds.EXCELLENT: + if score >= ScoreThresholds.EXCELLENT: return "green" elif score >= ScoreThresholds.GOOD: + return "blue" + elif score >= ScoreThresholds.AVERAGE: return "yellow" else: return "red"