model/config.py

"""
File: config.py
Author: Elena Ryumina and Dmitry Ryumin
Description: Configuration file.
License: MIT License
"""

import toml
from typing import Dict
from types import SimpleNamespace


def flatten_dict(prefix: str, d: Dict) -> Dict:
    result = {}

    for k, v in d.items():
        if isinstance(v, dict):
            result.update(flatten_dict(f"{prefix}{k}_", v))
        else:
            result[f"{prefix}{k}"] = v

    return result


config = toml.load("config.toml")

config_data = flatten_dict("", config)

config_data = SimpleNamespace(**config_data)

DICT_EMO = {
    0: "Neutral",
    1: "Happiness",
    2: "Sadness",
    3: "Surprise",
    4: "Fear",
    5: "Disgust",
    6: "Anger",
}

COLORS = {
    0: 'blue', 
    1: 'orange', 
    2: 'green', 
    3: 'red', 
    4: 'purple', 
    5: 'brown', 
    6: 'pink'
}