Spaces:
Runtime error
Runtime error
| """ | |
| Configuration manager for persistent settings (API keys, preferences). | |
| """ | |
| import json | |
| import os | |
| from pathlib import Path | |
| from typing import Dict, Optional | |
| from dotenv import load_dotenv | |
| import config | |
| class ConfigManager: | |
| """Manages persistent user configuration.""" | |
| def __init__(self): | |
| self.config_file = config.DATA_DIR / "user_config.json" | |
| self.config_data = self._load_config() | |
| def _load_config(self) -> Dict: | |
| """Load configuration from file.""" | |
| if self.config_file.exists(): | |
| try: | |
| with open(self.config_file, 'r', encoding='utf-8') as f: | |
| return json.load(f) | |
| except Exception: | |
| return self._default_config() | |
| return self._default_config() | |
| def _default_config(self) -> Dict: | |
| """Default configuration.""" | |
| return { | |
| "api_keys": { | |
| "gemini": "", | |
| "nvidia": "", | |
| "groq": "" | |
| }, | |
| "preferences": { | |
| "llm_provider": "gemini", | |
| "temperature": 0.7, | |
| "workflow": "Auto-Detect" | |
| }, | |
| "current_space": "General" | |
| } | |
| def save_config(self): | |
| """Save configuration to file.""" | |
| try: | |
| with open(self.config_file, 'w', encoding='utf-8') as f: | |
| json.dump(self.config_data, f, indent=2) | |
| except Exception as e: | |
| print(f"Error saving config: {e}") | |
| def get_api_key(self, provider: str) -> str: | |
| """Get API key for provider.""" | |
| stored_key = self.config_data.get("api_keys", {}).get(provider, "") | |
| if stored_key: | |
| return stored_key | |
| # Fallback to environment variables so users can configure keys in .env. | |
| # Reloading keeps behavior predictable after .env edits. | |
| for env_file in [config.PROJECT_ROOT / ".env", config.PROJECT_ROOT.parent / ".env"]: | |
| if env_file.exists(): | |
| load_dotenv(env_file, override=False) | |
| env_var_map = { | |
| "nvidia": "NVIDIA_API_KEY", | |
| "gemini": "GEMINI_API_KEY", | |
| "groq": "GROQ_API_KEY", | |
| } | |
| env_var = env_var_map.get(provider, "") | |
| env_value = os.getenv(env_var, "") if env_var else "" | |
| if env_value: | |
| return env_value | |
| # Local development fallback: allow keeping the keys in project-root text files. | |
| if provider == "nvidia": | |
| key_file = config.PROJECT_ROOT.parent / "nvidia_api.txt" | |
| if key_file.exists(): | |
| try: | |
| key = key_file.read_text(encoding='utf-8').strip() | |
| if key: | |
| return key | |
| except Exception: | |
| pass | |
| if provider == "gemini": | |
| key_file = config.PROJECT_ROOT.parent / "gemini_api.txt" | |
| if key_file.exists(): | |
| try: | |
| key = key_file.read_text(encoding='utf-8').strip() | |
| if key: | |
| return key | |
| except Exception: | |
| pass | |
| if provider == "groq": | |
| key_file = config.PROJECT_ROOT.parent / "groq_api.txt" | |
| if key_file.exists(): | |
| try: | |
| key = key_file.read_text(encoding='utf-8').strip() | |
| if key: | |
| return key | |
| except Exception: | |
| pass | |
| return "" | |
| def set_api_key(self, provider: str, api_key: str): | |
| """Save API key for provider.""" | |
| if "api_keys" not in self.config_data: | |
| self.config_data["api_keys"] = {} | |
| self.config_data["api_keys"][provider] = api_key | |
| self.save_config() | |
| def get_preference(self, key: str, default=None): | |
| """Get user preference.""" | |
| return self.config_data.get("preferences", {}).get(key, default) | |
| def set_preference(self, key: str, value): | |
| """Save user preference.""" | |
| if "preferences" not in self.config_data: | |
| self.config_data["preferences"] = {} | |
| self.config_data["preferences"][key] = value | |
| self.save_config() | |
| def get_current_space(self) -> str: | |
| """Get current workspace.""" | |
| return self.config_data.get("current_space", "General") | |
| def set_current_space(self, space_name: str): | |
| """Set current workspace.""" | |
| self.config_data["current_space"] = space_name | |
| self.save_config() | |