import json
import pandas as pd
import plotly.express as px

CO2_UNIT = "MtCO₂"

with open("data/method_groups.json") as _f:
    method_groups: dict[str, list[str]] = json.load(_f)

csv_taxonomy = pd.read_csv("data/inputs_aggregated.csv")

# resource_groups: group → [resources], order follows first appearance in CSV
resource_groups: dict[str, list[str]] = {}
for _, row in csv_taxonomy.drop_duplicates(subset=["Resource group", "Resource"]).iterrows():
    group, resource = row["Resource group"], row["Resource"]
    resource_groups.setdefault(group, [])
    if resource not in resource_groups[group]:
        resource_groups[group].append(resource)

# resource_unit_map: group → base unit (e.g. "Mha/MtCO2" → "Mha")
resource_unit_map: dict[str, str] = {}
for _, row in csv_taxonomy.drop_duplicates(subset=["Resource group"]).iterrows():
    resource_unit_map[row["Resource group"]] = str(row["Unit"]).split("/")[0]

resource_order = sum(resource_groups.values(), [])
method_order = sum(method_groups.values(), [])

# Resources whose name contains "other" are hidden from the standard display
# (tabs 1 and 3) but remain available as reference groups in the "Add resource"
# dialog so their coefficients can still be inherited by custom resources.
hidden_resource_names: set[str] = {
    r for r in resource_order
    if "other" in r.lower()
}

try:
    with open("data/secondary_resources.json") as _f:
        secondary_resources: set[tuple[str, str]] = {tuple(p) for p in json.load(_f)}
except FileNotFoundError:
    secondary_resources: set[tuple[str, str]] = set()

# Tooltips are loaded from data/tooltips.xlsx (two sheets: "Resources" and "Methods").
# To add or edit a tooltip, open that file — no code change needed.
# If the file is missing, tooltips are silently disabled (no hover text shown).
try:
    tooltips_xl = pd.read_excel("data/tooltips.xlsx", sheet_name=None)
    df_res = tooltips_xl.get("Resources", pd.DataFrame(columns=["Resource", "Tooltip"]))
    df_met = tooltips_xl.get("Methods",   pd.DataFrame(columns=["Method",   "Tooltip"]))
    resource_tooltips: dict[str, str] = {
        row["Resource"]: row["Tooltip"]
        for _, row in df_res.iterrows()
        if pd.notna(row.get("Tooltip")) and str(row.get("Tooltip", "")).strip()
    }
    method_tooltips: dict[str, str] = {
        row["Method"]: row["Tooltip"]
        for _, row in df_met.iterrows()
        if pd.notna(row.get("Tooltip")) and str(row.get("Tooltip", "")).strip()
    }
except FileNotFoundError:
    resource_tooltips: dict[str, str] = {}
    method_tooltips:   dict[str, str] = {}

CRRA_COLORS = {
    "off_white": "#eae9d8",
    "red": "#ff5f4d",
    "green_dark": "#298231",
    "green": "#55c95b",
    "green_light": "#92ef9b",
    "yellow": "#d9e210",
    "white": "#ffffff",
    "brown_grey": "#665d5b",
    "grey": "#a0a0a0",
    "black": "#000000",
}

CRRA_FONT = "Segoe UI, sans-serif"

PLOTLY_COLOR_SEQUENCE = [
    CRRA_COLORS["red"],
    CRRA_COLORS["green"],
    CRRA_COLORS["yellow"],
    CRRA_COLORS["green_dark"],
    CRRA_COLORS["brown_grey"],
    CRRA_COLORS["green_light"],
    CRRA_COLORS["grey"],
]

PLOTLY_HEATMAP_SCALE = [
    [0.0, CRRA_COLORS["white"]],
    [0.5, CRRA_COLORS["green_light"]],
    [1.0, CRRA_COLORS["green_dark"]],
]

PLOTLY_TEMPLATE = "plotly_white"