config/environment_config.py

from dataclasses import dataclass
from typing import Optional

@dataclass
class EnvironmentConfig:
    # Hugging Face configuration
    hf_model_path: str = "TeleologyHI/HIM-self"
    hf_token: Optional[str] = None
    
    # Hardware configuration
    device: str = "cuda"
    num_gpus: int = 1
    mixed_precision: bool = True
    
    # Logging configuration
    log_level: str = "INFO"
    enable_wandb: bool = False
    wandb_project: str = "HIM-self"
    
    # API configuration
    api_host: str = "0.0.0.0"
    api_port: int = 7860
    enable_cors: bool = True