"""
Configuration Module

This module provides configuration settings for the Mars Toolkit.
It includes API keys, endpoints, paths, and other configuration parameters.
"""

from typing import Dict, Any

class Config:

    @classmethod
    def as_dict(cls) -> Dict[str, Any]:
        """Return all configuration settings as a dictionary"""
        return {
            key: value for key, value in cls.__dict__.items()
            if not key.startswith('__') and not callable(value)
        }

    @classmethod
    def update(cls, **kwargs):
        """Update configuration settings"""
        for key, value in kwargs.items():
            if hasattr(cls, key):
                setattr(cls, key, value)


class General_Config(Config):
    """Configuration class for General MCP"""

    # Searxng
    SEARXNG_HOST="http://192.168.168.1:40032/"
    SEARXNG_MAX_RESULTS=10


class Material_Config(Config):


    # Materials Project
    MP_API_KEY = 'PMASAg256b814q3OaSRWeVc7MKx4mlKI'
    MP_ENDPOINT = 'https://api.materialsproject.org/'
    MP_TOPK = 3

    LOCAL_MP_PROPS_ROOT = '/home/ubuntu/sas0/LYT/paper_dataset/mp_cif/Props/'
    LOCAL_MP_CIF_ROOT = '/home/ubuntu/sas0/LYT/paper_dataset/mp_cif/MPDatasets/'
    # Proxy
    HTTP_PROXY = ''#'http://192.168.168.1:20171'
    HTTPS_PROXY = ''#'http://192.168.168.1:20171'

    # FairChem
    FAIRCHEM_MODEL_PATH = '/home/ubuntu/sas0/lzy/multi_mcp_server/sci_mcp/material_mcp/support/pretrained_models/fairchem_ckpt/eqV2_86M_omat_mp_salex.pt'
    FMAX = 0.05

    # MatterGen
    MATTERGENMODEL_ROOT = '/home/ubuntu/sas0/lzy/multi_mcp_server/sci_mcp/material_mcp/support/pretrained_models/mattergen_ckpt'
    MATTERGEN_ROOT='/home/ubuntu/sas0/lzy/multi_mcp_server/sci_mcp/material_mcp/mattergen_gen/mattergen'
    MATTERGENMODEL_RESULT_PATH = 'results/'

    # Dify
    DIFY_ROOT_URL = 'http://192.168.191.101:6080'
    DIFY_API_KEY = 'app-IKZrS1RqIyurPSzR73mz6XSA'

    #temp root
    TEMP_ROOT = '/home/ubuntu/sas0/lzy/multi_mcp_server/temp/material'


class Chemistry_Config(Config):


    TEMP_ROOT = '/home/ubuntu/sas0/lzy/multi_mcp_server/temp/chemistry'

    RXN4_CHEMISTRY_KEY='apk-8928522a146c2503f30b16d9909222d7583f412ee8f1049f08d32a089ba88d34'






material_config = Material_Config()
general_config = General_Config()
chemistry_config = Chemistry_Config()