PRIMARY_COLOR = "#AC482A"
DARK_BACKGROUND = "#082030"
DATASET_ID = "mmpr/open_model_evolution_data"
HF_PARQUET_URL_1 = "https://huggingface.co/datasets/mmpr/open_model_evolution_data/resolve/main/all_downloads_with_annotations.parquet"
HF_PARQUET_URL_2 = "https://huggingface.co/datasets/mmpr/open_model_evolution_data/resolve/main/one_year_rolling.parquet"

BUTTON_STYLE = {
    "display": "inline-block",
    "marginBottom": "10px",
    "marginRight": "15px",
    "marginTop": "30px",
    "padding": "6px 16px",
    "backgroundColor": DARK_BACKGROUND,
    "color": "white",
    "borderRadius": "6px",
    "textDecoration": "none",
    "fontWeight": "bold",
    "fontSize": "14px",
}

COMPANY_ICON_MAP = {
    "google": "../assets/icons/google.png",
    "distilbert": "../assets/images/hf.svg",
    "sentence-transformers": "../assets/images/hf.svg",
    "facebook": "../assets/icons/meta.png",
    "openai": "../assets/icons/openai.png",
    "amazon": "../assets/icons/amazon.png",
    "microsoft": "../assets/icons/microsoft.png",
}

COUNTRY_EMOJI_FALLBACK = {
    "User": "👤",
    "Organization": "🏢",
    "Model": "📦",
}

META_COLS_MAP = {
    "org_country_single": ["org_country_single", "total_downloads"],
    "author": [
        "org_country_single",
        "author",
        "total_downloads",
    ],
    "derived_author": [
        "org_country_single",
        "derived_author",
        "total_downloads",
    ],
    "model": [
        "org_country_single",
        "author",
        "derived_author",
        "merged_modality",
        "total_downloads",
    ],
}