#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from typing import Any
from AnP.Controllers.AIController import AIController
from AnP.Drivers.WebSocketServerDriver import WebSocketServerDriver
from AnP.Drivers.HTTPSocketServerDriver import HTTPSocketServerDriver
from AnP.Drivers.HTTPServerDriver import HTTPServerDriver
from AnP.Drivers.OllamaDriver import OllamaDriver
from AnP.Drivers.FilesDriver import FilesDriver
from AnP.Drivers.XMLScrapDriver import XMLScrapDriver
from AnP.Drivers.SeleniumDriver import SeleniumDriver
from AnP.Controllers.CloneController import CloneController
from AnP.Controllers.AITranslateController import AITranslateController

inputs:dict[str, dict[str, Any|None]] = {
    "default_models" : {
        "AIController" : AIController, 
        "CloneController" : CloneController,
        "AITranslateController" : AITranslateController,
        "WebSocketServerDriver" : WebSocketServerDriver,
        "HTTPSocketServerDriver" : HTTPSocketServerDriver,
        "HTTPServerDriver" : HTTPServerDriver, 
        "OllamaDriver" : OllamaDriver, 
        "FilesDriver" : FilesDriver, 
        "XMLScrapDriver" : XMLScrapDriver, 
        "SeleniumDriver" : SeleniumDriver
    }
}

try:

    from secrets import secrets as custom_secrets

    for key, value in dict(custom_secrets).items():
        if key not in inputs or isinstance(inputs[key], dict):
            inputs[key] = value
        elif isinstance(value, dict):
            for subkey, subvalue in value.items():
                inputs[key][subkey] = subvalue

except ImportError:
    pass