local-transcription/client/config.py

"""Configuration management for the local transcription application."""

import os
import yaml
from pathlib import Path
from typing import Any, Dict, Optional


class Config:
    """Manages application configuration with YAML file storage."""

    def __init__(self, config_path: Optional[str] = None):
        """
        Initialize configuration.

        Args:
            config_path: Path to configuration file. If None, uses default location.
        """
        self.app_dir = Path.home() / ".local-transcription"
        self.app_dir.mkdir(parents=True, exist_ok=True)

        if config_path is None:
            self.config_path = self.app_dir / "config.yaml"
        else:
            self.config_path = Path(config_path)

        self.config: Dict[str, Any] = {}
        self.load()

    def load(self) -> None:
        """Load configuration from file or create default if not exists."""
        if self.config_path.exists():
            with open(self.config_path, 'r') as f:
                self.config = yaml.safe_load(f) or {}
        else:
            # Load default configuration
            default_config_path = Path(__file__).parent.parent / "config" / "default_config.yaml"
            if default_config_path.exists():
                with open(default_config_path, 'r') as f:
                    self.config = yaml.safe_load(f) or {}
            else:
                self.config = self._get_default_config()

            # Save the default configuration
            self.save()

    def save(self) -> None:
        """Save current configuration to file."""
        with open(self.config_path, 'w') as f:
            yaml.dump(self.config, f, default_flow_style=False, indent=2)

    def get(self, key_path: str, default: Any = None) -> Any:
        """
        Get configuration value using dot notation.

        Args:
            key_path: Dot-separated path to config value (e.g., "audio.sample_rate")
            default: Default value if key not found

        Returns:
            Configuration value or default
        """
        keys = key_path.split('.')
        value = self.config

        for key in keys:
            if isinstance(value, dict) and key in value:
                value = value[key]
            else:
                return default

        return value

    def set(self, key_path: str, value: Any) -> None:
        """
        Set configuration value using dot notation.

        Args:
            key_path: Dot-separated path to config value (e.g., "audio.sample_rate")
            value: Value to set
        """
        keys = key_path.split('.')
        config = self.config

        # Navigate to the parent dict
        for key in keys[:-1]:
            if key not in config:
                config[key] = {}
            config = config[key]

        # Set the value
        config[keys[-1]] = value
        self.save()

    def _get_default_config(self) -> Dict[str, Any]:
        """Get hardcoded default configuration."""
        return {
            'user': {
                'name': 'User',
                'id': ''
            },
            'audio': {
                'input_device': 'default',
                'sample_rate': 16000,
                'chunk_duration': 3.0
            },
            'noise_suppression': {
                'enabled': True,
                'strength': 0.7,
                'method': 'noisereduce'
            },
            'transcription': {
                'model': 'base',
                'device': 'auto',
                'language': 'en',
                'task': 'transcribe'
            },
            'processing': {
                'use_vad': True,
                'min_confidence': 0.5
            },
            'server_sync': {
                'enabled': False,
                'url': 'ws://localhost:8000',
                'api_key': ''
            },
            'display': {
                'show_timestamps': True,
                'max_lines': 100,
                'font_size': 12,
                'theme': 'dark'
            }
        }

    def reset_to_default(self) -> None:
        """Reset configuration to default values."""
        self.config = self._get_default_config()
        self.save()

    def __repr__(self) -> str:
        return f"Config(path={self.config_path})"