|
""" |
|
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", |
|
} |
|
|