"""
Task name mapping utilities for displaying user-friendly task names.
"""

# Mapping from original task names to display names
task_display_names = {
    "Perception Temporal Action Loc": "Temporal Action Localisation",
    "Llm Merging": "LLM Merging",
    "Meta Learning": "Meta Learning",
    "Product Recommendation": "Next Product Recommendation",
    "Weather Forcast": "Rainfall Prediction",
    "Machine Unlearning": "Machine Unlearning",
    "Backdoor Trigger Recovery": "Backdoor Trigger Recovery"
}

def get_display_name(task_name):
    """
    Get the display name for a task
    
    Args:
        task_name (str): The original task name
        
    Returns:
        str: The display name for the task
    """
    return task_display_names.get(task_name, task_name)

def get_original_name(display_name):
    """
    Get the original task name for a display name
    
    Args:
        display_name (str): The display name
        
    Returns:
        str: The original task name
    """
    # Create a reverse mapping
    reverse_mapping = {v: k for k, v in task_display_names.items()}
    return reverse_mapping.get(display_name, display_name)