src/utils.py

import re

def get_metric_choices():
    """Return list of metric choices with emoji markers."""
    return [
        "Average ⭐",
        "Image Quality 📊",
        "Brightness Consistency 📊",
        "Color Temperature 📊",
        "Sharpness Retention 📊",
        "Motion Smoothness 📊",
        "Trajectory Accuracy 📊",
        "Memory Symmetry 📊",
        "Trajectory Alignment 📊",
    ]

def clean_metric_names(metrics):
    """Remove emoji markers from metric names."""
    cleaned = []
    for m in metrics:
        clean = m.replace(" ⭐", "").replace(" 📊", "").strip()
        cleaned.append(clean)
    return cleaned

def get_dimension_metrics():
    """Return mapping from dimension to list of metrics (for radar chart)."""
    return {
        "Generation Quality": ["Image Quality", "Brightness Consistency", "Color Temperature", "Sharpness Retention"],
        "Trajectory Following": ["Motion Smoothness", "Trajectory Accuracy"],
        "Memory Ability": ["Memory Symmetry", "Trajectory Alignment"]
    }