utils_display.py

import json 

with open("model_info.json", "r") as f:
    model_info = json.load(f)

def make_clickable_model(model_name):
    global model_info
    modified_model_name = model_name

    if model_name in model_info:
        info = model_info[model_name]
        pretty_name = info.get("pretty_name", model_name)
        hf_id = info.get("hf_model_id", False)

        if hf_id:
            link = f"https://huggingface.co/{hf_id}"
            modified_model_name = f'<a target="_blank" href="{link}" style="color: var(--link-text-color); text-decoration: underline; text-decoration-style: dotted;">{pretty_name}</a>'
        else:
            modified_model_name = f'{pretty_name}'

    return modified_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>"