src/display/formatting.py

def model_hyperlink(link, model_name):
    return f'<a target="_blank" href="{link}" style="color: var(--link-text-color); text-decoration: underline;text-decoration-style: dotted;">{model_name}</a>'


def make_clickable_model(model_name):
    # Custom link mappings for each model
    custom_links = {
        "ReSeek-Qwen2.5-7b-Instruct": "https://your-custom-link.com/reseek-7b",
        "ReSeek-Qwen2.5-3b-Instruct": "https://your-custom-link.com/reseek-3b",
        "ZeroSearch-Qwen2.5-3b-Instruct": "https://huggingface.co/Alibaba-NLP/ZeroSearch_wiki_V2_Qwen2.5_3B_Instruct",
        "ZeroSearch-Qwen2.5-7b-Instruct": "https://huggingface.co/Alibaba-NLP/ZeroSearch_wiki_V2_Qwen2.5_7B_Instruct",
        "Search-R1-Qwen2.5-7b-Instruct": "https://huggingface.co/PeterJinGo/SearchR1-nq_hotpotqa_train-qwen2.5-7b-it-em-ppo",
        "Search-R1-Qwen2.5-3b-Instruct": "https://huggingface.co/PeterJinGo/SearchR1-nq_hotpotqa_train-qwen2.5-3b-em-grpo",
        "Search-o1-Qwen2.5-7b-Instruct": "https://github.com/RUC-NLPIR/Search-o1",
        "RAG-Qwen2.5-7b-Instruct": "",
        "R1-Qwen2.5-7b-Instruct": "",
        "SFT-Qwen2.5-7b-Instruct": "",
        "CoT-Qwen2.5-7b-Instruct": "",
        "Direct-Inference-Qwen2.5-7b-Instruct": "",
    }

    if model_name in custom_links:
        link = custom_links[model_name]
        return model_hyperlink(link, model_name)
    else:
        # If no custom link, just return the model name
        return model_name


def styled_error(error):
    return f"<p style='color: red; font-size: 20px; text-align: center;'>{error}</p>"


def styled_warning(warn):
    return f"<p style='color: orange; font-size: 20px; text-align: center;'>{warn}</p>"


def styled_message(message):
    return f"<p style='color: green; font-size: 20px; text-align: center;'>{message}</p>"


def has_no_nan_values(df, columns):
    return df[columns].notna().all(axis=1)


def has_nan_values(df, columns):
    return df[columns].isna().any(axis=1)